Обученный YOLO, какая итерация у yolo_best.weights? - PullRequest
0 голосов
/ 22 апреля 2020

Я обучал YOLOv3 через систему Dar knet. Каждую 1000 итераций он сохранял веса, но в конце Dar knet оценивает все веса и использует лучшие. Они сохраняются в отдельном файле "yolov3_best.weights".

Я хочу выяснить, какая итерация использовалась для этого файла. Я пробовал до сих пор:

  • использовать весовые коэффициенты в тесте распознавания через терминал и проверять вывод
  • открывает файл best.weights через редактор и ищет его

но я не смог его найти.

У кого-нибудь есть решение?

Спасибо заранее.

1 Ответ

0 голосов
/ 26 апреля 2020

Итак, неясно, как узнать, какой шаг и эпоха итерации использовался для файла best_weights, поэтому я сделал следующее:

Я написал скрипт, который использует "yolov3_best.weights "-файл для обнаружения всех классов в наборе тестов, сравнил это с моими метками-данными и вычислил метрики отзыва, точности и f1. Я сделал это также для всех других весов, которые dar knet сохраняет по умолчанию (каждые 1000 шагов итерации), и сравнил результаты.

В конце я обнаружил, что "yolov3_best.weights" это не лучший показатель для моих метрик, поэтому я выбираю тот, который имеет наибольшее значение отзыва (но другие могут выбирать в соответствии с метрикой c, которая должна быть оптимизирована для необходимого случая).

Надеюсь, это поможет др.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...