如何更改 Yolo 2 (darflow) python 中的检查点间隔？

Question

我正在训练 Yolo2 (darkflow) 并想更改间隔（例如每 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 选项可以做到这一点。

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