Crop weed image recognition of UAV based on improved HRNet-OCRNet

Abstract

Aiming at the problem of low model recognition accuracy caused by high similarity and mutual occlusion between crops and weeds in unmanned aerial vehicle (UAV) images, a pixel-level weed recognition method based on improved HRNet-OCRNet is proposed. In this method, a multi-stage and multi-scale feature fusion method is added to HRNet to preserve more details and enhance semantic information at different levels, to solve the problem of high similarity between crops and weeds; The spatial self-attention module of polarized self-attention (PSA) is integrated to HRNet, enhance the network's learning of important features, and reduce the false identification caused by mutual occlusion of crops and weeds; The expansion prediction method is used to generate an accurate distribution map of crop weeds. Compared with Deeplabv3+, GCNet and K-Net, the experimental results show that the proposed method has higher recognition accuracy for crop weeds, and mean intersection over union (mIoU) reaches 85.76%.

Intelligent weight prediction of cows based on semantic segmentation and back propagation neural network

“…To further validate the advanced capabilities of the algorithm proposed in this study, a comparative analysis was performed against state-of-the-art algorithms commonly used in the field, namely PSPNet (Zhao et al, 2017 ), PSANet (Zhao et al, 2018 ), OCRNET (Yuan et al, 2020 ), and HRNET (Wang et al, 2020a ). These algorithms were chosen due to their prominent standing in the field and their demonstrated efficacy in similar tasks including fruits segmentation (Qiao et al, 2022 ; Qi et al, 2022 ), land segmentation (Yuan et al, 2021 ), crop and weed segmentation (Huang et al, 2021 ; Yang et al, 2023 ).…”