99网
您的当前位置:首页Torchvision目标检测模型训练过程记录

Torchvision目标检测模型训练过程记录

来源:99网

1、环境:

pytorch==1.6.0

torchvision==0.7.0

cudatoolkit==10.2

2、场景:

利用torchvision提供的maskrcnn训练自己的数据。

3、数据:转换成coco格式

示例代码:主要是没有mask的数据集,需要用coco函数制作。数据格式一定要对,否则很麻烦。

4、最大的坑是torchvision0.7.0版本有很多bug,都是non_zero函数的,我这边主要处理了几个,可以参考:

因篇幅问题不能全部显示,请点此查看更多更全内容