[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에서 계속 컴파일할 수 없음

이 내용에 흥미가 있습니까?

현재 기사가 여러분의 문제를 해결하지 못하는 경우 AI 엔진은 머신러닝 분석(스마트 모델이 방금 만들어져 부정확한 경우가 있을 수 있음)을 통해 가장 유사한 기사를 추천합니다:

ff-faster-rcnn 코드 이해

9개의 anchor가 있기 때문에 각각의 anchor는 4개의 값이 있는데 이 네 개의 값은 예측된 tx,ty,tw,th이다.그래서 이 출력feature의height와width는conv5-3의 크기와 일치합니다.이것...

텍스트를 자유롭게 공유하거나 복사할 수 있습니다.하지만 이 문서의 URL은 참조 URL로 남겨 두십시오.

CC BY-SA 2.5, CC BY-SA 3.0 및 CC BY-SA 4.0에 따라 라이센스가 부여됩니다.

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

좋은 웹페이지 즐겨찾기