Capsule Networks Towards Object Detection Capsule Object Detector (COD)
Although Convolutional Neural Networks performed better in object detection, CNNs does not care about spatial relationships existing in an image. In this paper, we try describe "capsule network based object detection" model COD based on the VGG16 model (as a base network) which presents a substantial result in many sections of object detection over Convolution Neural Network based model by achieving the problem of spatial relationships. We used matrix capsules and dynamic EM routing to classify object from different viewpoints. The whole model is grounded on "dynamic routing between capsules" which is suggested by Geoffrey E Hinton. Both proposed theories use capsules that maps feature properties of an object as information for detecting that object which is extracted by capsules and Dynamic routing groups the capsules of lower level into parent level capsules by an iterative dynamic routing process. We train and test our model on Pascal VOC 2007 and dataset. We implement this in python using Keras (Tensorflow as backend) and train our model in Google cloud compute engine. COD achieves an accuracy of 67.3 mAP on Pascal VOC 2007 dataset and performing a comparable performance with Fast R CNN. Keywords Object Detection, CNN, Capsule Networks, VGG16
International Journal of Computer Sciences and Engineering