使用yolov3进行自定义对象检测的预训练权重

时间:2020-06-15 21:30:00

标签: machine-learning deep-learning computer-vision object-detection yolo

在这里深入学习的新手,我正在尝试使用yolov3构建自定义对象检测模型。阅读一些文档/教程后,我发现通常建议使用用于其他数据集(例如ImageNetCOCO)的预先训练权重,即使您的自定义数据/标签与这些数据集没有任何关系。这是真的吗(如果是,为什么)?

我想对非常特定类型的图像(特别是网站的屏幕截图)进行对象检测。即使在这种情况下,我也应该使用预先训练的举重还是最好完全从头开始进行训练?

1 个答案:

答案 0 :(得分:1)

这听起来像您可能需要尝试的东西,但是我过去的经验和社区知识表明值得尝试。您可以检查此堆栈帖子以获取更多说明,但从本质上讲,它使网络有了一个良好的开端,使它可以对自身进行“微调”,从而消除了一些令人讨厌的品质,如果您从头开始训练模型,这些品质可能会使自己知道

https://stats.stackexchange.com/questions/193082/what-is-pre-training-a-neural-network