Как сделать перевод обучения Darknet YOLOv3 - PullRequest
0 голосов
/ 25 сентября 2019

Я начал обнаруживать объекты на изображении.У меня есть нейронная сеть 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)?Спасибо!

...