在detectron2的基础上, 训练mask-rcnn模型，完成工业视觉任务
分割管道中的绳子
- 安装 labelme
- 安装 Detectron2
- 运行
labelme选择polygons 进行分割标, 得到annotation.json - 注册
detectron2可用的数据集具体参见dataset.py - 训练 mask-rcnn, 参见
train.py需要修改的
cfg.merge_from_file模型选择cfg.DATASETS.TRAIN训练数据集为步骤4中注册的数据集cfg.MODEL.WEIGHTS预训练模型位置cfg.MODEL.ROI_HEADS.NUM_CLASSES类别总数
- infer 过程
infer.py
