Skip to content

Wolfman1219/streamlit-onnx

Repository files navigation

Yolov8 Real-time Inference using Streamlit

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.

demo of the dashboard

Features

  • Supports uploading model files (<200MB) and downloading models from URL (any size)
  • Supports videos.

How to run

After cloning the repo:

  1. Install requirements
    • pip install -r requirements.txt
  2. Add sample images to data/sample_images
  3. Add sample video to data/sample_videos and call it sample.mp4 or change name in the code.
  4. Add the model file to models/ and change cfg_model_path to its path.
git clone https://github.com/Wolfman1219/yolo_with_streamlit.git
cd yolo_with_streamlit
streamlit run app.py

References

https://discuss.streamlit.io/t/deploy-yolov5-object-detection-on-streamlit/27675

About

streamlit-onnx

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published