Для начала я перешел по этой ссылке , и обучение проходит без ошибок.
Однако после тренировки модель ничего не обнаруживает, даже когда я ее даю примеры из учебного набора.
Я использую очень маленький набор данных из 5 обучающих выборок и 1 проверочного образца, просто чтобы посмотреть, смогу ли я запустить модель, независимо от переоснащения. Есть только 1 класс. И я использую крошечную архитектуру YOLO v3.
Я сделал следующие шаги по ссылке:
- Скопируйте
yolov3-tiny.cfg - Настройте размер партии и подразделение:
batch=5, subdivison=5 - Настройка фильтров:
filter=18 - Настройка классов:
classes=1
И я также читал, что якорь Ящики должны быть скорректированы (без дополнительных подробностей), поэтому я добавил 2 якоря с размером не менее объекта, который я хочу обнаружить:
anchors = 81,82, 135,169, 344,319, 431,432
И отрегулировал входное разрешение в соответствии с разрешением моих изображений подробнее:
Для тестирования я сократил шаги до:
Что еще нужно настроить, чтобы тест выполнялся?
Я работаю YOLOv3 на моем процессоре в данный момент (да, ужасно медленный), и когда я его собрал, make-файл был сконфигурирован с GPU=0 и CUDNN=0.
Мои данные в настоящее время состоят из созданного документа Это как изображения с точно таким же объектом, просто в разных местах текста, так что ничего особенного.