search

chenyuntc / dsod.pytorch

Reimplement DSOD: Learning Deeply Supervised Object Detectors from Scratch.
MIT License
70 stars 12 forks source link

About nms #5

Open zeroMrCc opened 5 years ago

zeroMrCc commented 5 years ago

I have met the problem: ImportError: No module named 'torchcv.models.ssd.nms._ext' how to implement the nms module?

ShangyinGao commented 5 years ago

got the same error, anybody have a solution?

ShangyinGao commented 5 years ago

just figured out the reason. ._ext includes a gpu version of box_nms, to use this method, you should first compile the source code in torchcv/models/ssd/nms/ by using make.sh

another solution is just comment line 9 in box_coder.py and change line 122 in same file to keep = box_gpu(torch.cat([box,score[:,None]],1).cuda(),nms_thresh).cpu() as here