4

我已经使用 Darkflow 和 yolov2.weights 进行了定制训练。检查点文件夹中有四个文件。它们是：

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 文件。

4

1 回答 1

2

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

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

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

于 2018-06-08T10:52:37.427 回答