Back to projects

Public project · Computer vision

Hymenoptera Recognizer

Recognizing bees, wasps, and hornets in images or video.

Model predictions on a batch of hymenoptera images
Project previewGitHub

Context

The project

Hymenoptera Recognizer trains and uses a YOLOv5 model to distinguish five categories: bee, wasp, European hornet, Asian hornet, and Oriental hornet. The repository covers images, videos, and webcam streams.

Objectives

Design challenges

  1. 01

    Build a dedicated dataset for five visually similar classes.

  2. 02

    Train and version models with a reproducible workflow.

  3. 03

    Make inference usable on images, video, or webcam.

Architecture

Technical flow

Collection and labeling → YOLO dataset → YOLOv5 training → validation → versioned weights → image or video inference

Scroll through the flow

Collection and labeling
YOLO dataset
YOLOv5 training
validation
versioned weights
image or video inference
  • Python
  • YOLOv5
  • PyTorch
  • Roboflow
  • Computer Vision
  • GPU

Approach

Working steps

  1. 01

    Assemble and label the dataset with dedicated tools.

  2. 02

    Train two model versions and retain their weights.

  3. 03

    Compare labels and predictions on the published validation batches.

Public repository

What the sources document

  • Two model versions, their weights, and their confusion matrices are published in the repository.
  • Validation examples allow annotations and predictions to be compared directly.
github.com/SylvJalbExplore the repository

Let’s talk CONTACT / 01

Let’s talk about your next product.

Share a little context. I will reply to you directly by email.

contact@sylvain-jalbert.com

This information is used only to reply to your message. It is not stored on this website.