PyTorch Deep Learning for Food Image Classification with Food Dataset

Abstract

Classification of food images is crucial in today's increasingly digitally connected world. In the rapidly evolving mobile applications and social media era, the demand for an automated system that can recognize food types from an image is intensifying. This study employs deep learning and the PyTorch framework to develop a dependable and efficient solution for classifying food images. This research is motivated by the growing complexity of food introduction challenges. The primary challenge is improving the accuracy of food type recognition and overcoming variations in the visual presentation of food, such as lighting, shooting angles, and proportional and textural differences. Convolutional Neural Networks (CNN) are effective for image classification and are incorporated into the methods utilized. In addition, we employ ResNet101 transfer learning techniques to capitalize on the knowledge of trained models for large image datasets. The primary objective of this study is to develop a food image classification model that is accurate, training-efficient, and capable of accurately recognizing various types of food. In testing and evaluation, the developed model could realize multiple types of food with satisfactory accuracy. The accuracy of training reached 99.35%, while the accuracy of testing reached 94.65%. This study also reveals how Resnet101 transfer learning is utilized by deep learning technology.