Recognition of tea sprouts based on improved deeplabV3 + model
摘要
The recognition of tea sprouts is the premise of realize the intelligence of the premium tea picking. DeepLabV3 + , as the latest semantic segmentation algorithm of DeepLab family, can well recognize the tea sprouts in images. However, the model still has the problem of large number of parameters, and there is still room for improvements in segmentation accuracy and speed. To solve the multi-label recognition and segmentation of background, single bud, one leaf and two leaves in the tea sprout images, an improved deep convolutional neural network algorithm is proposed. This method realizes the recognition of different postures of high-quality tea sprouts on the premise of efficiency and accuracy. The DMResNet18 model is used as the feature extraction network to lightweight the network model and increase the segmentation speed. The feature joint attention network is added to optimize the spatial weights and channel weights of the model, which improves the accuracy of segmentation. Additionally, the low-level feature fusion method is applied to obtain rich initial global information of features and improves the overall performance of the model. The results show that the segmentation accuracy of improved DeepLabV3 + model reaches 92.58%, which is improved by 4.26% compared with the original DeepLabV3 + model. The model size and the speed of detection are improved by 80.67% and 52.24%, respectively. The proposed method offers a solution for the intelligent recognition of tea sprouts.