Head Detector using YoloV8

Notice: It is OUTDATED.

This is a head detection model based on the Ultralytics YoloV8 model. Please clone the repository to use it or use the Hugging Face deployment.

Documentation

1. Install requirements

Install dependencies using the following command. The model is implemented using Python version 3.10.

pip install -r requirements.txt

2. Run inference from CLI and show the results using matplotlib

python ./main.py --image_path=image/test.jpg

3. Run inference using Streamlit

streamlit run ./streamlit_webapp.py

4. Run inference using Streamlit on Docker

Build the image from the Dockerfile:

docker build --tag abel_head_detector .

Or pull the image from docker hub:

docker pull abelkidane/reports

Run the image using docker run:

docker run -it abel_head_detector streamlit run streamlit_webapp.py --server.port 8080

Or using docker compose (not completed yet)

docker build --tag abel_head_detector .
docker compose up