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, Turkey, 2 - 04 February 2017, pp.10-14 identifier

  • Publication Type: Conference Paper / Full Text
  • Doi Number: 10.1145/3045714.3045716
  • City: Kayseri
  • Country: Turkey
  • Page Numbers: pp.10-14
  • Keywords: Convolutional neural network, Deep neural network models, Fine-tuning, Object dataset, Object recognition, Robotics
  • Yıldız Technical University Affiliated: Yes

Abstract

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.