Как потренировать крошечный йолов2 с тензорным потоком? - PullRequest
Новая
       23

Как потренировать крошечный йолов2 с тензорным потоком?

0 голосов
/ 27 июня 2019

Я действительно мало знаю о машинном обучении.Я только что скачал плагин tenorflow sharp для единства и попробовал его с предварительно обученной моделью yolov2.Теперь я хочу обучить свою собственную модель обнаружению определенного типа объекта.

Я действительно чувствую себя инопланетянином.Что я должен делать?Нужно ли учить «тензорный поток»?Что на самом деле означает «тренировка yolov2 с tenorflow»?

Я нашел хорошую статью здесь: https://timebutt.github.io/static/how-to-train-yolov2-to-detect-custom-objects/

Но если я не ошибаюсь, он тренируется с даркнетом, а не с тензорным потоком.Поэтому я думаю, что не могу использовать вывод с плагином tenorflowsharp.Я не мог найти простой учебник по этой теме.Любая помощь будет оценена ..

1 Ответ

1 голос
/ 28 июня 2019

Хорошо.Для новичков, как я, вот что вы должны сделать:

Алгоритм YoloV2, написанный в Darknet.Darknet - это инфраструктура нейронной сети с открытым исходным кодом, написанная на C и CUDA.Если вы хотите использовать YoloV2 с плагином unity tenorflowsharp, вам нужна реализация Tensorflow YoloV2.

И darkflow .(Darknet + Tensorflow = Darkflow. Забавно, а?) Делает свою работу.Итак, вот схема того, что вы должны сделать, чтобы обучить свой собственный алгоритм yolov2 использовать в единстве с tenorflow:

1-) Установить среду anaconda и python с tenorflow 2-) Скачать darkflow с github 3-)Обучите yolov2 с darkflow 4-) Конвертируйте обучающие файлы в .pb, затем .bytes 5-) Используйте .bytes with tenorflowsharp

Для первых 3 шагов я настоятельно рекомендую серию видео, начинающуюся с этого: https://www.youtube.com/watch?v=PyjBd7IDYZs

Надеюсь, это поможет.Не стесняйтесь комментировать, когда вы застряли.

...