Как захватить видео с веб-камеры и передать его в мою обученную модель Yolov5 (в PyTorch) внутри веб-приложения (внутри браузера)?

#python #machine-learning #yolo

#python #машинное обучение #yolo

Вопрос:

Я работаю над системой обнаружения выбоин, и я обучил свою модель с использованием Yolov5 (от Ultralytics, полностью реализованной в PyTorch). После обучения модели в Google Colab у меня есть файл окончательного веса в формате .pt. Теперь я хочу создать какое-то веб-приложение, которое будет принимать входные данные с веб-камеры и передавать их в мою модель в режиме реального времени. Я много искал, но не нашел удовлетворительного решения.

Комментарии:

1. Вы нашли решение?

Ответ №1:

Для этого нет открытого исходного кода веб-приложения. Но вы можете изменить ultralitycs detect.py чтобы создать детектор, занесите информацию о вашем обнаружении в базу данных и используйте ее для отображения на веб-странице.

https://github.com/ultralytics/yolov5/blob/master/detect.py

Ответ №2:

YOLOv5 предоставляет простую команду для этого. Команда:-

 python detect.py --weights 'your-path-to-weights-file' --conf any-confidence-percentage-you-like --source 0
  

Приведенная выше команда используется, если вы запускаете ее на своем локальном компьютере.