search

xingyizhou / CenterNet

Object detection, 3D detection, and pose estimation using center point detection:
MIT License
7.25k stars 1.92k forks source link

Using CenterNet with Torch 1.x #446

Open deandiakov opened 4 years ago

deandiakov commented 4 years ago

Has anyone had any experience / success porting CenterNet to work with PyTorch 1.0 or higher? From the attempts I've done there seems to be errors with a missing Nvidia library link every time. Please let me know!

mathpopo commented 4 years ago

And I use https://github.com/CharlesShang/DCNv2.git replace your code pytorch1.2 cuda10.0 is ok!!!!!!!!!!!!!!!!!!!

LaCandela commented 4 years ago

@mathpopo HI! Did you just simply replace the CenterNet/src/lib/models/networks/DCNv2 directory with the pytorch1.0 version of DCNv2 taht you linked above? Does it generate the models.networks.DCNv2._ext module as well which is reguired to run for example the demo.py script?

mathpopo commented 4 years ago

@LaCandela yes, i just mv the old DCNv2 file folder,replace the new git code ,left else operations also as same as before.

gtfaiwxm commented 4 years ago

@mathpopo After I simply replace the CenterNet/src/lib/models/networks/DCNv2 directory, the command line returns the Segmentation fault (core dumped) error. Could you help me?

LaCandela commented 4 years ago

@mathpopo thanks, it works fine! with pytorch1.2 cuda10.0

kp-1 commented 4 years ago

excuse me,dose anyone have this issue:ModuleNotFoundError: No module named '_ext'?

LaCandela commented 4 years ago

@kp-1 Did you install the deformable convolution repo?

FranciscoReveriano commented 4 years ago

Yeah i just got that issue.

sohail759 commented 4 years ago

RESOLVED https://www.gitmemory.com/issue/xingyizhou/CenterNet/7/486653333

zhang295498 commented 4 years ago

Thank you very much!!! I pass it~