vacancy / PreciseRoIPooling

Precise RoI Pooling with coordinate gradient support, proposed in the paper "Acquisition of Localization Confidence for Accurate Object Detection" (https://arxiv.org/abs/1807.11590).
MIT License
772 stars 152 forks source link

CPU implementation #57

Open maksad opened 3 years ago

maksad commented 3 years ago

I have tried PreciseRoIPooling in my research and was satisfied with the result, now I would like to deploy my model using TorchServe and Amazon Elastic Inference (EI). One of the requirements on EI is "CPU compatibility" and having that variation of PreciseRoIPooling would greatly help. Is there any timeline about it?

vacancy commented 3 years ago

Currently, there is no plan for CPU forward/backward support.

cybaol commented 2 years ago

Is there any plan to achieve CPU version?

evekeen commented 2 years ago

Without a CPU implementation, there is little practical use in the library besides research. I'd really like to try porting it to CPU however, I'm far from understanding how it works. Can anyone help?