Mask R-CNN训练自己的数据集（pytorch）

1、使用labelme进行标注。

关于labelme的安装与使用，可以参考：labelme安装使用教程_哔哩哔哩_bilibili

labelme安装完成后，参考(93条消息) 45、实例分割的labelme数据集转coco数据集以及coco数据集转labelme数据集_天昼AI实验室的博客-CSDN博客_labelme 实例分割

将将标注文件转化成coco格式。

2、转化完成后

借助Mask R-CNN源码解析（Pytorch）_哔哩哔哩_bilibili​​​​​​

进行代码调试。

跑通自己的代码前，一定要先跑通开源的coco数据集的代码，跑通方法，霹雳巴拉up主有讲。如果能够跑通，则可以跑一下自己标注的数据集。

踩坑：

跑自己标注的数据集时，一定要将num_class的类别个数进行更换。否则，将会出现：if 'caption' in anns[0]: IndexError: list index out of range这样的报错

如果标注完数据，运行labelme2coco.py提示缺少某种包，对我来说，是忘记用conda的虚拟环境。