高级检索+

基于RCL-UNet的机直播油菜幼苗覆盖度估计

Estimation of vegetation coverage for mechanically direct-seeded rapeseed seedlings using RCL-UNet

  • 摘要: 针对大田环境下机直播油菜幼苗覆盖度估计方法不完善而制约油菜播种质量精准评估的问题,该研究提出一种改进U-Net语义分割模型RCL-UNet(RCL-U-sharped Network)。该模型采用ResNet50-PPM-S(resnet50-pyramid pooling module)作为编码器以增强特征提取能力,在跳跃连接中引入CBAM(convolutional block attention module)注意力机制以强化空间与通道特征融合,并采用Diceloss损失函数以改善小目标分割性能。试验结果表明,RCL-UNet在油菜幼苗图像分割任务中的平均交并比mIoU、平均像素精度mPA、平均精准率mP和平均召回率mR分别为89.88%、94.98%、93.96%和93.92%,相比原始U-Net模型分别提升了16.75、15.33、8.21和14.69个百分点。油菜覆盖度估计结果表明,在机械条播和飞机撒播条件下,改进模型预测植被覆盖度与人工真值的RMSE均不高于0.5%,决定系数R2均不低于0.94,改进模型在实际场景中具有较好的应用价值与鲁棒性,可为机直播油菜播种性能评价体系的建立提供技术支持。

     

    Abstract: Accurate estimation of seedling vegetation coverage for mechanically direct-seeded rapeseed under field conditions is a critical prerequisite for evaluating sowing quality and guiding precise management practices such as gap filling and thinning. However, existing methods remain inadequate due to the small target size, dense distribution, complex soil backgrounds, and high visual similarity between rapeseed seedlings and weeds at early growth stages. To address these challenges, this study proposes an improved semantic segmentation model named RCL-UNet (RCL-U-shaped Network) based on the classical U-Net architecture. A dedicated dataset was constructed from high-resolution RGB images acquired by a DJI Mavic 3 Pro UAV over typical mechanically direct-seeded rapeseed fields in Anhui Province, China. After image cropping, screening, and pixel-level annotation, a total of 3,140 valid image tiles were obtained, including 2,960 training tiles (after data augmentation), 120 validation tiles, and 60 test tiles. Three key improvements were introduced into the U-Net baseline: First, the original encoder was replaced with a ResNet50-PPM-S structure, which integrates a 50-layer deep residual network with a pyramid pooling module to extract multi-scale contextual features and enhance sensitivity to small targets. Second, a convolutional block attention module (CBAM) was embedded into the skip connections to adaptively recalibrate feature responses through the synergistic effect of channel and spatial attention, effectively suppressing background noise while preserving fine boundary details of seedlings. Third, the Dice loss function was adopted instead of cross-entropy loss to alleviate the severe class imbalance in small-target segmentation, thereby improving pixel-wise coverage of seedling regions. Ablation experiments confirmed that all three modules contributed positively to segmentation performance. The RCL-UNet achieved a mean Intersection over Union (mIoU) of 89.88%, a mean Pixel Accuracy (mPA) of 94.98%, a mean Precision (mP) of 93.96%, and a mean Recall (mR) of 93.92% on the test set, representing absolute improvements of 16.75, 15.33, 8.21, and 14.69 percentage points over the original U-Net, respectively. Compared with PSPNet and DeepLabv3+, RCL-UNet outperformed both with mIoU gains of 10.78 and 3.33 percentage points, respectively. Visual assessments under varying planting densities (300,000–600,000 plants·ha-1) and imaging distances (0.8 m and 2.5 m) showed that RCL-UNet produced more complete and boundary-accurate segmentation masks with significantly fewer omissions and adhesions, especially in distant-view and high-density scenes. The improved model was deployed on a laptop equipped with an RTX 4060 GPU via a PyQt5 interface, achieving an average inference time of 49 ms per image. Field validation was conducted at an independent site under two sowing patterns: mechanical drill sowing and aerial broadcasting, with 100 sampling units of 1 m × 1 m established for each pattern. Vegetation coverage estimation results showed that under mechanical drill sowing, the root mean square error (RMSE) between model predictions and manual ground truth was 0.42% with a coefficient of determination (R2) of 0.98; under aerial broadcasting, the RMSE was 0.50% and R2 was 0.94. Across both sowing patterns, the RMSE values were consistently below 0.5% and the R2 values remained above 0.94, demonstrating that the improved model achieved high pixel-wise segmentation accuracy and exhibited robust generalization capabilities when deployed in complex and variable field environments. The proposed method can provide reliable technical support for establishing a scientific evaluation system for sowing performance of mechanically direct-seeded rapeseed. Nevertheless, the current model has only been validated at the four-leaf stage and has not yet covered the full growth cycle where canopy overlap becomes more pronounced. Moreover, the slightly lower prediction accuracy under aerial broadcasting compared with mechanical drill sowing indicates that further adaptation to irregular sowing patterns is needed. Future work will extend the dataset to multiple growth stages, incorporate multi-source imagery, and explore lightweight architectures for edge deployment to further enhance model generalization and operational efficiency.

     

/

返回文章
返回