Xavier_NX（Xavier）配置CenterNet

1.安装pytorch

下载torch-1.1.0-cp36-cp36m-linux_aarch64.whl

sudo chmod -R 777 torch-1.1.0-cp36-cp36m-linux_aarch64.whl
python3，用以下命令安装：

pip3 install torch-1.1.0-cp36-cp36m-linux_aarch64.whl

python2，用以下命令安装：

pip install torch-1.1.0-cp36-cp36m-linux_aarch64.whl

安装完成后，验证是否安装成功：

终端输入：

python3

import torch

torch.cuda.is_available()

true #显示true说明安装成功#

2.安装cocoapi

git clone https://github.com/cocodataset/cocoapi.git
cd /home/nvidia/cocoapi/PythonAPI
python3 setup.py install –user
手动安装了matplotlib

sudo apt-get install python3-matplotlib或者pip3 install matplotlib

pip3 install pycocotools

终端输入：

python3

import matplotlib #无报错说明安装成功#

安装完成后，验证安装是否成功

终端输入：

pyhon3

import pycocotools #无报错说明安装成功#

3.安装DCNv2

git clone https://github.com/CharlesShang/DCNv2
下载的DCNv2拷贝到/home/nvidia/CenterNet-master/src/lib/models/networks目录下（此目录下已有的DCNv2文件夹删除）
修改make.sh

python3 setup.py build develop

原来的是python setup.py build develop，如果使用的是python2的环境可以不用修改

4.安装编译CenterNet

git clone https://github.com/Duankaiwen/CenterNet #需要翻墙下载#
编译DCNv2
cd /CenterNet-master/src/lib/models/networks/DCNv2
./make.sh
编译NMS

cd CenterNet/src/lib/external

make

5.运行Demo测试

python3 demo.py ctdet --demo ../images/19064748793_bb942deea1_k.jpg --load_model ../models/ctdet_coco_dla_2x.pth

说明：ctdet_coco_dla_2x.pth为模型文件，可以提前下载好。另外运行Demo测试时下载dla34-ba72cf86.pth也需要需要很长时间。

Demo测试结果如下图：

6.报错总结

ImportError: torch.utils.ffi is deprecated. Please use cpp extensions instead.

修改：build.py文件

from torch.utils.ffi import create_extension

修改为:from torch.utils.cpp_extension import BuildExtension

ffi = create_extension修改为：ffi = BuildExtension

ModuleNotFoundError: No module named 'progress'

pip3 install progress

ModuleNotFoundError: No module named 'torchvision'

pip3 install torchvision

编译DCNv2 [Errno 13] Permission denied: '/usr/local/lib/python3.6/dist-packages/test-easy-install-31464.write-test'

sudo ./make.sh

Xavier_NX（Xavier）配置CenterNet

相关推荐