0

我是一个有一些编程经验的初学者。我正在尝试用我自己的数据集训练暗流。我正在按照这些说明进行操作。

https://github.com/thtrieu/darkflow

到目前为止，我已经完成了以下步骤。

  • 安装了darkflow和相关模块

  • 创建测试图像并进行注释（Pascal VOC）。

               https://ibb.co/y4HmtGz  

           https://ibb.co/GkxLshK

如果我理解正确，暗流训练需要 Pascal VOC？

我的问题是我不知道如何开始培训。如何开始训练过程以及如何测试神经网络是否正常工作？我应该通过训练获得重量吗？

4

1 回答 1

0

您可以从此处选择使用预训练的权重。下载cfg和weights.

假设你安装了 darkflow，你可以像这样训练你的网络：

flow --model cfg/<your-config-filename>.cfg --load bin/<filename>.weights --train --annotation train/Annotations --dataset train/Images --epoch 100 --gpu 1.0

如果你想从头开始训练你的网络而不使用任何预训练的权重，你可以这样做：

flow --model cfg/<your-config-filename>.cfg --train --annotation train/Annotations --dataset train/Images --epoch 100 --gpu 1.0

训练开始后，模型检查点保存在ckpt目录中。您可以加载最新的检查点并在示例图像上进行测试。

于 2020-03-30T04:14:19.230 回答