Развертывание модели глубокого обучения компьютерного зрения PyTorch в настольном приложении windows

Question

Я обучил модель yolov3 в PyTorch с помощью своего набора данных, а также написал несколько служебных кодов для нее, которые работают вместе с моделью на языке python. Теперь я хочу развернуть эту модель и мои утилиты в настольном приложении windows, которое снимает видео и запускает модель на своих кадрах. Как мне выполнить эту задачу с минимальным изменением кода или переписать его на другом языке? Какой фреймворк является лучшим вариантом для разработки пользовательского интерфейса приложения? Спасибо.

Answer 1

1. Сначала я бы использовал вашу модель с PyTorch для обнаружения каждого кадра и использовал numpy ImageDraw для рисования вокруг вашего объекта (чтобы его можно было обнаружить). Вот статья об этом: Рисование прямоугольника внутри массива 2D numpy

2. Затем я бы использовал OpenCV (cv2), чтобы добавить все кадры вместе в сделать видео, вы также можете использовать ffmpeg. Вот статья об этом (OpenCV): Как сделать mov ie из изображений в python

3. Тогда для вашей инфраструктуры пользовательского интерфейса вы можете используйте PyQt5 для отображения вашего видео: Загрузите видео в opencv кадр за кадром, используя PyQT . Но вы также можете использовать Kivy с Gstreamer: Kivy VideoPlayer fullscreen, l oop и скрыть элементы управления

4. Наконец, чтобы превратить ваш .py файл в .exe (исполняемый файл для windows) Я бы использовал PyInstaller для этого: http://www.pyinstaller.org/