Free cookie consent management tool by TermsFeed

Louise Helary

Sheep detection model v1

25/03/2024

🐏 Sheep Detection Model v1

This repository contains code for a sheep detection model using YOLO v8.

💻 Requirements

  • GPU: The model requires a GPU to train efficiently. Please ensure that you have access to a GPU before proceeding.
  • PyTorch Library: Refer to the PyTorch website to install the appropriate version based on your operating system. You can find the installation instructions here: https://pytorch.org/
  • YOLO Library: You can install the YOLO library by running pip install ultralytics.
  • yaml file: Give the access directory to the dataset.

📄 Dataset

This model was built using two roboflow universe dataset. Both contained aerial photos of sheeps

Dataset 1: Aerial Sheep Dataset 2: Sheep Gen4

All images from both are organized in train/validation/test:

Train Validation Test
Dataset 1 3 609 350 174
Dataset 2 777 78 39
Total 4 386 428 213

Model parameters

  • pre-trained model: yolov8m.pt
  • batch: 8
  • optimizer: auto
  • lr: 0.01
  • momentum: 0.937

Results

  • recall = 0.76
  • precision = 0.93

https://github.com/ICAERUS-EU/UC3_Livestock_Monitoring/assets/99217487/0bbd56ab-905b-473e-91db-b5fb0aa479ef