中国农业机械化科学研究院集团有限公司 主管

北京卓众出版有限公司 主办

基于半监督学习的秧苗计数与定位算法研究

Seedling counting and localization algorithm based on semi-supervised learning

  • 摘要: 准确测定水稻株数和位置在水稻育种和栽培中具有至关重要的作用。传统的田间抽样调查法费时费力、准确率低,无人机遥感技术为水稻秧苗的智能精准计数和定位提供了有效途径。针对无人机图像处理需要大量数据标注的问题,提出了一种基于半监督学习的秧苗计数与定位算法,称为稀疏点标注网络(SPANet)。SPANet根据Teacher-Student训练范式构建,并针对稀疏点注释生成伪标签不准确的问题,提出的点聚合模块(PAM)能从有限的点注释信息生成高质量的伪标签。为了验证SPANet的性能,构建了一个包含300张高分辨率(1600像素×1600像素)的水稻计数与定位数据集,并在50%和80%的注释比例下进行了详尽的试验。结果表明,SPANet在80%标注比例下,平均绝对误差(MAE)和均方根误差(RMSE)分别为15.7和20.2,从而实现了水稻秧苗的精准、高效计数与定位。

     

    Abstract: Accurately measuring number and location of rice plants plays a vital role in rice breeding and cultivation.Traditional field sampling surveys are time-consuming, labor-intensive, and inaccurate.UAV remote sensing technology provides an effective way to intelligently and accurately count and locate rice seedlings.Aiming at a problem of UAV image processing required a large amount of data annotation, a seedling counting and localization algorithm based on semi-supervised learning was proposed, called sparse point annotation network(SPANet).SPANet was built according to Teacher-Student training paradigm, and with a view to problem of inaccurate generation pseudo-labels from sparse point annotations, the proposed point aggregation module(PAM)could generate high-quality pseudo-labels from limited point annotation information.To verify performance of SPANet, a rice counting and localization dataset containing 300 high-resolution(1600 pixel ×1600 pixel)images was constructed, and conducted detailed experiments at annotation ratios of 50% and 80%.Results showed that when annotation ratio was 80%, mean absolute error(MAE)and mean square error(RMSE)of SPANet were 15.7 and 20.2 respectively, thus achieving accurate and efficient counting and localization of rice seedlings.

     

/

返回文章
返回