Hello everyone, I wanted to share with you my first Streamlit app.
It’s a simple interface for yolov5 inference.
Live Demo
Features
- Caches the model for faster inference on both CPU and GPU
- Supports both images and videos.
- Supports both CPU and GPU inference.
- Supports Custom Classes and changing Confidence.
How to run
deploy using Streamlit Cloud or:
github repo
clone repo:
git clone https://github.com/moaaztaha/Yolo-Interface-using-Streamlit
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 the name in the code. - Add the model file to
models/
and changecfg_model_path
to its path.
cd Yolo-Interface-using-Streamlit
streamlit run app.py