NMS without CUDA - Githubissues

multimodallearning / pytorch-mask-rcnn

Other

2.01k stars 557 forks source link

NMS without CUDA #92

Closed kahya-se closed 4 years ago

kahya-se commented 4 years ago

Hello, I'd like to use Mask-RCNN in a CPU only mode. Yet still, the NMS module has to be installed using CUDA, which seems to me unnecessary, as I don't intend to use a GPU [for testing purposes on this machine]. Isn't it possible to run the model without CUDA (as it is possible in the tf-mask-rcnn)?

kahya-se commented 4 years ago

Duplicate to issue #15

wangutian commented 3 years ago

Have you solved the problem yet？