将生成的检查点转换为 .weights darkflow

Question

我已经使用 Darkflow 和 yolov2.weights 完成了定制培训。检查点 folder.These 中有四个文件是：

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

我想转换这些文件.weights?

目前检查点加载如下：

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

}

我想将 'load' 密钥加载为 .weights 文件。

Answer 1

训练好的checkpoint文件yolov2-3c-5500.data文件其实就是你的权重文件。如果要转换为 .pb 文件，请使用以下命令

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

例如：flow --model cfg/Your_ConfigFile.cfg --load 5500 --savepb

Converting checkpoints generated to .weights darkflow