参考上一篇博客detectron2-入门安装,然后进行以下操作:
1、首先在OWOD文件目录运行以下代码,完成内部构建
python -m pip install -e ./
2、数据集准备
首先下载数据集Annotations和JPEGImages:下载链接,需要翻墙,下载方法可以看我的博客,然后:
(1)主文件夹下新建datasets/VOC2007文件夹
(2)下载的Annotations与JPEGImages文件夹放在VOC2007下
(3)新建datasets/VOC2007/ImageSets/Main文件夹
(4)将datasets/OWOD_imagesets下的所有文件放到datasets/VOC2007/ImageSets/Main下
1、首先在OWOD文件目录运行以下代码,完成内部构建
python -m pip install -e ./
2、运行代码,我只有一块gpu
python tools/train_net.py --num-gpus 1 --config-file ./configs/OWOD/t1/t1_train.yaml SOLVER.IMS_PER_BATCH 8 SOLVER.BASE_LR 0.0025 OUTPUT_DIR “./output/t1”
3、出现以下情况
No module named ‘reliability’
解决办法:
pip install reliability
No module named ‘shortuuid’
解决办法:
pip install shortuuid