Laser & Optoelectronics Progress, Volume. 62, Issue 2, 0228002(2025)

Fine Classification of Tree Species Based on Improved U-Net Network

Yulin Cai1、*, Hongzhen Gao1, Xiaole Fan1, Huiyu Xu1, Zhengjun Liu2, and Geng Zhang2
Author Affiliations
  • 1College of Geodesy and Geomatics, Shandong University of Science and Technology, Qingdao 266590, Shandong , China
  • 2Chinese Academy of Surveying and Mapping, Beijing 100036, China
  • show less

    In this study, a new method is proposed by improving an existing deep-learning network, where aerial high-resolution hyperspectral data and LiDAR data are combined for the fine classification of tree species. First, feature extraction and fusion are performed for different data sources. Subsequently, a classification network named CA-U-Net is constructed based on the U-Net network by adding a channel-attention-mechanism module to adjust the weights of different features adaptively. Finally, we attempt to address the problem of low identification precision for small-sample species by modifying CA-U-Net in class-imbalance cases. The research results show that 1) the CA-U-Net network performs well, with an overall classification accuracy of 96.80%. Compared with the FCN, SegNet, and U-Net networks, the CA-U-Net network shows improvements of 8.56, 11.99, and 3.31 percent points, respectively, in terms of classification accuracy. Additionally, the network exhibits a higher convergence speed. 2) Replacing the original loss function in the CA-U-Net network with a cross-entropy loss function based on the class-sample-size balance can improve the classification accuracy for tree species with fewer samples. The proposed methodology can serve as an important reference in small-scale forestry, such as orchard management, urban-forest surveys, and forest-diversity surveys.

    Keywords
    Tools

    Get Citation

    Copy Citation Text

    Yulin Cai, Hongzhen Gao, Xiaole Fan, Huiyu Xu, Zhengjun Liu, Geng Zhang. Fine Classification of Tree Species Based on Improved U-Net Network[J]. Laser & Optoelectronics Progress, 2025, 62(2): 0228002

    Download Citation

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

    Category: Remote Sensing and Sensors

    Received: Apr. 26, 2024

    Accepted: May. 24, 2024

    Published Online: Jan. 6, 2025

    The Author Email:

    DOI:10.3788/LOP241175

    CSTR:32186.14.LOP241175

    Topics