Polytechnic and design, Vol. 8 No. 1, 2020.
Professional paper
https://doi.org/10.19279/TVZ.PD.2020-8-1-18
KLASIFIKACIJA AUTOMOBILA KORISTEĆI TRANSFERIRANO UČENJE NA RESNET ARHITEKTURI NEURONSKE MREŽE
Stjepan Ložnjak
; Zagreb University of Applied Sciences, Zagreb, Croatia, Student
Tin Kramberger
; Zagreb University of Applied Sciences, Zagreb, Croatia
Ivan Cesar
; Zagreb University of Applied Sciences, Zagreb, Croatia
Renata Kramberger
; Zagreb University of Applied Sciences, Zagreb, Croatia
Abstract
Classification is one of the most common problems that neural networks are used for. In the case of higher resolution image classification, convolutional neural networks are commonly used. Due to the reason that convolutional neural networks are so often used in classification, there are many pretrained models that can be adapted for new domains using a technique called transfer learning. This paper shows how excellent results in classification accuracy can be achieved by applying transfer learning to pretrained convolution neural network. This paper presents the results of the learning transfer of the ResNet-152 convolution neural network on the Stanford Cars dataset. The results show accuracy over 88% only by training the last fully connected layer.
Keywords
Transfer learning; ResNet; Stanford Car dataset
Hrčak ID:
242768
URI
Publication date:
17.6.2020.
Visits: 1.065 *