From Few Images to High Accuracy: Augmentation and Embedding Methods for Date Fruit Ripeness

نوع مقاله : مقاله پژوهشی

نویسندگان

1 عضو هیات علمی گروه مهندسی مکانیک بیوسیستم- دانشکده منابع طبیعی و فناوری کشاورزی-دانشگاه محقق اردبیلی - اردبیل – ایران

2 گروه مهندسی کانپیوتر-دانشگاه صنعتی شزیف

3 گروه مکانیک بیوسیستم دانشگاه گرگان

چکیده

Manual date harvesting and sorting remain labor-intensive and error-prone, particularly when distinguishing intermediate ripeness stages such as Rotab. We present an image-based classification pipeline for the Berhi cultivar that assigns fruit to three ripeness stages—Khalal, Rotab, and Tamar—using compact deep structures and training strategies suited to small datasets. Rather than relying on generative or adversarial methods, our approach emphasizes (i) careful augmentation (classical transforms, automated policies, and sample-mixing), (ii) transfer and self-supervised pre training, and (iii) embedding- and metric-learning alternatives, with ensembles and test-time augmentation used as optional accuracy/robustness boosters. On a 150-image dataset (50 images per class) evaluated with 5-fold cross-validation, a ResNet18 baseline reaches about 95% average accuracy. Automated augmentation combined with MixUp/CutMix improves accuracy to 97%, and self-supervised pre training plus advanced augmentation and ensembling attain peak performance near 98%. Improvements are most pronounced for the visually ambiguous Rotab class. We also report practical robustness measures (common corruptions, geometric stability, and calibration), which show that augmentation and pre training substantially increase stability under realistic input variability. These results indicate that, for small and visually subtle datasets, augmentation and pre training—rather than synthetic data generation—offer a pragmatic path to high accuracy and robust behavior.

کلیدواژه‌ها

موضوعات


عنوان مقاله [English]

From Few Images to High Accuracy: Augmentation and Embedding Methods for Date Fruit Ripeness

نویسندگان [English]

  • Raziyeh Pourdarbani 1
  • Omid Daliran 2
  • Sajad Sabzi 3
1 Dept. of Biosystem engineering, University of Mohaghegh Ardabili
2 2Department of Computer Engineering, Sharif University of Technology, Tehran 14588-89694, Iran.
3 3Department of Biosystems Engineering, Gorgan University of Agricultural Sciences and Natural Resources, Gorgan, Iran.
چکیده [English]

Manual date harvesting and sorting remain labor-intensive and error-prone, particularly when distinguishing intermediate ripeness stages such as Rotab. We present an image-based classification pipeline for the Berhi cultivar that assigns fruit to three ripeness stages—Khalal, Rotab, and Tamar—using compact deep structures and training strategies suited to small datasets. Rather than relying on generative or adversarial methods, our approach emphasizes (i) careful augmentation (classical transforms, automated policies, and sample-mixing), (ii) transfer and self-supervised pre training, and (iii) embedding- and metric-learning alternatives, with ensembles and test-time augmentation used as optional accuracy/robustness boosters. On a 150-image dataset (50 images per class) evaluated with 5-fold cross-validation, a ResNet18 baseline reaches about 95% average accuracy. Automated augmentation combined with MixUp/CutMix improves accuracy to 97%, and self-supervised pre training plus advanced augmentation and ensembling attain peak performance near 98%. Improvements are most pronounced for the visually ambiguous Rotab class. We also report practical robustness measures (common corruptions, geometric stability, and calibration), which show that augmentation and pre training substantially increase stability under realistic input variability. These results indicate that, for small and visually subtle datasets, augmentation and pre training—rather than synthetic data generation—offer a pragmatic path to high accuracy and robust behavior.

کلیدواژه‌ها [English]

  • Date fruit
  • Ripeness
  • Deep learning
  • Self-supervised learning
  • Metric learning
  • Robustness

مقالات آماده انتشار، پذیرفته شده
انتشار آنلاین از تاریخ 01 آذر 1404
  • تاریخ دریافت: 04 آبان 1404
  • تاریخ بازنگری: 22 آبان 1404
  • تاریخ پذیرش: 01 آذر 1404
  • تاریخ اولین انتشار: 01 آذر 1404
  • تاریخ انتشار: 01 آذر 1404