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函数的,我这边主要处理了几个,可以参考:
因篇幅问题不能全部显示,请点此查看更多更全内容