【centernet】コンパイルcenternetはpytorch 1をサポートする.2.0

説明:centernetプロジェクトはpytorch 0である.4.1でテストしました.ここでpytorch 1にします.2.0でコンパイルします.主な問題はDCNv 2拡張です.
転載先:https://github.com/xingyizhou/CenterNet/issues/7

1. build nms

cd CenterNet\src\lib\external
#python setup.py install
python setup.py build_ext --inplace

2. clone and build original DCN2

cd CenterNet\src\lib\models
etworks
rm -rf DCNv2
git clone https://github.com/CharlesShang/DCNv2
cd DCNv2

python setup.py build develop

説明:

ここではdcnを変更する必要はありません.v2_cuda.Cu、DCNv 2プロジェクトが

に更新されたため
可能性のある問題error: command '/usr/local/cuda/bin/nvcc' failed with exit status 1

これはgccバージョンの問題によるもので、エラー時に6.3でコンパイルされます.gccを5.4に切り替えるとOKです.

3. test

cd CenterNet/src
python demo.py ctdet --demo ../images/17790319373_bd19b24cfc_k.jpg --load_model ../models/ctdet_coco_dla_2x.pth --debug 2
python demo.py multi_pose --demo ../images/17790319373_bd19b24cfc_k.jpg --load_model ../models/multi_pose_dla_3x.pth --debug 2

4.質問

なぜpytorch 1にいるのか分かりません.1.0では、を介してコンパイルできません.