Repozytorium publikacji - Politechnika Gdańska

In the paper we investigate the problem of food classification from images, for the Food-101 dataset extended with 31 additional food classes from Polish cuisine. We adopted transfer learning and firstly measured training times for models such as MobileNet, MobileNetV2, ResNet50, ResNet50V2, ResNet101, ResNet101V2, InceptionV3, InceptionResNetV2, Xception, NasNetMobile and DenseNet, for systems with NVIDIA Tesla V100 (Volta) and NVIDIA GTX 1060 (Pascal) GPUs. We presented inference times corresponding to training the various considered network models, both using a desktop NVIDIA GTX 1060 GPU and an Intel i7-7000 CPU. Subsequently, we investigated the InceptionV3 model in more detail, best in the preliminary tests, regarding the impact of both learning rates (including both various fixed and variable rates) as well as batch sizes on the accuracy of classification, along with training times for various batch sizes. This allowed to identify better learning rate configurations as well as classification performance versus training time.

Autorzy

• Ewa Tusień link otwiera się w nowej karcie ,
• Aleksandra Wilke link otwiera się w nowej karcie ,
• Joanna Woźna link otwiera się w nowej karcie ,
• dr hab. inż. Paweł Czarnul link otwiera się w nowej karcie