Real-world image datasets, especially in domains like food classification, often suffer from class imbalance and limited data diversity, which can significantly impact model performance. To address these challenges, we enhance the YOLOv8m-cls model by integrating advanced data augmentation techniques, specifically CutMix and MixUp, to improve robustness and generalization. Our experimental results demonstrate that the augmented dataset significantly enhances model performance. Compared to training on the original dataset, the model trained with augmentation achieves a lower validation loss (0.57687 vs. 0.93444), higher Top-1 accuracy (87.17% vs. 85.51%), and higher Top-5 accuracy (96.67% vs. 95.84%). Although the training time increases due to the larger effective dataset size, the improvements in generalization and accuracy justify the additional computational cost. This work underscores the importance of combining state-of-the-art models with tailored augmentation strategies to address the inherent challenges of real-world image classification tasks.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

CutMix and MixUp: Effective Data Augmentation Strategies for Enhancing YOLOv8m-Cls in Food Image Classification

  • Nam Mai Hoai Tran,
  • Giang Truong Le,
  • Huy Anh Huynh,
  • Tuyen Bich Thi Vu,
  • Tung Quang Trinh,
  • Vinh Dinh Nguyen

摘要

Real-world image datasets, especially in domains like food classification, often suffer from class imbalance and limited data diversity, which can significantly impact model performance. To address these challenges, we enhance the YOLOv8m-cls model by integrating advanced data augmentation techniques, specifically CutMix and MixUp, to improve robustness and generalization. Our experimental results demonstrate that the augmented dataset significantly enhances model performance. Compared to training on the original dataset, the model trained with augmentation achieves a lower validation loss (0.57687 vs. 0.93444), higher Top-1 accuracy (87.17% vs. 85.51%), and higher Top-5 accuracy (96.67% vs. 95.84%). Although the training time increases due to the larger effective dataset size, the improvements in generalization and accuracy justify the additional computational cost. This work underscores the importance of combining state-of-the-art models with tailored augmentation strategies to address the inherent challenges of real-world image classification tasks.