Mask-Rcnn-实例分割模型在Keras当中的实现
目录
所需环境
tensorflow-gpu==1.13.1
keras==2.1.5
文件下载
这个训练好的权重是基于coco数据集的,可以直接运行用于coco数据集的实例分割。
链接: 提取码: 15cj
这个数据集是用于分辨图片中的圆形、正方形、三角形的,格式已经经过了处理,可以让大家明白训练集的格式。
链接: 提取码: 9457
训练步骤
1、准备数据集
a、利用labelme标注数据集,注意标注的时候同一个类要用不同的序号,比如画面中存在两个苹果那么一个苹果的label就是apple1另一个是apple2。
b、标注完成后将jpg文件和json文件放在根目录下的before里面。
c、之后运行json_to_dataset.py就可以生成train_dataset文件夹了。
2、修改训练参数
a、dataset.py内修
1