python - 如何更改 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 的第 70 行（https://github.com/thtrieu/darkflow/blob/5fe906b347ab8584354e375a14ea2e7bdf5188d7/darkflow/net/flow.py#L70）

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 步骤时保存检查点。