项目地址https://github.com/xingyizhou/CenterNet
一、配置环境
基础环境 ubuntu16.04 cuda9.2 gtx1080ti
- PYTHON虚拟环境搭建
conda create -n CenterNet python=3.6
- 安装pytorch版本0.4.1
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ #设置清华源
conda install pytorch=0.4.1 cuda92 #安装对应版本的torch
conda install torchvision
- 修改functional.py中的torch.batch_norm函数中的torch.backends.cudnn.enabled 为False
- 安装requirements.txt中的依赖项
遇到问题
安装numba的时候遇到的问题，看了下是LLVM的版本不对。从官网选择对应的LLVM版本https://github.com/llvm/llvm-project/releases下载。一定要下载10.0.1 这个版本，我下载了11.0.1不行
下载之后解压，然后设置LLVM_CONFIG路劲
export LLVM_CONFIG=/home/nemo/software/package/clang+llvm-10.0.1-x86_64-linux-gnu-ubuntu-16.04/bin/llvm-config - 编译
cd $CenterNet_ROOT/src/lib/models/networks/DCNv2
./make.sh
二、运行测试代码
- 下载模型
https://github.com/xingyizhou/CenterNet/blob/master/readme/MODEL_ZOO.md
放到models目录下 - 测试
直接下载太慢了，这里复制了下载地址，然后用迅雷下载很快就下载好了。下载之后把模型文件放到指定的目录下就可以了。
python demo.py ctdet --demo /home/nemo/DeepLearningFramework/CenterNet/images/33823288584_1d21cf0a26_k.jpg --load_model ../models/ctdet_coco_dla_2x.pth
