6 ms·
It's a bit disappointing, when caffe2 was released it was stated that mobile is a big focus, but things like this don't support mobile (even though some of this
by haeffin 9y ago
It's a bit disappointing, when caffe2 was released it was stated that mobile is a big focus, but things like this don't support mobile (even though some of this was demoed by FB on a phone).
- ajtulloch 9y agoFWIW we have some similar R-CNN models using this codebase running in the FB apps on iOS/Android with Caffe2 - with some neat bells and whistles (eg full execution of the compute graph on the device GPU where available via Metal/OpenGL). I’ll look into adding a mobile tutorial for Detectron if you’re interested.
- lovelearning 9y agoNot the OP, but +1 from me for such a mobile tutorial. And thank you in advance!
- haeffin 9y agoThat would be awesome. But - is a tutorial enough? From what I gather from the readme the custom ops have cuda implementations and no CPU ones, which you'd like to have as a fallback on mobile (even if metal/opengl implementations exist), or am I wrong?
- ajtulloch 9y agoThey exist and are (nearly all) open-sourced for CPU & Metal (I wrote some of them).
- haeffin 9y agoNice - the "Detectron operators currently do not have CPU implementation; a GPU system is required." on https://github.com/facebookresearch/Detectron/blob/master/INSTALL.md https://github.com/facebookresearch/Detectron/blob/master/IN... is outdated then? Or is it the "nearly all" that's behind this statement?
- stzpz 9y agoWe have open sourced several operators for detectron so that the model could run on GPU/CPU/mobile (https://github.com/facebookresearch/Detectron/commit/757d77c239fbef9d6a341cdfcea2f43219175709 https://github.com/facebookresearch/Detectron/commit/757d77c...). We also open sourced a conversion tool (https://github.com/facebookresearch/Detectron/commit/757d77c239fbef9d6a341cdfcea2f43219175709 https://github.com/facebookresearch/Detectron/commit/757d77c...) to help converting the model.