tensorflow - 将生成的检查点转换为 .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

}

我想将“加载”键加载为 .weights 文件。

Answer 1

训练好的检查点文件 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