我已在ubuntu中安装了darknet,现在正尝试在自定义数据集上使用yolo v2实施对象检测。在yolo论文中,他们告诉他们已经使用图像网络数据集对网络进行了预训练。所以,我的问题是我们还应该对网络进行预训练吗?
抱歉,如果我迟钝了。 有人可以回复我吗?
答案 0 :(得分:1)
在大多数情况下,如果您的数据集在预训练权重(例如人,汽车)中具有很多相似的特征,则应使用诸如darknet53.conv.74
或darknet19_448.conv.23
之类的预训练网络。
但是您也可以在不使用那些预先训练的网络(从头开始训练)的情况下训练网络,例如,通过消除以下命令的权重:
./darknet detector train data/obj.data yolo-obj.cfg