При реализации yolo в darknet, должны ли мы тренироваться на наборе данных image net?

Question

Я установил darknet в ubuntu и сейчас пытаюсь реализовать обнаружение объектов с помощью yolo v2 в моем настраиваемом наборе данных.В статье Yolo они рассказали, что они предварительно обучили сеть с использованием набора данных image net.Итак, мой вопрос: стоит ли нам также предварительно обучать сеть?

Извините, если я притупляюсь.Может кто-нибудь ответить мне?

Answer 1

В большинстве случаев, если в вашем наборе данных имеется много похожих функций в предварительно обученном весе (например, человек, автомобиль), вам следует использовать предварительно обученную сеть, например darknet53.conv.74 или darknet19_448.conv.23.

Но вы также можете обучить сеть, не используя эти предварительно обученные сети (обучение с нуля), например, удалив вес из команды:

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