1、前期工作
下载源码https://github.com/matterport/Mask_RCNN
下载coco预训练weight https://github.com/matterport/Mask_RCNN/releases
下载数据集
气球数据集分为训练集和测试集
2、在sample 文件夹下有一个bollon文件夹,这是我们要做的项目
把我们下载好的dataset 和权重文件放到bollon目录下
3、安装依赖环境
pip3 install -r requirements.txt
python3 setup.py install
4、训练数据
注意要先切换到balloon.py文件目录下训练
放模型和数据集的路径要正确
Python ./balloon.py/ train --dataset=./balloon/ --weight=./mask_coco.h5
训练好的模型在logs文件夹下