We found a match
Your institution may have access to this item. Find your institution then sign in to continue.
- Title
改进胶囊网络的小样本图像分类算法.
- Authors
王飞龙; 刘萍; 张玲; 李钢
- Abstract
In order to address the problem that the capsule network can not classify complex small sample images effectively, a classification model is proposed on the basis of fusing the improved Darknet with the capsule network. Firstly, the Darknet is upgraded containing both the shallow level extractor and the deep level extractor. The shallow level extractor adopts a 5×5 convolution kernel to capture long-distance edge contour features and the deep level extractor uses a 3×3 convolution kernel to capture deeper semantic features. Then, the extracted edge features and semantic features are fused to preserve effective features of images. Next, the capsule network is used to vectorize these effective features to work out the loss of spatial representation. Finally, L2 regularization is added in the loss function to avoid the over-fitting. Experimental results show that, on the small sample dataset, the classification accuracy of the proposed model is 28.51 percentage points and 24.40 percentage points higher than that of the models of the capsule network and the DCaps respectively, 21.57 percentage points and 18.02 percentage points higher than that of the ResNet50 and the Xception respectively. Hence it suggests that the method proposed in this paper gains a better performance in classifying complex small sample images. Meanwhile, on the large sample dataset, the classification accuracy of the proposed model has also been improved to a certain extent.
- Subjects
CAPSULE neural networks; PERCENTILES; CLASSIFICATION
- Publication
Journal of Frontiers of Computer Science & Technology, 2022, Vol 16, Issue 10, p2387
- ISSN
1673-9418
- Publication type
Article
- DOI
10.3778/j.issn.1673-9418.2102026