【发布时间】:2019-03-10 13:32:51
【问题描述】:
我已经在 ubuntu 中安装了暗网,现在正尝试在我的自定义数据集上使用 yolo v2 实现对象检测。在 yolo 论文中,他们说他们已经使用图像网络数据集对网络进行了预训练。那么,我的问题是我们是否也应该对网络进行预训练?
对不起,如果我生硬了。 有人可以回复我吗?
【问题讨论】:
标签: conv-neural-network object-detection yolo
我已经在 ubuntu 中安装了暗网,现在正尝试在我的自定义数据集上使用 yolo v2 实现对象检测。在 yolo 论文中,他们说他们已经使用图像网络数据集对网络进行了预训练。那么,我的问题是我们是否也应该对网络进行预训练?
对不起,如果我生硬了。 有人可以回复我吗?
【问题讨论】:
标签: conv-neural-network object-detection yolo
对于大多数情况,如果您的数据集在预训练的权重中有很多相似的特征(例如人、车),您应该使用预训练的网络,例如 darknet53.conv.74 或 darknet19_448.conv.23。
但您也可以在不使用那些预训练网络的情况下训练网络(从头开始训练),例如通过从命令中删除权重:
./darknet detector train data/obj.data yolo-obj.cfg
【讨论】: