@[TOC] (自己的数据集训练detectron)
先建立自己的训练数据:
注:
annotations下放了两个json文件:train.json 和 val.json(里面什么都没有写,因为没有验证集)
train下放的是训练集的原始图片
val下放的是验证集的原始图片(我的下面什么都没放,因为没有验证集)
1. 选择的训练网络:
我选择的是:configs/12_2017_baselines/e2e_mask_rcnn_R-50-FPN_1x.yaml
2. 修改 datasets/dataset_catalog.py
在datasets 里面加上自己的数据集
##3. 修改 configs/12_2017_baselines/e2e_mask_rcnn_R-50-FPN_1x.yaml
4. 运行代码
python tools/train_net