The RaspGrade Dataset: Towards Automatic Raspberry Ripeness Grading with Deep Learning
Mohamed Lamine Mekhalfi, Paul Chippendale, Fabio Poiesi, Samuele Bonecher, Gilberto Osler, Nicola Zancanella
TL;DR
This work tackles automatic five-class raspberry ripeness grading on a moving packaging line using RGB vision. It introduces the RaspGrade dataset, comprising pixel-level masks and five ripeness labels, and employs YOLOv8 for real-time instance segmentation to grade raspberries and punnets. Baseline results show moderate overall accuracy (mAP around 62–66%), with severe class imbalance: Grade 2 dominates while Grade 5 remains difficult; targeted loss-weight adjustments yield improvements but reveal trade-offs among classes. The dataset is released publicly, enabling further research, and future work will expand data coverage, explore advanced architectures, and possibly multi-modal sensing to enhance robustness for industrial deployment. The practical impact lies in improved throughput and non-invasive quality control in fruit processing lines.
Abstract
This research investigates the application of computer vision for rapid, accurate, and non-invasive food quality assessment, focusing on the novel challenge of real-time raspberry grading into five distinct classes within an industrial environment as the fruits move along a conveyor belt. To address this, a dedicated dataset of raspberries, namely RaspGrade, was acquired and meticulously annotated. Instance segmentation experiments revealed that accurate fruit-level masks can be obtained; however, the classification of certain raspberry grades presents challenges due to color similarities and occlusion, while others are more readily distinguishable based on color. The acquired and annotated RaspGrade dataset is accessible on Hugging Face at: https://huggingface.co/datasets/FBK-TeV/RaspGrade.
