Abstract:
Supporting vital research by the University of Pretoria's Social Insects Research Group (SIRG), the project uses the YOLOv8 CNN object detection model to identify different species of bees. We employed several approaches such as shearing, brightness, rotation, breadth, and height changes to gather and enhance a dataset of approximately one thousand bee images. Metrics like the F1 score, precision, recall, and confusion matrices were used to assess the YOLOv8 model. To make our work accessible, we implemented an augmented visualisation with Streamlit and a computer vision model. Our results demonstrate that YOLOv8 is capable of precisely classifying various bee species and determining which enhancement methods work best. For additional study, the project's code and documentation are accessible on GitHub.