search

pytorch / android-demo-app

PyTorch android examples of usage in applications
1.47k stars 606 forks source link

Object detection yolov5 in Android apps adding extra features #149

Open ys31jp opened 3 years ago

ys31jp commented 3 years ago

I have a question which might need your help or guidance on this. I am new to this. I followed your object detection with yolov5 to implement in android. Right now I'm think to implement this A model + B model then implement on android apps. Example: A is YOLO for food item detection or classification. B is from A the results of detection, and predict is it eatable

Q: how am I going to implement part B?

Thanks in advance if anyone who could guide me.

midasklr commented 3 years ago

I have a question which might need your help or guidance on this. I am new to this. I followed your object detection with yolov5 to implement in android. Right now I'm think to implement this A model + B model then implement on android apps. Example: A is YOLO for food item detection or classification. B is from A the results of detection, and predict is it eatable

Q: how am I going to implement part B?

Thanks in advance if anyone who could guide me.

crop the roi from bitmap, than use a classification model to predict,it is the same as the HelloWordAPP(https://github.com/pytorch/android-demo-app/tree/master/HelloWorldApp)