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

how to generate checkpoint file .ckpt? #584

Open liuhantao9 opened 6 years ago

liuhantao9 commented 6 years ago

@thtrieu I saw that when generating .pb file. The check point will be stored into the pb file. In order to get the frozen GraohDef file, I will need both .ckpt and .pb. How am I able to get .ckpt which stores serialized variables from a TensorFlow graph?

prateethvnayak commented 5 years ago

@thtrieu Using the following command flow --model cfg/yolo.cfg --load bin/yolo.weights --savepb I was able to get a.pb and .meta file. Is this .pb file equivalent to the Frozen tensorflow Graph ? where the variables are converted to constants ?

If not, how do I get the frozen graph, since I don't have the .ckpt file for using the freeze_graph tensorflow tool ?