Fungi Classification using YOLOv8 with Auto-Labeling

Logo

YOLOv8 for classifying fungi images with the help of auto-labeling techniques using Label Studio.

View the Project on GitHub alxmares/Fungi_Classification_YOLOv8__Auto-Labeling

๐Ÿ› ๏ธ Tools and Technologies Used

Python YOLOv8 Label Studio Conda OpenCV

๐Ÿ“ Description of YOLOv8s

YOLOv8s is the latest version of the โ€œYou Only Look Onceโ€ (YOLO) family of real-time object detection models. It offers improved accuracy and performance over its predecessors, making it ideal for detecting and classifying various objects in complex environments.

YOLOv8

๐Ÿ“ˆ Auto-Labeling with Label Studio

This project utilized Label Studio in combination with YOLOv8s for automated image labeling. Over 6000 images were auto-labeled through active learning, streamlining the data preparation process and significantly reducing manual effort.

Label Studio Workflow

๐Ÿ–ผ๏ธ Image Preprocessing

To ensure consistency and optimal model performance, all images were preprocessed to a uniform size of 640x640 pixels. This standardization was crucial for effective training and inference.

๐Ÿ‹๏ธ Training with YOLOv8s

The YOLOv8s model was trained using the preprocessed and auto-labeled images. The training process involved fine-tuning various hyperparameters to achieve the best possible performance.

๐Ÿ“Š Results

The trained YOLOv8s model demonstrated exceptional performance in classifying different types of fungi. Key metrics and visual examples of the modelโ€™s predictions are highlighted below, showcasing its accuracy and robustness.

Model Performance

Model Results

Class Images Instances Box (P) Box (R) mAP50 mAP50-95
all 1370 1795 0.961 0.943 0.980 0.806
Apendiculatus 252 361 0.991 0.907 0.986 0.800
Cronartium 307 370 0.977 0.986 0.991 0.803
Melanocepphala 231 333 0.901 0.929 0.969 0.721
Phragmidium 220 276 0.989 0.951 0.975 0.790
Pucciniastrum 167 229 0.951 0.929 0.982 0.837
Hemileia 192 226 0.957 0.956 0.978 0.886

Model Results

Example Detections

Example Detection 1


๐Ÿš€ Explore More Projects!

๐Ÿ“š Check out all my projects on GitHub Pages