基于人工蜂群算法的W SN 覆盖优化研究

AMethod for Image Retrieval with Capsule Network

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.