0、首先创建文件夹。
a、mkdir detect
b、cd detect
1、安装conda。
a、./Anaconda3-2019.10-Linux-x86_64.sh
2、创建并激活CenterNet环境。
a、conda create --name CenterNet python=3.7
b、conda activate CenterNet
3、安装pytorch。
a、conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
4、安装CocoAPI。
a、git clone https://github.com/cocodataset/cocoapi.git
b、cd PythonAPI
c、make
出现错误:“gcc: error: pycocotools/_mask.c: No such file or directory”
解决方法:pip install cython
d、python setup.py install --user
5、安装CenterNet和其他依赖。
a、cd ../..
b、git clone https://github.com/xingyizhou/CenterNet
c、cd CenterNet
d、pip install -r requirements.txt
6、编译DCNV2。
a、cd src/lib/models/networks/DCNv2
b、./make.sh
出现错误:“ImportError: torch.utils.ffi is deprecated. Please use cpp extensions instead.”
解决方法:cd ..
mv DCNv2/ DCNv2_old/
git clone https://github.com/CharlesShang/DCNv2.git
cd DCNv2
./make.sh
7、编译NMS。
a、cd ../../../external
b、make