Unmanned aerial vehicle remote sensing dead standing tree detection method based on double-flow feature integrated network
By adopting a dual-stream feature integrated network in the remote sensing of dry wood detection in drone, combining multiple branch modules and separation and enhancing attention modules, the occlusion, sparse and small sample problems of dry wood detection in dense target scenarios are solved, and higher detection accuracy and more comprehensive detection results are achieved.
Patent Information
- Application Number
- CN202411685620.3
- Authority / Receiving Office
- CN · China
- Patent Type
- Applications(China)
- Current Assignee / Owner
- Filing Date
- 2024-11-23
- Publication Date
- 2025-02-28
- Estimated Expiration
- 2044-11-23
Smart Images

Figure CN119540801A_ABST
Abstract
Description
Technical Field
[0001] The invention belongs to the technical field of unmanned aerial vehicle image processing, and in particular relates to a method for detecting standing dead trees by remote sensing using an unmanned aerial vehicle based on a dual-stream feature integration network. Background Art
[0002] Forests are complex ecosystems, and the health and life cycle of trees play a critical role in their composition and function. Tree mortality leads to a reduction in the total leaf area of the community, which affects soil processes such as solar radiation and nutrient cycling, fungal activity, soil erosion, and evaporation rates. At the same time, large-scale tree mortality events can have a negative impact on forest carbon storage, biodiversity, and the livelihoods of people who rely on forest resources. In the context of global climate change, extreme events such as droughts and wildfires can cause large-scale tree deaths, further amplifying these impacts. However, trees do not usually fall down immediately after death, but become standing dead trees (SDTs). Therefore, accurate detection and monitoring of standing dead trees is crucial to understanding these impacts, optimizing forest management, and reducing forest fires and their associated carbon emissions.
[0003] There are many reasons for tree death, such as stress, pests and diseases, and interspecific competition. Pests and diseases have the characteristic of rapid outbreaks, so many scholars also identify trees infected by pests and diseases in order to control forest health in advance. For example, Mo Dengkui et al. disclosed a large-scale intelligent detection method for dead standing trees caused by pine wilt disease based on unmanned aerial vehicles and convolutional neural networks, which can realize the rapid detection, counting and evaluation of dead standing trees in a large area; Chen Heng et al. proposed an integrated sky-ground remote sensing monitoring method for pine wilt disease epidemics, which realized comprehensive monitoring of pine wilt disease epidemic areas and improved the accuracy and efficiency of monitoring dead trees caused by pine wilt disease; Chen Xiaohua et al. disclosed a sky-ground integrated monitoring method for pine wilt disease monitoring, which has high efficiency in screening epidemic areas and can realize large-scale efficient monitoring; Zhang Fei et al. disclosed a method, device, medium and equipment for patrolling and identifying dead trees and diseased trees, which can accurately identify dead trees and diseased trees; Jiang Miaomiao et al. proposed an improved Yolov5 dead wood detection method combined with a defogging algorithm, which organically combined the target detection model with the defogging algorithm network, so that it can still maintain a relatively high target detection rate when facing foggy images. Accuracy; Xu Guoqing et al. provided a dead tree identification method and equipment based on visible light images. By reducing the bit rate and image conversion of visible light images, and then clustering and denoising the images, the dead trees in the jungle can be identified without relying on infrared images; Chen Farong et al. provided a pine wood nematode disease standing tree detection method and device based on UAV remote sensing images. The detection accuracy of pine wood nematode disease standing trees was improved by fusing multiple model results; Lan Yubin et al. disclosed a pine wood nematode disease dead tree detection and positioning method based on deep learning. The method can quickly, efficiently and accurately detect diseased pine trees and determine the location of diseased pine trees; Xu Guoqing disclosed a visible light remote sensing image dead tree identification software system and identification method. Through the remote sensing image segmentation module, advanced deep learning methods are combined with tree remote sensing image object recognition to realize the identification of dead trees in remote sensing images.
[0004] Currently, the dual-branch structure has been widely used in processing images, videos and other data. For example, Xu Xiaolong et al. disclosed a classroom student posture recognition method based on dual-trunk feature fusion, which effectively mined and fused multi-scale visual features, and realized accurate and efficient recognition of multi-target student postures in complex scenes; Zhou Hangxia et al. disclosed a solar radiation prediction method and system based on dual-branch feature extraction, which combined multi-scale convolution and bidirectional gated recurrent networks to extract meteorological features and time series features respectively, and combined with the attention mechanism to optimize the weighted fusion of each branch, and the prediction accuracy was significantly improved; Hua Chunjian et al. disclosed an RGB-D saliency detection method based on progressive weighted decoding, which extracted multi-level RGB image features and depth image features respectively through a symmetrical dual-stream feature extraction backbone network, and adopted a cross-modal feature fusion module to enhance the cross-modal information interaction between different branches. Experimental results show that the invention has high target detection accuracy for various scenarios.
[0005] The above inventions detect trees infected by pests and diseases and dead trees respectively, and explore some applications of dual-branch networks, but lack inventions related to the detection of dead trees in dense scenes. Therefore, the present invention proposes a dual-stream feature integration network to deal with problems such as mutual occlusion between tree crowns in dense scenes, background information introduced by sparse tree crowns, incomplete dead trees in the edge area of the image, small targets and few samples. Summary of the invention
[0006] The purpose of the present invention is to solve the problems of close connection between the crowns of dead trees in dense target scenes, serious occlusion problems, sparse crowns of dead trees after the dead branches fall and easy to be affected by the background, incomplete crowns of dead trees in the edge area of the image, less feature information of small targets and few samples, and provide a UAV remote sensing dead tree detection method based on a dual-stream feature integration network.
[0007] To achieve the above purpose, the technical solution adopted by the present invention is as follows:
[0008] A method for detecting standing dead trees by remote sensing using an unmanned aerial vehicle (UAV) based on a dual-stream feature integration network, the method comprising:
[0009] Step 1: Obtain images of dead trees using drones, and use annotation software combined with field surveys to annotate the dead trees;
[0010] Step 2: Build a dead tree detection dataset, which includes the collected dataset and the public dataset. The collected dataset is divided into training set, validation set and test set. During the training process, online data augmentation (including rotation, scaling, flipping and cropping) is used to augment the training set to enhance the generalization performance of the model. The public dataset is used as an additional test set to verify the generalization performance of the model.
[0011] Step 3: Use the Diverse Branch Block (DBB) to replace the traditional convolution module to fully explore the features of dead trees; use the Separated and Enhanced Attention Module (SEAM) to process the shallow features extracted from the trunk to enhance the attention to the details of the dead trees and obtain the enhanced features F SEAM ; Combining CBLinear and CBFuse to build a dual-branch structure backbone network, the left backbone of the network is responsible for extracting basic features, while the right backbone not only extracts features, but also integrates and fuses the features of the left backbone, and finally obtains feature F;
[0012] Step 4: The feature F obtained in step 3 is sent to the model detection head for training to reduce the difference between the predicted results and the true labels, and the detection performance of the model is evaluated on the validation set to ensure that the model not only learns the characteristics of the data, but also can generalize to unseen data to obtain the optimal model.
[0013] Step 5: Apply the trained model to the new image to perform real-time dead tree detection, and finally obtain an output image containing candidate boxes and confidence levels.
[0014] The beneficial effects of the present invention relative to the prior art are as follows: the present invention designs a novel dual-branch feature extraction fusion backbone network for detecting dead trees through RGB images, and by fusing features at different levels, significantly improves the recognition accuracy of small targets and targets in the edge area of the image. In addition, in order to solve the problem of densely connected crown occlusion, the present invention introduces a Diverse Branch Block (DBB) module to replace the traditional convolution module. The DBB module contains multiple branches of different scales and complexities, which enhances the feature expression ability of the convolution block and effectively reduces the missed detection rate of the dense crown occlusion area. At the same time, the effective combination of the dual-branch feature fusion backbone and the DBB module fully explores the characteristics of the dead trees, effectively enhances the propagation of the features in the network, and overcomes the problem of small samples. In addition, considering that the crown is sparse after the dead branches and leaves fall, the introduction of a large amount of background information reduces the detection accuracy, and the Separated and Enhancement Attention Module (SEAM) is introduced. The SEAM module effectively improves the model's recognition ability of the dead branch features in the canopy by enhancing the attention to the detailed features of the dead trees, such as color, edges and texture, and reduces the interference of background information. BRIEF DESCRIPTION OF THE DRAWINGS
[0015] Figure 1 This is the dual-backbone feature extraction and fusion network structure diagram. DETAILED DESCRIPTION
[0016] In order to better understand the content of the present invention, the present invention will be further described below in conjunction with specific embodiments and drawings. The following embodiments are implemented based on the technology of the present invention, and detailed implementation methods and operating steps are given, but the protection scope of the present invention is not limited to the following embodiments.
[0017] Embodiment 1:
[0018] A method for detecting standing dead trees by remote sensing using an unmanned aerial vehicle (UAV) based on a dual-stream feature integration network, the method comprising:
[0019] Step 1: Obtain images of dead trees using drones, and use annotation software combined with field surveys to annotate the dead trees;
[0020] Step 2: Build a dead tree detection dataset, which includes the collected dataset and the public dataset. The collected dataset is divided into training set, validation set and test set. During the training process, online data augmentation (including rotation, scaling, flipping and cropping) is used to augment the training set to enhance the generalization performance of the model. The public dataset is used as an additional test set to verify the generalization performance of the model.
[0021] Step 3: Use the Diverse Branch Block (DBB) to replace the traditional convolution module to fully explore the features of dead trees; use the Separated and Enhanced Attention Module (SEAM) to process the shallow features extracted from the trunk, enhance the attention to the details of the dead trees, and obtain the enhanced features F SEAM ; Combining CBLinear and CBFuse to build a dual-branch structure backbone network, the left backbone of the network is responsible for extracting basic features, while the right backbone not only extracts features, but also integrates and fuses the features of the left backbone, and finally obtains feature F;
[0022] Step 4: Send the feature F obtained in step 3 to the model detection head, train the model to reduce the difference between the predicted result and the true label, and evaluate the detection performance of the model on the validation set to ensure that the model not only learns the characteristics of the data, but also can generalize to unseen data to obtain the optimal model;
[0023] Step 5: Apply the trained model to the new image to perform real-time dead tree detection, and finally obtain an output image containing candidate boxes and confidence levels.
[0024] In step 3, the multi-branch module adopts a multi-branch topology structure, combined with multi-scale convolution, serialized 1x1 and KxK convolution, average pooling and branch addition, which greatly enriches the feature space, but at the same time increases the complexity of training. It is worth noting that the DBB module can be converted into a regular convolution layer during inference, so there is no additional inference cost. This allows DBB to be used as a substitute for existing convolution layers to improve performance without affecting inference efficiency. In step 3, the training process is shown in formula (1):
[0025]
[0026] In step 5, the output during inference is shown in formula (2):
[0027] Y=Conv merged (X)(2)
[0028] In formula (1), X and Y represent the input and output of the DBB module, w1, w2, w3, w4 are the weights of each branch, and Conv 1×1 and Conv K×K denote convolution layers with kernel sizes of 1×1 and K×K, respectively; BN denotes a batch normalization layer; and AvgPool denotes an average pooling layer. In formula (2), Conv merged Represents a new convolution operation, which recalculates the weights and biases of each branch of the original DBB module to ensure that the new convolution operation is as consistent as possible with the original multi-branch output.
[0029] The SEAM module uses multiple residual convolution blocks to enhance the expressiveness of features, and recalibrates feature channels through a channel attention mechanism consisting of an adaptive average pooling layer and a fully connected layer, thereby highlighting important features and suppressing unimportant features. This method not only reduces the number of parameters, but also improves computational efficiency. The output of the residual convolution block is added to the input features through a residual connection. This structure helps the network capture the residual between the input features and the transformed features, thereby effectively compensating for the information loss caused by occlusion and sparse branches, enhancing the attention and expression of dead tree features, and improving the model's ability to recognize dead trees. The main functional blocks and related formulas are shown below:
[0030] (1) Residual convolution block
[0031] The residual convolution block consists of a standard convolution operation followed by a GELU activation function and batch normalization. The output is then added to the input to form the final output. The formula is as follows:
[0032] output=Conv(input)+input(3)
[0033] Among them, Conv is a standard convolution operation, input is the input feature map, and output is the output feature map;
[0034] (2) Adaptive average pooling layer
[0035] Reduce the spatial dimension of the input feature map to 1×1, that is, globally average the features in each channel;
[0036] y=AvgPool(x)(4)
[0037] Where x represents the input feature map, i.e., the output of the residual convolution block in formula (3), and AvgPool represents the adaptive average pooling operation, which reduces the spatial dimension of the feature map to 1×1;
[0038] (3) Fully connected layer
[0039] The feature recalibration is achieved through two fully connected layers. The first fully connected layer compresses the number of channels, and the second fully connected layer restores the number of channels. Finally, the weight factor is output through the Sigmoid function.
[0040] s=σ(FC2(ReLU(FC1(y))))(5)
[0041] Where y is the output of the adaptive average pooling of formula (4); FC1 represents the first fully connected layer for dimensionality reduction, from C in Down to C in / reduction; ReLU is an activation function used to introduce nonlinearity and help the network learn complex functions; FC2 represents the second fully connected layer, which reduces the dimension from C in / reduction restores to C in ; σ is the Sigmoid activation function, which outputs a weight factor s between 0 and 1 for subsequent feature recalibration;
[0042] (4) Feature recalibration
[0043] Use the weight factor output by the fully connected layer to scale each channel of the input feature map;
[0044] output = x exp(s) (6)
[0045] Where x represents the input original feature map, s represents the weight factor calculated by formula (5), and exp(s) represents the application of an exponential function to s, which is used to enhance the model's sensitivity to important features.
[0046] The main function of the CBLinear module is to perform convolution operations on the input feature map and split the output results into multiple feature maps according to the predefined number of channels; this design helps to disperse information to different branches, facilitating subsequent feature fusion.
[0047] For the input feature map
[0048] (1) Use a convolutional layer with a kernel size of k, a stride of s, and a padding of p:
[0049] Y = Conv2d(X, k, s, p) (7)
[0050] Among them, X represents the input feature map, C in represents the number of channels of the input feature map, H represents the height of the input feature map, and W represents the width of the input feature map. indicates that X is a C in Channels, a three-dimensional tensor of height H and width W, whose elements belong to the set of real numbers Conv2d represents a two-dimensional convolution operation, Y is the output feature map after the convolution operation, Σc out It represents the total number of channels of all output feature maps, which is obtained by the sum of the elements in c2s (which contains the number of channels that each output branch should have). H' and W' represent the height and width of the output feature map. It means that Y is a out Channel, a three-dimensional tensor of height H′ and width W′, whose elements belong to the set of real numbers
[0051] (2) Split output channels: Split the convolution output Y into m tensors Y i ,in Each segmented feature map represents the features of different branches of the network, which are used for subsequent fusion (that is, the feature map generated after the convolution operation will be divided into several parts, each of which represents the feature information of a branch in the network. This design allows different network branches to focus on processing different types of information, and then these different information will be recombined to enhance the network's understanding and processing capabilities of the overall data); the mathematical expression is:
[0052] Y=[Y1,Y2,...,Y m ](8)
[0053] Among them, Y i The number of channels is and Represents Y i is a A three-dimensional tensor of a channel with a height of H' and a width of W', where the elements belong to the set of real numbers
[0054] The role of the CBFuse module is to fuse multi-scale and multi-channel feature maps into a unified feature map to meet the different feature requirements of the network; it first performs interpolation adjustment on feature maps of different sizes and then adds them in the feature map dimension to achieve fusion.
[0055] There are currently n input feature maps {Y1, Y2, …, Y n}, with different sizes; Y n is the target size H'×W', and the other feature maps are interpolated and adjusted to this size;
[0056] (1) Interpolation adjustment: For each Y i (where i < n), use nearest neighbor interpolation to adjust it to the target size H'×W':
[0057] Y i ' = Interpolate(Y i , size=(H', W')) (9)
[0058] where Y i ' is the feature map after interpolation adjustment, and Interpolate represents the interpolation operation;
[0059] (2) Add all the adjusted feature maps (including the last one) element-wise to obtain the fused output feature map F:
[0060]
[0061] where F is the feature map obtained after fusing all feature maps;
[0062] In step 3, the process of feature fusion is specifically as follows:
[0063] (1) Feature extraction and preliminary fusion: The left backbone is used to extract the basic feature F L , and the features of each layer are segmented through CBLinear (where i = 1, 3, 5, 7, 9) for subsequent fusion; the right backbone first processes the image through a convolutional layer and the SEAM module to enhance the representation of key features and obtains the feature F SEAM ; subsequently, through the CBFuse module, F SEAM is preliminarily fused with the features of five key levels (i.e., the 1st, 3rd, 5th, 7th, and 9th layers) of the left backbone to generate the feature FF1; this is the first fusion operation in the network.
[0064] (2) Further feature fusion: FF1 undergoes further feature extraction through a DBB module, and then undergoes a second fusion with the features of the 3rd, 5th, 7th, and 9th layers of the left backbone through a CBFuse module to obtain feature FF2; similarly, FF2 undergoes another c2f and DBB module, and then undergoes a third fusion with the features of the 5th, 7th, and 9th layers of the left backbone to generate feature FF3; this process continues, and FF3 undergoes a fourth fusion with the features of the 7th and 9th layers of the left backbone after passing through c2f and DBB modules to obtain FF4; finally, FF4 undergoes a fifth fusion with the features of the 9th layer of the left backbone after passing through c2f and DBB modules to generate feature FF5; FF5 passes through the last c2f module and enters the Neck part of the network, where feature fusion is performed again to maximize feature utilization efficiency, obtaining feature F.
[0065] CBLinear and CBFuse are combined to build a dual-branch feature extraction and fusion backbone. This dual-branch structure enhances the network's expressiveness and ability to adapt to different scales, helping to capture and preserve important information at different scales. In addition, it mitigates the loss of image features during downsampling and enhances feature propagation throughout the network, helping to obtain more accurate and comprehensive detection results.
[0066] Ablation experiment
[0067] In order to verify the effectiveness of each module, an ablation comparison experiment was designed on the test set. Since the proposed DSFI-YOLO is based on the YOLOv8n model, YOLOv8n is used as the benchmark for the ablation experiment. The results are shown in Table 1. After fusing the features of each layer using the dual-branch feature extraction fusion backbone structure, mAP 50 The improvement was 1.0%, which enhanced the recognition effect of small targets and dead trees with incomplete features in the edge area of the image; after replacing the convolution module with DBB, the problem of missed detection caused by occlusion between dead trees was solved, and the mAP 50 In addition, the introduction of the SEAM attention mechanism improves the model's recognition effect on sparse tree crowns, and mAP 50 It increased by 0.8%. In summary, the proposed algorithm mAP 50 Reaching 92.5%, compared with the YOLOv8n algorithm, mAP 50 Improved mAP by 3.1% 50-95 It increased by 0.9%, and P and R increased by 3.7% and 4.9% respectively.
[0068] Table 1 Comparison of ablation experiment results
[0069] Yolov8n DSFI-backbone DBB SEAM P / % R / % <![CDATA[mAP 50 / %]]> <![CDATA[mAP 50-95 / %]]> √ - - - 85.5 80.9 89.4 45.2 - √ - - 87.7 83.0 90.4 45.2 - √ √ - 90.0 85.2 91.7 46.7 - √ √ √ 89.2 85.8 92.5 46.1
[0070] “√” means the module is added, and “-” means no operation is performed.
[0071] Comparative test
[0072] The comparison results of DSFI-YOLO with SSD, FasterR-CNN, LDS-YOLO and ImprovedYOLOv7 on the test set and public datasets are shown in Table 2 and Table 3 respectively.
[0073] Table 2 shows that DSFI-YOLO has the best recognition effect in the dead tree test set. Its mAP50 is 92.5%, which is 9.0%, 12.0%, 5.5% and 6.4% higher than SSD, FasterR-CNN, LDS-YOLO and ImprovedYOLOv7 respectively. The P value is 9.2%, 8.7%, 2.5% and 3.6% higher than SSD, FasterR-CNN, LDS-YOLO and ImprovedYOLOv7; the R value is 2.0%, 6.9% and 7.0% higher than FasterR-CNN, LDS-YOLO and ImprovedYOLOv7, but 0.9% lower than SSD. mAP50-95 is 8.3%, 7.1%, 3.4% and 5.7% higher than SSD, FasterR-CNN, LDS-YOLO and ImprovedYOLOv7 respectively. In terms of FPS, it is only lower than LDS-YOLO, but much higher than other comparison models; in terms of Params, it is only higher than LDS-YOLO, but much lower than other comparison models. Comprehensively comparing average P, R, mAP50, mAP50-95, FPS, and Params, the DSFI-YOLO algorithm shows better performance.
[0074] Table 2 Performance comparison of different target detection models on the test set
[0075] Model P / % R / % <![CDATA[mAP 50 / %]]> <![CDATA[mAP 50-95 / %]]> <![CDATA[FPS / F·s -1 ]]> Params / MB SSD 80.0 86.7 83.5 37.8 39.3 181.0 FasterR-CNN 80.5 83.8 80.5 39.0 42.4 460.0 LDS-YOLO 86.7 78.9 87.0 42.7 131.6 8.0 ImprovedYOLOv7 85.6 79.0 86.1 40.4 46.1 71.3 DSFI-YOLO 89.2 85.8 92.5 46.1 83.3 15.9
[0076] Table 3 shows that in the public data set, LDS-YOLO has the best recognition effect, while the algorithm proposed in this invention is second. From the analysis of mAP50, mAP50-95, FPS and Params, this algorithm is better than SSD, FasterR-CNN and Improved YOLOv7. In addition, combining Tables 2 and 3, it can be found that LDS-YOLO focuses more on P value, while this algorithm is more stable in terms of R value, and mAP50 reaches 81.1%, which is a good performance. This proves that the algorithm proposed in this study has good generalization in the detection of dead trees.
[0077] Table 3 Performance comparison of different target detection models on public datasets
[0078] Model P / % R / % <![CDATA[mAP 50 / %]]> <![CDATA[mAP 50-95 / %]]> <![CDATA[FPS / F·s -1 ]]> Params / MB SSD 77.2 83.4 77.2 37.7 39.2 181.0 FasterR-CNN 68.7 88.5 68.3 39.9 37.3 460.0 LDS-YOLO 86.4 76.6 82.1 42.8 140.8 8.0 ImprovedYOLOv7 68.9 77.7 74.9 41.7 78.1 71.3 DSFI-YOLO 78.8 82.8 81.1 64.0 87.0 15.9
Claims
1. A method for detecting standing dead trees by remote sensing using an unmanned aerial vehicle (UAV) based on a dual-stream feature integration network, characterized in that: The method is: Step 1: Obtain images of dead trees using drones, and use annotation software combined with field surveys to annotate the dead trees; Step 2: Build a dead tree detection dataset, which includes the collected dataset and the public dataset. The collected dataset is divided into training set, validation set and test set. During the training process, online data augmentation (including rotation, scaling, flipping and cropping) is used to augment the training set to enhance the generalization performance of the model. The public dataset is used as an additional test set to verify the generalization performance of the model. Step 3: Use the Diverse Branch Block (DBB) to replace the traditional convolutional module to fully explore the characteristics of dead trees; The Separated and Enhanced Attention Module (SEAM) is used to process the shallow features extracted from the trunk to enhance the attention to the details of the dead trees and obtain the enhanced features F SEAM ; Combining CBLinear and CBFuse to build a dual-branch structure backbone network, the left backbone of the network is responsible for extracting basic features, while the right backbone not only extracts features, but also integrates and fuses the features of the left backbone, and finally obtains feature F; Step 4: Send the feature F obtained in step 3 to the model detection head, train the model to reduce the difference between the predicted result and the true label, and evaluate the detection performance of the model on the validation set to ensure that the model not only learns the characteristics of the data, but also can generalize to unseen data to obtain the optimal model; Step 5: Apply the trained model to the new image to perform real-time dead tree detection, and finally obtain an output image containing candidate boxes and confidence levels.
2. The method for detecting standing dead trees by remote sensing using a UAV based on a dual-stream feature integration network according to claim 1, characterized in that: In step 3, the multi-branch module adopts a multi-branch topology structure, combining multi-scale convolution, serialized 1x1 and KxK convolution, average pooling and branch addition; the training process is shown in formula (1): In step 5, the output during inference is shown in formula (2): Y = Conv merged (X) (2) In formula (1), X and Y represent the input and output of the DBB module, w1, w2, w3, w4 are the weights of each branch, and Conv 1×1 and Conv K×K denote convolution layers with kernel sizes of 1×1 and K×K, respectively; BN denotes a batch normalization layer; and AvgPool denotes an average pooling layer. In formula (2), Conv merged Represents a new convolution operation, which recalculates the weights and biases of each branch of the original DBB module to ensure that the new convolution operation is as consistent as possible with the original multi-branch output.
3. The method for detecting standing dead trees by remote sensing using a UAV based on a dual-stream feature integration network according to claim 1, characterized in that: In step 3, the SEAM module uses multiple residual convolution blocks to enhance the expressiveness of features, and recalibrates feature channels through a channel attention mechanism consisting of an adaptive average pooling layer and a fully connected layer, thereby highlighting important features and suppressing unimportant features; the output of the residual convolution block is added to the input feature through a residual connection; the main functional blocks and related formulas are as follows: (1) Residual convolution block The residual convolution block consists of a standard convolution operation followed by a GELU activation function and batch normalization. The output is then added to the input to form the final output. The formula is as follows: output=Conv(input)+input (3) Among them, Conv is a standard convolution operation, input is the input feature map, and output is the output feature map; (2) Adaptive average pooling layer Reduce the spatial dimension of the input feature map to 1×1, that is, globally average the features in each channel; y=AvgPool(x) (4) Where x represents the input feature map, i.e., the output of the residual convolution block in formula (3), and AvgPool represents the adaptive average pooling operation, which reduces the spatial dimension of the feature map to 1×1; (3) Fully connected layer The feature recalibration is achieved through two fully connected layers. The first fully connected layer compresses the number of channels, and the second fully connected layer restores the number of channels. Finally, the weight factor is output through the Sigmoid function. s=σ(FC2(ReLU(FC1(y)))) (5) Where y is the output of the adaptive average pooling of formula (4); FC1 represents the first fully connected layer for dimensionality reduction, from C in Down to C in / reduction; ReLu is an activation function used to introduce nonlinearity and help the network learn complex functions; FC2 represents the second fully connected layer, which reduces the dimension from C in / reduction restores to C in ; σ is the Sigmoid activation function, which outputs a weight factor s between 0 and 1 for subsequent feature recalibration; (4) Feature recalibration Use the weight factor output by the fully connected layer to scale each channel of the input feature map; output=x·exp(s) (6) Where x represents the input original feature map, s represents the weight factor calculated by formula (5), and exp(s) represents the application of an exponential function to s, which is used to enhance the model's sensitivity to important features.
4. The method for detecting standing dead trees by remote sensing using a UAV based on a dual-stream feature integration network according to claim 1, characterized in that: In step 3, the feature map of the original branch is split into multi-channel feature maps by the CBLinear module, and then these maps are fused with the features of the right trunk by the CBFuse module; The main function of the CBLinear module is to perform convolution operations on the input feature map and split the output result into multiple feature maps according to the predefined number of channels; For the input feature map (1) Use a convolutional layer with a kernel size of k, a step size of s, and a padding of p: Y=Conv2d(X,k,s,p) (7) Among them, X represents the input feature map, C in represents the number of channels of the input feature map, H represents the height of the input feature map, and W represents the width of the input feature map. indicates that X is a C in Channels, a three-dimensional tensor of height H and width W, whose elements belong to the set of real numbers Conv2d represents a two-dimensional convolution operation, Y is the output feature map after the convolution operation, Σc out It represents the total number of channels of all output feature maps, which is obtained by the sum of the elements in c2s (which contains the number of channels that each output branch should have). H' and W' represent the height and width of the output feature map. It means that Y is a out Channel, a three-dimensional tensor of height H′ and width W′, whose elements belong to the set of real numbers (2) Split output channels: Split the convolution output Y into m tensors Y i ,in Each segmented feature map represents the features of different branches of the network, which are used for subsequent fusion; the mathematical expression is: Y=[Y1,Y2,...,Y m ] (8) Among them, Y i The number of channels is and Represents Y i is a Channel, a three-dimensional tensor of height H′ and width W′, whose elements belong to the set of real numbers The function of the CBFuse module is to fuse multi-scale and multi-channel feature maps into a unified feature map to meet the different feature requirements of the network. It first interpolates and adjusts the feature maps of different sizes, and then adds them in the feature map dimension to achieve fusion. There are n input feature maps {Y1,Y2,…,Y n }, sizes vary; Y n If the target size is H′×W′, then all other feature maps are interpolated and adjusted to this size; (1) Interpolation adjustment: For each Y i (where i < n), adjust it to the target size H′×W′ using nearest neighbor interpolation: Y i '=Interpolate(Y i ,size=(H′,W′)) (9) Among them, Y i ' is the feature map after interpolation adjustment, Interpolate represents the interpolation operation; (2) All adjusted feature maps (including the last feature map) are added element by element to obtain the fused output feature map F: Among them, F is the feature map obtained by fusion of all feature maps; Combine CBLinear and CBFuse to build a dual-branch feature extraction and fusion backbone.
5. The method for detecting standing dead trees by remote sensing using an unmanned aerial vehicle (UAV) based on a dual-stream feature integration network according to claim 1 or 4, characterized in that: In step 3, the feature fusion process is as follows: (1) Feature extraction and preliminary fusion: The left backbone is used to extract basic features F L And through CBLinear, the features F of each layer Li (where i = 1, 3, 5, 7, 9) segmentation for subsequent fusion; the right trunk first processes the image through a convolutional layer and the SEAM attention mechanism to enhance the representation of key features and obtain feature F SEAM ; Then, through the CBFuse module, F SEAM It is preliminarily fused with the features of the five key levels (i.e., the 1st, 3rd, 5th, 7th, and 9th layers) of the left trunk to generate feature FF1; (2) Further feature fusion: FF1 undergoes further feature extraction through a DBB module, and then undergoes a second fusion with the features of the 3rd, 5th, 7th, and 9th layers of the left backbone through a CBFuse module to obtain feature FF2; similarly, FF2 undergoes another c2f and DBB module, and then undergoes a third fusion with the features of the 5th, 7th, and 9th layers of the left backbone to generate feature FF3; this process continues, and FF3 undergoes a fourth fusion with the features of the 7th and 9th layers of the left backbone after passing through c2f and DBB modules to obtain FF4; finally, FF4 undergoes a fifth fusion with the features of the 9th layer of the left backbone after passing through c2f and DBB modules to generate feature FF5; FF5 passes through the last c2f module and enters the Neck part of the network, where feature fusion is performed again to maximize feature utilization efficiency, obtaining feature F.
Citation Information
Patent Citations
Large-range pine wood nematode disease dead standing tree intelligent detection method based on unmanned aerial vehicle and CNN
CN112084989A
Weed detection method based on multi-scale fusion module and feature enhancement
CN113657326A
Infrared remote sensing image target detection method based on YOLOv8, electronic equipment and computer readable storage medium
CN118887511A
Method, device, and storage medium for deep learning based domain adaptation with data fusion for aerial image data analysis
US20220092420A1
AU2020103026A4