Back to Search Start Over

Food Image Classification Based on CBAM-Inception V3 Transfer Learning

Authors :
DU Hui-jiang
CUI Xiao-yi
WANG Yi-meng
SUN Li-ping
Source :
Liang you shipin ke-ji, Vol 32, Iss 1, Pp 91-98 (2024)
Publication Year :
2024
Publisher :
Academy of National Food and Strategic Reserves Administration, 2024.

Abstract

To improve the accuracy of automatic recognition and classification of food images, a classification model CBAM- InceptionV3 is proposed, which embeds the Convolutional Block Attention Module. The specific method is to split the Inception V3 model with ImageNet pre-trained weight parameters into blocks, embed CBAM modules after each Inception block, and reassemble them into a new model, embedding a total of 11 CBAM modules. This new model is used for transfer learning of Food-101 food image dataset padded and scaled to 299 pixels in both length and width, with the highest accuracy of 82.01%. Compared with the original Inception V3 model, the CBAM module can effectively improve the model's feature extraction and classification capabilities. At the same time, transfer learning can significantly improve the accuracy rate and shorten the training time compared with the training from scratch. Compared with several other mainstream convolutional neural network models, the results show that this new model has higher recognition accuracy and can provide strong support for food image classification and recognition.

Details

Language :
English, Chinese
ISSN :
10077561
Volume :
32
Issue :
1
Database :
Directory of Open Access Journals
Journal :
Liang you shipin ke-ji
Publication Type :
Academic Journal
Accession number :
edsdoj.8610d15e5cb140358d6032700d19530c
Document Type :
article
Full Text :
https://doi.org/10.16210/j.cnki.1007-7561.2024.01.012