Laser & Optoelectronics Progress, Volume. 57, Issue 16, 161022(2020)

Ship Classification Method for Point Cloud Images Based on Three-Dimensional Convolutional Neural Network

Yongmei Ren1,2, Jie Yang1、*, Zhiqiang Guo1, and Yilei Chen3
Author Affiliations
  • 1Hubei Key Laboratory of Broadband Wireless Communication and Sensor Networks, School of Information Engineering, Wuhan University of Technology, Wuhan, Hubei 430070, China
  • 2School of Electrical and Information Engineering, Hunan Institute of Technology, Hengyang, Hunan 421002, China
  • 3School of Artificial Intelligence, Xidian University, Xi'an, Shaanxi 710071, China
  • show less

    In order to further improve the classification accuracy of ship classification method for point cloud images, a new ship classification method based on three-dimensional convolutional neural network (3D CNN) is proposed. First, the point cloud image is transformed into a voxel grid image by the density grid method and the voxel grid image is taken as the input object of a 3D CNN. Then, the high-level features of the voxel grid image are extracted by the designed 6-layer 3D CNN to capture its structural information. Finally, the classification results are obtained using the Softmax function in the output layer. The experimental results show that the classification accuracy of the proposed method can reach 96.14% on the self-build point cloud image ship dataset, 5.97% higher than that of the 3D ShapeNets method and 2.46% higher than that of the VoxNet method. Compared with some existing methods, the proposed method has higher classification accuracy on Sydney urban object dataset. These results show that the proposed method has a good classification performance.

    Tools

    Get Citation

    Copy Citation Text

    Yongmei Ren, Jie Yang, Zhiqiang Guo, Yilei Chen. Ship Classification Method for Point Cloud Images Based on Three-Dimensional Convolutional Neural Network[J]. Laser & Optoelectronics Progress, 2020, 57(16): 161022

    Download Citation

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

    Category: Image Processing

    Received: Dec. 10, 2019

    Accepted: Jan. 16, 2020

    Published Online: Aug. 5, 2020

    The Author Email: Yang Jie (jieyang@whut.edu.cn)

    DOI:10.3788/LOP57.161022

    Topics