Asian Food Image Classification Based on Deep Learning

• 作者: Bing Xu Xiaopei He Zhijian Qu
• 来源期刊: 电脑和通信(英文)
• 学科: 工学
• 关键词: Asian Food Image Classification Convolutional Neural Network Attention Mechanism Data Enhancement
• 年，卷（期）: dnhtxyw_2021,（3）
• 页码范围: 10-28
• 页数: 19页
• 分类号: TP3

摘要

To improve Asian food image classification accuracy, a method that combined Convolutional Block Attention Module (CBAM) with the Mobile NetV2, VGG16, and ResNet50 was proposed for Asian food image classification. Additionally, we proposed to use a mixed data enhancement algorithm (Mixup) to have a smoother discrimination ability. The effects of introducing the attention mechanism (CBAM) and using the mixed data enhancement algorithm (Mixup) were shown respectively through experimental comparison. The combination of these two and the final test set Top-1 accuracy rate reached 87.33%. Moreover, the information emphasized by CBAM was reflected through the visualization of the heat map. The results confirmed the classification method’s effectiveness and provided new ideas that improved Asian food image classification accuracy.