Search code examples
tensorflowdarkflow

Converting checkpoints generated to .weights darkflow

I have done customized training using Darkflow and yolov2.weights. There are four files in checkpoint folder.These are:

1.yolov2-3c-5500.data-00000-of-00001
2.yolov2-3c-5500.index
3.yolov2-3c-5500.meta
4.yolov2-3c-5500.profile

I want to convert these file .weights?

Currently the checkpoint are loaded like this:

options = {
    'model': 'cfg/yolov2-4c.cfg',
    'load': 5500,
    'threshold': 0.3,
    'gpu':1.0

}

I want to load 'load' key as .weights file.

Solution

  • The trained checkpoint files yolov2-3c-5500.data file is actually your weight file. If you want to convert to .pb file, use the below command

    flow --model cfg/Your_ConfigFile.cfg --load ckpt/Your_required_checkpoint-chkptNumber --savepb

    Ex: flow --model cfg/Your_ConfigFile.cfg --load 5500 --savepb