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

How to use own data training model? #3

Closed xieyulai closed 7 years ago

xieyulai commented 7 years ago

Hi, natanielruiz. I am using darknet to traing my own data. I only detect one class. Could guide me how to modify your code to detect only one class? Do I need to modify the parameters in android-yolo/app/src/main/java/org/tensorflow/demo/TensorflowClassifier.java? or anything else? Thank you.

natanielruiz commented 7 years ago

Unfortunately you have to change some other things to make it work such as the c++ native code output part (the output of the model) and the TensorFlowClassifier.java (which makes sense of the c++ output). Hope this helps.