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
张洁 苏倩 韩忠泰.
基于人工蜂群算法的W SN 覆盖优化研究[J]. 电脑与电信. 2020, 1(6): 52-56
HUANG Jing YANG Shu-guo LIU Zi-zheng.
AMethod for Image Retrieval with Capsule Network[J]. Computer & Telecommunication. 2020, 1(6): 52-56
{{custom_sec.title}}
{{custom_sec.title}}
{{custom_sec.content}}