Optics and Precision Engineering, Volume. 28, Issue 1, 200(2020)
Straw detection algorithm based on semantic segmentation in complex farm scenarios
The traditional segmentation algorithms for straw coverage detection basing on thresholds or texture features were difficult to get rid of the disadvantages of low accuracy, high complexity and time-consuming, and the effect of segmentation on complex farmland scenes containing a lot of interference factors was not good. Therefore, this paper proposed a semantic segmentation algorithm (DSRA-UNet) with high accuracy, a small mount of training parameters and high running speed. Combined with UNet′s symmetric codec architecture, this algorithm used standard convolution in shallow feature maps, and depthwise separable convolution in deep ones. Residual structure was built in each layer to increase the network depth,which can reduce the number of parameters and improve the accuracy at the same time. In addition, the global maximum pooling attention mechanism was added during the skip connection process to further improve the segmentation accuracy of the network. The algorithm was verified on the straw datasets, and the experiment results showed that the mean of intersection over union reached to 94.3% in the proposed algorithm of this paper. The number of training parameters of the algorithm was only 0.76 M, and the test time of single picture was less than 0.05 s. The algorithm could accurately segment the straw and soil, and separate the interference information in the complex environment, especially solving the shadow problem in image.
Get Citation
Copy Citation Text
LIU Yuan-yuan, ZHANG Shuo, YU Hai-ye, WANG Yue-yong, WANG Jia-mu. Straw detection algorithm based on semantic segmentation in complex farm scenarios[J]. Optics and Precision Engineering, 2020, 28(1): 200
Category:
Received: Aug. 27, 2019
Accepted: --
Published Online: Mar. 25, 2020
The Author Email: Yuan-yuan LIU (liuyuanyuan1980@126.com)