search

natanielruiz / android-yolo

Real-time object detection on Android using the YOLO network with TensorFlow
Apache License 2.0
667 stars 212 forks source link

To customize it for other trained model files #21

Closed abhigoku10 closed 7 years ago

abhigoku10 commented 7 years ago

@hashimshafiq @natanielruiz Hi can u pls guide what all parameters to be changed if we change the model file trained for custom object detection. I have converted my model file using darkflow to .pb format i have to modify the current android yolo to detect other objects

natanielruiz commented 7 years ago

Please check the Disclaimer on the project Readme. Cheers

abhigoku10 commented 7 years ago

@natanielruiz i have read the disclaimer of the project readme and its mentioned that it works only for 20 class. Can you list / suggest the steps to modify tat from 20 classes to any other classes , that would be of great help

natanielruiz commented 7 years ago

Hi, I believe the Disclaimer also gives some pointers as to what code to check and change if you were to try to modify it to detect a different number of classes. Also the darkflow model should be tiny-yolo (v1). If you are able to make it more general (be able to work with different number of classes) please make a pull request and I can merge it! Cheers!