Food Classification from Images Using a Neural Network Based Approach with NVIDIA Volta and Pascal GPUs
Abstrakt
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.
Cytowania
-
0
CrossRef
-
0
Web of Science
-
0
Scopus
Autorzy (4)
Cytuj jako
Pełna treść
- Wersja publikacji
- Accepted albo Published Version
- Licencja
- Copyright (2022 The Author(s), under exclusive license to Springer Nature Switzerland AG)
Słowa kluczowe
Informacje szczegółowe
- Kategoria:
- Aktywność konferencyjna
- Typ:
- publikacja w wydawnictwie zbiorowym recenzowanym (także w materiałach konferencyjnych)
- Język:
- angielski
- Rok wydania:
- 2022
- Opis bibliograficzny:
- Tusień E., Wilke A., Woźna J., Czarnul P.: Food Classification from Images Using a Neural Network Based Approach with NVIDIA Volta and Pascal GPUs// / : , 2022,
- DOI:
- Cyfrowy identyfikator dokumentu elektronicznego (otwiera się w nowej karcie) 10.1007/978-3-031-10539-5_20
- Źródła finansowania:
-
- Publikacja bezkosztowa
- Weryfikacja:
- Politechnika Gdańska
wyświetlono 106 razy