0

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

https://github.com/thtrieu/darkflow

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

  • 安装了darkflow和相关模块

  • 创建测试图像并进行注释(Pascal VOC)。

               https://ibb.co/y4HmtGz  
    
               https://ibb.co/GkxLshK
    

如果我理解正确,暗流训练需要 Pascal VOC?

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

4

1 回答 1

0

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

假设你安装了 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 回答