khoi03 / Counting-Sheep

Counting Sheep without Sleep. This project utilizing YOLO and several image processing techniques to accurately track, count and segment sheep.

Counting-Sheep

Counting Sheep without Sleep. This project utilizing YOLO and several image processing techniques to track, count and segment sheep.

How to run this repository

I recommend creating an anaconda environment:

conda create --name [environment-name] python=3.9

Then, install Python requirements:

pip install -r requirements.txt

Finally, to reproduce the results, from the [environment-name] project root, run:

python counting_sheeps.py

Important: The code within the drawing_bounds.py file handles the crucial task of delineating and extracting the detection area. It will need adjustments to accommodate various videos and camera angles, or possibly even be disabled. Additionally, employing a stable camera in the video setup, ideally capturing all sheep from a high angle (even though I lack this type of video), will significantly improve accuracy.

Demo

Here are some demos showcasing the results obtained from videos sourced from YouTube.

output_sheeps2.mp4

output_Off.ewe.go.sheep.sorting.mp4