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
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.
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.
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.
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.
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.
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 |