AMethod for Image Retrieval with Capsule Network

HUANG Jing YANG Shu-guo LIU Zi-zheng

Computer & Telecommunication ›› 2020, Vol. 1 ›› Issue (6) : 52-56.

Computer & Telecommunication ›› 2020, Vol. 1 ›› Issue (6) : 52-56.

AMethod for Image Retrieval with Capsule Network

Author information +
History +

Abstract

As an emerging network structure, the capsule network uses vector output instead of scalar output, which can capture the spatial relationship between image features and improve the limitations of convolutional neural network. This paper firstly trains the capsule network to achieve image classification, obtains the predictive label of the image, determines the category of the query im- age, and then uses the feature parameters in the digital capsule layer of the network as the feature vector of the image. The feature vector is used to find images similar to the query image in the category set of the query image. In this paper, experiments are carried out on the FASHION-MNIST and CIFAR10 datasets respectively. The experimental results show that the proposed method can bet- ter extract the features of the images and obtain good image retrieval results.

Key words

image retrieval / capsule network / feature extraction

Cite this article

Download Citations
HUANG Jing YANG Shu-guo LIU Zi-zheng. AMethod for Image Retrieval with Capsule Network[J]. Computer & Telecommunication. 2020, 1(6): 52-56

Accesses

Citation

Detail

Sections
Recommended

/