Segmentation of field grape bunches via an improved pyramid scene parsing network  

作　　者：Shan Chen Yuyang Song Jinya Su Yulin Fang Lei Shen Zhiwen Mi Baofeng Su 

机构地区：[1]College of Mechanical and Electronic Engineering,Northwest A&F University,Yangling 712100,Shaanxi,China [2]Key Laboratory of Agricultural Internet of Things,Ministry of Agriculture and Rural Affairs,Yangling 712100,Shaanxi,China [3]Shaanxi Key Laboratory of Agricultural Information Perception and Intelligent Services,Yangling 712100,Shaanxi,China [4]College of Enology,Northwest A&F University,Yangling 712100,Shaanxi,China [5]School of Computer Science and Electronic Engineering,University of Essex,Colchester,CO43SQ,UK

出　　处：《International Journal of Agricultural and Biological Engineering》2021年第6期185-194,共10页国际农业与生物工程学报（英文）

基　　金：supported by the Key R&D Project of Ningxia Hui Autonomous Region(Grant No.2019BBF02013);Guangxi Key R&D Program Project(Grant No.Gui Ke AB21076001).

摘　　要：With the continuous expansion of wine grape planting areas,the mechanization and intelligence of grape harvesting have gradually become the future development trend.In order to guide the picking robot to pick grapes more efficiently in the vineyard,this study proposed a grape bunches segmentation method based on Pyramid Scene Parsing Network(PSPNet)deep semantic segmentation network for different varieties of grapes in the natural field environments.To this end,the Convolutional Block Attention Module(CBAM)attention mechanism and the atrous convolution were first embedded in the backbone feature extraction network of the PSPNet model to improve the feature extraction capability.Meanwhile,the proposed model also improved the PSPNet semantic segmentation model by fusing multiple feature layers(with more contextual information)extracted by the backbone network.The improved PSPNet was compared against the original PSPNet on a newly collected grape image dataset,and it was shown that the improved PSPNet model had an Intersection-over-Union(IoU)and Pixel Accuracy(PA)of 87.42%and 95.73%,respectively,implying an improvement of 4.36%and 9.95%over the original PSPNet model.The improved PSPNet was also compared against the state-of-the-art DeepLab-V3+and U-Net in terms of IoU,PA,computation efficiency and robustness,and showed promising performance.It is concluded that the improved PSPNet can quickly and accurately segment grape bunches of different varieties in the natural field environments,which provides a certain technical basis for intelligent harvesting by grape picking robots.

关 键 词：grape bunches semantic segmentation deep learning improved PSPNet 

分 类 号：S66[农业科学—果树学]