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

Question

我使用Darkflow和yolov2.weights完成了定制培训。 checkpoint文件夹中有四个文件。这些文件是：

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

}

我想加载＆＃39;加载＆＃39;键为.weights文件。

Answer 1

经过训练的检查点文件yolov2-3c-5500.data文件实际上是您的体重档案。如果要转换为.pb文件，请使用以下命令

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

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