Centernet: Using CenterNet with Torch 1.x

[deandiakov]

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]

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

[LaCandela]

@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]

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

[gtfaiwxm]

@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]

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

[kp-1]

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

[LaCandela]

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

[FranciscoReveriano]

Yeah i just got that issue.

[sohail759]

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

[zhang295498]

Thank you very much!!! I pass it~