Создание файла весов YOLO для пользовательского набора данных

Question

введите описание изображения здесь Я использовал крошечный YOLO для создания пользовательской модели. 4 файла были сгенерированы в папке ckpt, соответствующей моей последней контрольной точке. Обнаружение объектов работает правильно, когда я даю файл .cfg и номер загрузки.

Я должен использовать cv2.dnn.rea dnet (), чтобы загрузить модель и файл cfg для использования в raspberrypi, но Функция rea dnet принимает только 2 аргумента: файл .weights и файл .cfg. Итак, как мне преобразовать мою модель в один файл весов? введите описание изображения здесь

Я использую python 3.7 и opencv 4.

Answer 1

Это не точный ответ, но используемый вами репозиторий выглядит старым. Если у вас есть данные, вы можете переобучить свою пользовательскую модель в yolo v3. Этот репозиторий содержит всю необходимую информацию для обучения пользовательской модели, включая yolo v3 tiny, https://github.com/AlexeyAB/darknet.

. Это будет генерировать .weights каждые 1000 итераций в папке резервных копий и последние веса для возобновления обучения. , Вы можете использовать эти веса после достаточных итераций вместе с файлом .cfg в cv2.dnn.rea dnet ().