期刊
IEEE ACCESS
卷 9, 期 -, 页码 108162-108174出版社
IEEE-INST ELECTRICAL ELECTRONICS ENGINEERS INC
DOI: 10.1109/ACCESS.2021.3101247
关键词
Feature extraction; Generative adversarial networks; Grippers; Generators; Object segmentation; Image segmentation; Vegetation; Detection of wood defects; data augmentation; GAN; layered deformable Mask R-CNN
资金
- Heilongjiang Province Research and Development Plan of Applied Technology [GA19A402]
- Central University Basic Scientific Research Business Expenses Special Funds [2572020DY12]
In this study, a novel detection method using generative adversarial networks and data augmentation is proposed to address the issues of imbalance in defect categories and difficult access to defective data in wood defect detection. Additionally, a layered deformable mask region-based convolutional neural network is constructed to tackle the challenges of modeling irregular defects and extracting contextual information.
The detection of wood defects plays an important role in the processing and production of wood, the deep learning methods have achieved outstanding results in this field in recent years. There are still two unresolved problems, one is that the category of defects is unbalanced and difficult to obtain, and the other is the effective modeling of defects with different sizes and irregular shapes. Here an alternative detection method is proposed. For the problem of imbalance in defect categories, a generative data augmentation method, cycle generative adversarial network (GAN), is used to generate new defective images, while seven online data augmentation schemes are used to solve the problem of difficult access to defective data. The layered deformable mask region-based convolutional neural network (Mask R-CNN) is constructed to address the challenges of modeling irregular defects and contextual information extraction. By creating layered connections between residual modules, a larger receptive field is achieved at each layer of the network, while deformable convolution is used to better fit the shape of the defects. Finally, the upgraded model is trained and used to locate and segment defects simultaneously. Ablation experiments show that the enhanced model performs better than the original model in terms of localization and segmentation. Meanwhile, it is excellent relative to baseline object detection algorithms, instance segmentation algorithms, and other methods for detecting wood defects.
作者
我是这篇论文的作者
点击您的名字以认领此论文并将其添加到您的个人资料中。
推荐
暂无数据