[centernet] 컴파일centernet은pytorch1을 지원합니다.2.0

설명:centernet 프로젝트는pytorch0.4.1에서 테스트한 것은pytorch1에 있습니다.2.0에서 컴파일합니다.주요 문제는 DCNv2 확장입니다.
전재: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

설명:

여기는 dcnv2_cuda.cu, DCNv2 프로젝트가 업데이트되었기 때문

가능한 문제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에서 계속 컴파일할 수 없음