A web interface for real-time yolo inference using streamlit. It supports CPU and GPU inference, supports both images and videos and uploading your own custom models.
- Supports uploading model files (<200MB) and downloading models from URL (any size)
- Supports videos.
After cloning the repo:
- Install requirements
pip install -r requirements.txt
- Add sample images to
data/sample_images
- Add sample video to
data/sample_videos
and call itsample.mp4
or change name in the code. - Add the model file to
models/
and changecfg_model_path
to its path.
git clone https://github.com/Wolfman1219/yolo_with_streamlit.git
cd yolo_with_streamlit
streamlit run app.py
https://discuss.streamlit.io/t/deploy-yolov5-object-detection-on-streamlit/27675