AlexeyAB / darknet

YOLOv4 / Scaled-YOLOv4 / YOLO - Neural Networks for Object Detection (Windows and Linux version of Darknet )
http://pjreddie.com/darknet/
Other
21.76k stars 7.96k forks source link

Convert darknet weights to tensorflow #6699

Open vqbang opened 4 years ago

vqbang commented 4 years ago

Hi @AlexeyAB, I'm trying to find a repo to convert the weights that I'd trained using your repo, with a custom config based on yolov3.cfg file. After some researches, these are the repos that I've tried:

But It seems like none of them have the same result as yours. The most accurate one is mystic123, but still have some slight differences (about 2-3% accuracy) per prediction. Have you tried to convert a successful one to tensorflow with no changes in result? Have a good day my friend :D

GotG commented 4 years ago

I have the same problem. My original yolo model has more detections (on the same image) in darknet than in the converted tensorflow model using mystic123. It would be nice to obtain (more) similar results on the same image with both original and converted model.