开发环境搭建

  • Python 3.6
  • PyTorch v0.4.1.
  1. 创建一个虚拟环境并激活
  1. # 创建一个虚拟环境
  2. conda create --name CenterNet python=3.6
  3. # 激活环境
  4. conda activate CenterNet
  1. 安装 PyTorch
  1. # 安装 pytorch0.4.1
  2. conda install pytorch=0.4.1 torchvision -c pytorch
  1. 安装 COCOAPI
  1. git clone https://github.com/cocodataset/cocoapi.git 
  2. cd cocoapi/PythonAPI
  3. make
  4. python setup.py install --user
  1. git 项目
  1. git clone https://github.com/FLyingLSJ/CenterNet.git
  2. cd CenterNet
  1. 安装必要包
  1. pip install -r requirements.txt
  1. 编译可形变卷积 (from DCNv2).
  1. cd CenterNet/src/lib/models/networks/DCNv2
  2. ./make.sh
  1. [可选：如果想使用多尺度进行训练或者测试的话] 编译 NMS
  1. cd CenterNet/src/lib/external
  2. make
  1. 下载预训练模型（只针对目标检测）
  1. cd CenterNet/models
  2. sh download_model.sh
  1. 测试
  1. python demo.py ctdet --demo ./images --load_model ../models/ctdet_coco_dla_2x.pth