配置环境
-
按照我上一篇unbiased teacher复现博客,配置pytorch和安装detectron的依赖包(不用下载detectron,只需要安装依赖包即可)
-
下载OWOD代码
-
进入项目文件夹,激活环境,输入语句配置detectron
python setup.py build develop
-
会出现两个包的报错,
No module named 'reliability'
和No module named 'shortuuid'
,分别使用pip install reliability
和pip install shortuuid
安装即可。
下载数据集和模型
- 下载PASCAL VOC 2007的train/val数据集(Image和Anno)
- 下载好的数据按照这个回答放到正确路径下:
https://github.com/JosephKJ/OWOD/issues/59#issuecomment-897747744 - 下载预训练模型,这个回答里面有预训练模型的链接https://github.com/JosephKJ/OWOD/issues/86,将模型下载好后,将模型路径替换这里的weight:
训练代码
我只有一块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"
即可完成训练
所有的训练sh链接
https://github.com/JosephKJ/OWOD/blob/master/run.sh