Laser & Optoelectronics Progress, Volume. 57, Issue 24, 241002(2020)

Texture Images Classification Algorithm Combining Wavelet Transform and Capsule Network

Zhiyong Tao1, jie Li1,2、*, and Xiaoliang Tang2
Author Affiliations
  • 1School of Electronic Information Engineering, Liaoning Technical University, Huludao, Liaoning 125105, China
  • 2Quanzhou Institute of Equipment Manufacturing Haixi Institutes, Chinese Academy of Sciences, Quanzhou, Fujian 362000, China
  • show less

    Capsule network is a new type of deep learning network, capsule structure can encode information such as posture, texture, hue, etc. of the feature, and has a good ability to express the texture feature of the image. Aiming at the problem that the primary feature extraction network of the capsule network is too simple and the feature expression ability is insufficient, a discrete wavelet capsule network (DWTCapsNet) that combines the feature expression capabilities of deep convolutional neural networks with wavelet transform multi-resolution analysis capabilities is propose in this work. First, the feasibility of the capsule network in the application of texture image classification is studied. Second, the ability of each part of DWTCapsNet on the improvement of capsule network classification performance is studied. Finally, the robustness of DWTCapsNet is analyzed through anti-rotation and anti-noise experiments. The classification accuracy is used as the standard model evaluation criteria, and the experimental results on the commonly used texture image data sets show that the classification accuracy of DWTCapsNet is higher.

    Tools

    Get Citation

    Copy Citation Text

    Zhiyong Tao, jie Li, Xiaoliang Tang. Texture Images Classification Algorithm Combining Wavelet Transform and Capsule Network[J]. Laser & Optoelectronics Progress, 2020, 57(24): 241002

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Image Processing

    Received: Apr. 21, 2020

    Accepted: May. 22, 2020

    Published Online: Dec. 2, 2020

    The Author Email: Li jie (leej95@163.com)

    DOI:10.3788/LOP57.241002

    Topics