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 GitHub - cocodataset/cocoapi: COCO API - Dataset @ http://cocodataset.org/
• 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 GitHub - CharlesShang/DCNv2: Deformable Convolutional Networks v2 with Pytorch
• 下载的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 GitHub - Duankaiwen/CenterNet: Codes for our paper "CenterNet: Keypoint Triplets for Object Detection" .  #需要翻墙下载#
• 编译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