pytorch / android-demo-app

PyTorch android examples of usage in applications
1.46k stars 604 forks source link

increase live speed detection #221

Open blackCmd opened 2 years ago

blackCmd commented 2 years ago

Is there a way to increase the detection speed during the live object detection?

It is so slower than tensorflow lite object detection.

Avant-Gardiste commented 2 years ago

Yolov5 is too big to run it on a mobile device in reasonable time. I would suggest to train it on Yolov5n (nano) instead of Yolov5s (small).