CenterNet环境搭建

torch 1.3.1

torchvision 0.4.2

按照官方安装步骤:https://github.com/xingyizhou/CenterNet/blob/master/readme/INSTALL.md

改动部分:

     由于torch是1.x以上版本,与官方的0.4不相符,编译DCNv2时会出现导入"torch.utils.ffi"失败的情况

     解决办法:

             进入src/lib/models/networks文件夹下,删除DCNv2文件夹,

             在当前目录下运行: git clone https://github.com/CharlesShang/DCNv2.git

             再进入DCNv2文件夹下进行编译

     还遇到Failed to load module "canberra-gtk-module"

     解决办法:

            sudo apt-get install libcanberra-gtk-module

测试:

      下载官方提供的模型:如  ctdet_coco_dla_2x.pth放在/models文件夹下

       在 src/目录下 运行 python demo.py ctdet --demo ../images/ --load_model ../models/ctdet_coco_dla_2x.pth,出现效果图