Я начал обнаруживать объекты на изображении.У меня есть нейронная сеть YOLOv3 с платформой Darknet.Сеть предварительно обучена на основе набора данных COCO.Теперь мне нужно пройти обучение для того, чтобы улучшить результаты.Что у меня есть:
Я создал свой пользовательский набор данных и перенес обучение, следуя этому руководству: https://github.com/ultralytics/yolov3/wiki/Example:-Transfer-Learning
Через несколько часов он выплюнул какой-то файл .pt
.Из того, что я понял, .pt
- это формат PyTorch, но моя программа использует .weights
, который является форматом Darknet.Сначала я попытался найти способ преобразования .pt
в .weights
и наткнулся на это: https://github.com/marvis/pytorch-caffe-darknet-convert, но, видимо, это работает только с YOLOv2.
Затем я попытался что-то искатьо том, как переводить обучение с помощью Darknet, но безуспешно.
С тех пор я застрял и не могу двигаться вперед, кто-нибудь из вас может дать мне несколько советов и указаний о том, как добиться перевода обучения на YOLOLv3 +Darknet (или как конвертировать .pt
в .weights
, так как я уже знаю, как создать .pt
)?Спасибо!