关键词: 树枝识别, YOLOv4, 语义分割, 数据集训练, BIOU边框回归损失函数

Abstract: In computer vision, it was difficult to train and recognize objects with unclear features, and improve the detection in many fields. In order to identify the branches with less obvious characteristics, mainly the branches that covered the apple position when the manipulator picked the apple, a method to obtain the branch semantic skeleton and identify the branch position box by combining semantic segmentation and YOLOv4 algorithm was proposed. Before using the data set for training, the method of semantic segmentation to divide the rectangular envelope of branches, eliminate the small branches and branches that affected the effect of branch recognition, and then label the data set with labelimg and labelme tools was used; Three layers of maximum pooling layer were added to the trained network model, and the CIOU of YOLOv4 was improved in terms of regression loss. A confidence correlation function BIOU was proposed to improve the prediction accuracy according to the complex characteristics and suitable range. The final experiment showed that the F1 and AP of the tree branches with occluded apple positions trained by the improved YOLOv4 network model were 20.00 precentage and 23.36 precentage higher than those of all the trees trained by the original network.

Key words: branch recognition, YOLOv4, semantic segmentation, dataset training, BIOU frame regression loss function