在Darknet中实施yolo时,我们应该在图像网络数据集上进行训练吗?

时间:2019-03-10 13:32:51

标签: conv-neural-network object-detection yolo

我已在ubuntu中安装了darknet,现在正尝试在自定义数据集上使用yolo v2实施对象检测。在yolo论文中,他们告诉他们已经使用图像网络数据集对网络进行了预训练。所以,我的问题是我们还应该对网络进行预训练吗?

抱歉,如果我迟钝了。 有人可以回复我吗?

1 个答案:

答案 0 :(得分:1)

在大多数情况下,如果您的数据集在预训练权重(例如人,汽车)中具有很多相似的特征,则应使用诸如darknet53.conv.74darknet19_448.conv.23之类的预训练网络。

但是您也可以在不使用那些预先训练的网络(从头开始训练)的情况下训练网络,例如,通过消除以下命令的权重:

./darknet detector train data/obj.data yolo-obj.cfg