Application of Attention Mechanism Object Detection Algorithm in Walnut Fruit Recognition
The recognition of walnut fruits can be applied to the yield prediction work of large walnut planting parks.Accurately identifying the fruits on walnut trees is of great significance for improving the accuracy of yield prediction.To solve the existing challenges,this paper proposes an Attention-YOLO walnut object detection algo-rithm for accurately identifying walnuts in unmanned aerial vehicle remote sensing images,so as to achieve the purpose of providing technical support for walnut yield prediction.First,the CBAM attention-enhanced Backbone network is introduced to improve the feature extraction ability of the algorithm.Then,the multiple attention mechanism module is used to reshape the original Head layer and optimize the target positioning accuracy.Exper-iments show that the detection accuracy(mAP0.5)of the Attention-YOLO algorithm for walnuts reaches 92.6％,and the recall rate(R)reaches 96％.Compared with other mainstream models,it has obvious advantages.
NETL
NSTL
万方数据
