search

remydubois / lsnms

Large Scale Non maximum Suppression. This project implements a O(nlog(n)) approach for non max suppression, useful for object detection ran on very large images such as satellite or histology, when the number of instances to prune becomes very large.
65 stars 6 forks source link

Will cuda support come? #10

Closed kadirnar closed 2 years ago

kadirnar commented 2 years ago

I want to use lsnms library for Yolo models. But it doesn't work when cuda is active.

Error Message: 

I want to use lsnms library for Yolo models. But it doesn't work when cuda is active.
remydubois commented 2 years ago

Jitclasses are not supported on CUDA yet, as numba's doc states:

Support for jitclasses are available on CPU only. (Note: Support for GPU devices is planned for a future release.)

I will port the implementation on CUDA as soon as jitclass kernels can be implemented