如何在Yolo 2（darflow）python中更改检查点间隔？

Question

我正在训练Yolo2（暗流）并希望更改间隔（例如每100步）以制作检查点。

训练命令：

python flow --model cfg/yolo.cfg --load bin/yolo.weights --train --annotation train/XML --dataset train/Images

这样的事情：

--set ckpt 100

我找不到命令。谢谢！

Answer 1

您可以尝试更改文件flow.py（https://github.com/thtrieu/darkflow/blob/5fe906b347ab8584354e375a14ea2e7bdf5188d7/darkflow/net/flow.py#L70）

的第70行

ckpt = (i+1) % (self.FLAGS.save // self.FLAGS.batch)

到

  ckpt = (i+1) % 100

Answer 2

也许我有点迟了，但是没有理由更改代码。在https://github.com/thtrieu/darkflow/blob/5fe906b347ab8584354e375a14ea2e7bdf5188d7/darkflow/defaults.py#L31处，您可以看到有执行此操作的cli选项。

请记住，流将在每次保存// batch_size步骤时保存检查点。