【缺陷检测】基于Mask R-CNN框架,训练自己的数据集

基于Mask R-CNN框架,训练自己的数据集

(一):安装labelme

  1. pip install labelme
  2. pip install pyqt5
  3. pip install pillow==4.0.0

(二):标注数据

  1. 使用labelme得到json文件
  2. 使用labelme_json_to_dataset1.json得到json文件夹
  3. 也可以使用批处理脚本得到所有json文件夹
  4. 得到4个文件夹标注信息

(三):代码中需要改动的地方

  1. NUM_CLASSES:类别的个数+1
  2. 后续再更新

你可能感兴趣的:(cnn,python,json)