thtrieu / darkflow

Translate darknet to tensorflow. Load trained weights, retrain/fine-tune using tensorflow, export constant graph def to mobile devices
GNU General Public License v3.0
6.13k stars 2.08k forks source link

No bounding boxes are drawn during real time custom object detection in Andorid YoloV2 #800

Open medasuryatej opened 6 years ago

medasuryatej commented 6 years ago

I have trained ( Github repo link: https://github.com/thtrieu/darkflow.git) using pre-trained weights (tiny-yolo-voc.weights) for 2 of my custom objects, over 30000 steps with a loss of 0.8. When tested on windows prompt using python, Yolo was able to detect the objects and perfectly draw the bounding boxes. image

I saved the weights as .pb files and when tested on android (using the https://github.com/szaza/android-yolo-v2), No bounding boxes are drawn, the android real time is not detecting my custom object at all. Also I don't see any error or exception message on the log-cat, what could be the issue

Any reference links for custom object detection on Android using yolo V2 would be very helpful

darkflow #Yolov2 #android #object-detection

xinyee1997 commented 4 years ago

do u sove the problem?