search

szaza / android-yolo-v2

Android YOLO real time object detection sample application with Tensorflow mobile.
Do What The F*ck You Want To Public License
227 stars 86 forks source link

GPU support for android #25

Open spaul13 opened 5 years ago

spaul13 commented 5 years ago

How to enable GPU support for this project on Android?

szaza commented 5 years ago

In case of tfLite just follow this tutorial: https://www.tensorflow.org/lite/performance/gpu; I did not try it out, however I'm pretty sure that it should work. I think that is quite impossible to add GPU support for the android-yolo-v2 project, because it was written with tensorflow-android and as far as I know it doesn't have any GPU support: https://github.com/tensorflow/tensorflow/tree/master/tensorflow/contrib/android. If it is possible please try to use tfLite, because it is optimized for mobile devices, uses less resources and I believe that tensorflow-android is getting outdated. I would also be very happy, if I could review a merge request on a tfLite-yoloV2 project.

Good luck for your work!