detectron2-训练coco数据集
- Detectron是Facebook人工智能研究的软件系统,它实现了最先进的目标检测算法,包括Mask R-CNN。它是用Python编写的,由Caffe2深度学习框架提供支持。
1.下载coco数据集到对应得目录
2.到项目的tools文件夹下面运行train_net.py,我这里只用了一个gpu
./train_net.py \
--config-file ../configs/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_1x.yaml \
--num-gpus 1 SOLVER.IMS_PER_BATCH 2 SOLVER.BASE_LR 0.0025
这里可能会出现找不到train2017文件的情况,进行软链接操作:
ln -s /home/mrp/hyy/OWOD/detectron2/datasets /home/mrp/hyy/OWOD/detectron2/tools
根据自己的目录来设置
3.继续运行代码,可以跑通了!!!
./train_net.py \
--config-file ../configs/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_1x.yaml \
--num-gpus 1 SOLVER.IMS_PER_BATCH 2 SOLVER.BASE_LR 0.0025