Блог

Я ищу оптимальный способ обучения предварительно обученных моделей для YOLOv4 У меня есть локальная среда Debian 10 OS,

• GeForce RTX 2060 SUPER
• GeForce GTX 750 Ti

Я планирую обучать модели на основе изображений разного размера и обученных моделей, которые я собираюсь использовать как часть микросервисов, разработанных на Java или python.

Должен ли я использовать какие-либо сторонние сервисы, такие как Google colab? и второй вопрос, какую структуру лучше использовать? (pytorch, tensorflow и т. Д.)

Спасибо за предложение

Вы можете использовать это репозиторий, https://github.com/AlexeyAB/darknet . В нем есть все инструкции по пользовательскому обучению, обучению передаче, а также обучению colab, сценарию вывода.

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

Пользовательские инструкции по обучению,

https://github.com/AlexeyAB/darknet#how-to-train-to-detect-your-custom-objects

Обучение и вывод в colab,

https://colab.research.google.com/drive/1_GdoqCJWXsChrOiY8sZMr_zbr_fH-0Fg