我已经在ubuntu中安装了darknet,现在正在尝试在我的自定义数据集上使用yolo v2实现对象检测。在yolo论文中,他们告诉我们,他们已经使用图像网络数据集对网络进行了预训练。所以,我的问题是,我们是否也应该预先训练网络?
抱歉,我说得太直白了。有人能回复我吗?
发布于 2019-03-11 01:00:49
在大多数情况下,如果您的数据集在预先训练的权重中有许多相似的特征(例如人、车),则应该使用预先训练的网络,如darknet53.conv.74
或darknet19_448.conv.23
。
但是,您也可以在不使用这些预先训练的网络(从头开始训练)的情况下训练网络,例如,通过从命令中删除权重:
./darknet detector train data/obj.data yolo-obj.cfg
https://stackoverflow.com/questions/55088212
复制