Improved weed identification algorithm based on YOLOv5-SPD
Accurate identification of weeds is the primary prerequisite for achieving machine replacement of manual weeding.The target of nascent weeds is small,making identification difficult.YOLOv5-SPD has good performance in small target recognition,but its robustness and accuracy still need to be improved.Adding channel attention mechanism on the basis of YOLOv5-SPD can strengthen the weight value of effective features,making the learning of the network more targeted.At the same time,replacing the generalized intersection over union(GIoU)loss function with complete intersection over union(CIoU)can effectively solve the problem of border coincidence,the height width ratio of the target box and the prediction box,and the relationship between the center point,there by making the weed prediction box closer to the real box.The experimental results on the weed dataset show that the improved network detection accuracy reaches 70.3%with an accuracy rate of 94.1%which is 4.7%and 2.8%higher than the original YOLOv5-SPD.
attention mechanismboundary loss functionYOLOv5-SPD algorithmweed identification
NETL
NSTL
万方数据
维普
