小小甜菜centernet爬坑记

小小甜菜centernet爬坑记

晴天霹雳，yolov3作者宣布退出CV界用来表达对老美将他的算法用于军事，本萌新看着永远搬不了yolov4，只好转而投入centernet的怀抱了。

CenterNet

照着官方说明安装即可，注意的是DCNv2安装:

cd CenterNet\src\lib\external
python3 setup.py build_ext --inplace
cd CenterNet\src\lib\models\networks
rm -rf DCNv2
git clone https://github.com/CharlesShang/DCNv2
cd DCNv2
python3 setup.py build develop

使用及训练

python3 demo.py ctdet --demo webcam --load_model ../models/ctdet_coco_dla_2x.pth
python3 demo.py ctdet --demo webcam --load_model ../models/ctdet_coco_dlav0_1x.pth

官方指定训练方法1
官方指定训练方法2
我个人喜欢训练不带DCN的模型，方便在服务器部署。

#在opts.py的61行
self.parser.add_argument('--arch', default='dlav0_34',

因疫情影响暂时用不了服务器，后续未完