Tehničko veleučilište u Zagrebu · Zagreb

Automobile Classification Using Transfer Learning on ResNet Neural Network Architecture

stručni rad

stručni rad

Automobile Classification Using Transfer Learning on ResNet Neural Network Architecture

Vrsta prilog u časopisu
Tip stručni rad
Godina 2020
Časopis Polytechnic and design
Volumen 8
Svesčić 1
Stranice str. 59-64
DOI 10.19279/TVZ.PD.2020-8-1-18
ISSN 1849-1995
EISSN 2459-6302
Status objavljeno

Sažetak

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.

Ključne riječi

transfer learning, ResNet, Stanford Car dataset