About nms

[zeroMrCc]

trafficstars

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

[ShangyinGao]

got the same error, anybody have a solution?

[ShangyinGao]

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