Tailoring the AI for robotics: Fine-tuning predefined deep convolutional neural network model for a narrower class of objects

BAYRAKTAR E., Yigit C. B., Boyraz P.

2017 International Conference on Mechatronics Systems and Control Engineering, ICMSCE 2017, Kayseri, Türkiye, 2 - 04 Şubat 2017, ss.10-14 identifier

  • Yayın Türü: Bildiri / Tam Metin Bildiri
  • Doi Numarası: 10.1145/3045714.3045716
  • Basıldığı Şehir: Kayseri
  • Basıldığı Ülke: Türkiye
  • Sayfa Sayıları: ss.10-14
  • Anahtar Kelimeler: Convolutional neural network, Deep neural network models, Fine-tuning, Object dataset, Object recognition, Robotics
  • Yıldız Teknik Üniversitesi Adresli: Hayır

Özet

Object recognition has been one of the greatest challenges for robotic tasks particularly in indoor environments. In order to perform this task, so called hand-crafted features requiring high computational effort which rely on expert knowledge are used until achieving better results by learned features using convolutional neural networks. In this study, we have given a powerful object classification model with 2.72% top-1 error rate which is achieved by fine-tuning a predefined model for 10 classes. We built our model on top of VGG16 architecture (trained on a larger dataset which consists of 1000 classes) and froze the layers except the last classification layer which we trained for 10 classes. The training data consists of 10000 images (1000 per class) and there are 4000 images for validation (400 per class). The object classes in our dataset are book, bottle, bowl, cup, eyeglass, keyboard, laptop, monitor, teapot and vase which may exist on the desktop in an indoor environment.