Как тренировать YOLOv3 только на слоях dar knet -53? - PullRequest
0 голосов
/ 19 января 2020

В настоящее время я работаю с реализацией YOLOv3 из:

https://github.com/pythonlessons/YOLOv3-object-detection-tutorial/tree/master/YOLOv3-custom-training

Пока все работает нормально, однако я хотел бы усовершенствовать свои тренировки и улучшить. У меня есть набор данных, специально предназначенный для обучения классификации, и один, который я хотел бы использовать для обучения обнаружению.

Я читаю статью YOLOv3 и другие блоги, однако не могу выяснить, к каким слоям в этой реализации и к yolov3 в целом относятся. darknet53. Я хотел бы заморозить все слои, кроме dar knet -53, обучить их классификации и после этого заморозить все не обнаруживающие слои и обучить их обнаружению.

Кто-нибудь знает, как выяснить, какие К какой позиции относятся слои, чтобы я мог соответственно изменить код в train.py? Мне трудно это выяснить из yolo.cfg и одной газеты.

Заранее спасибо.

...