Yolov5-Real-Time-Object-Detection
Yolov5-Real-Time-Object-Detection copied to clipboard
Published
20 hours ago •
ngzhili
ngzhili
Real Time Video Feed with Object Detection using Yolov5 (custom or pre-trained model) deployed on web browser using Flask backend.
Yolov5 Object Detection on Live Stream Video via Flask
The Yolov5s pretained model is deployed using flask. This repo contains example apps for exposing the yolo5 object detection model from pytorch hub via a flask api/app.
Web app
Simple app that enables live webcam detection using pretrained YOLOv5s weights and see real time inference result of the model in the browser.
Run & Develop locally
Run locally and dev:
conda create -n <VENV>conda activate <VENV>(<VENV>) $ pip install -r requirements.txt(<VENV>) $ flask run
Docker
The example dockerfile shows how to expose the rest API:
# Build
docker build -t yolov5 .
# Run
docker run -p 5000:5000 yolov5-flask:latest
reference
- https://github.com/ultralytics/yolov5
- https://github.com/jzhang533/yolov5-flask
- https://github.com/avinassh/pytorch-flask-api-heroku
ngzhili