search

gliese581gg / YOLO_tensorflow

tensorflow implementation of 'YOLO : Real-Time Object Detection'
Other
1.72k stars 656 forks source link

Why not use .pb format weights? #36

Open anonym24 opened 6 years ago

anonym24 commented 6 years ago

Why not use .pb format weights? (https://github.com/thtrieu/darkflow)

APratham commented 6 years ago

According to me, weights is generated by the neural network itself during training. So to generate the weights file you would need to train the network on Darknet (as said in readme.md of this repo). Later on for the file to be usable by the Neural Net you need to convert it to a specific format (.ckpt). I guess if you could tweak the code to accept .pb instead, you can use the files