训练黑网立即完成

Question

我想使用yolo体系结构进行对象检测。在用我的定制数据训练网络之前，我按照以下步骤对Pascal VOC数据进行了培训：https://pjreddie.com/darknet/yolo/

说明非常清楚。 但经过最后一步

./darknet detector train cfg/voc.data cfg/yolo-voc.cfg darknet19_448.conv.23

暗网立即停止训练，并宣布权已被写入backups/目录。

起初我以为预训练太好了，而且会立刻达到停止标准。 所以我用./darknet detect命令在这些测试图像data/dog之一上的权重。没有发现。

如果我不使用任何预训练的权重，网络确实训练。 我已经编辑CFG/YOLO-voc.cfg使用

# Testing 
#batch=1 
#subdivisions=1 
# Training 
batch=32 
subdivisions=8 

现在的训练过程已经捉迷藏多个小时，并让我的GPU温暖。

这是培养黑网的预期方式吗？ 如何正确使用预训练砝码，而不需要进行训练？

是否有任何设置来创建检查点，或获取进度的想法？

Answer 1

这是一个老问题，所以我希望你现在有你的答案，但这里是我的，以防万一它有帮助。

在使用darknet大约一个月后，我遇到了人们在论坛上发帖或发帖的大部分障碍。在你的情况下，我非常确定这是因为已经为最大批次数量训练了权重，并且当在黑网中读取预先训练的权重时假定训练已经完成。

相关个人经验：当我使用预训练的权重文件之一时，它从迭代40101开始，一直运行到40200，然后切断。

如果您有自定义数据，我会坚持从头开始培训，但如果您想再次尝试预先训练的权重，您可能会发现更改cfg文件中的最大批次有助于。