论文标题

从合成的大豆豆荚数据集中转移学习,以用于分支的原位大豆豆荚的原位分割

Transfer Learning from Synthetic In-vitro Soybean Pods Dataset for In-situ Segmentation of On-branch Soybean Pod

论文作者

Yang, Si, Zheng, Lihua, Chen, Xieyuanli, Zabawa, Laura, Zhang, Man, Wang, Minjuan

论文摘要

成熟的大豆植物具有复杂的建筑,豆荚经常互相接触,这对原位分割的分支大豆豆荚构成了挑战。基于深度学习的方法可以实现准确的培训和强大的概括能力,但是它需要大量标记的数据,这通常是一个限制,尤其是对于农业应用。由于缺少标记的数据来训练分支大豆豆荚的原位分割模型,因此我们提出了从合成的维特罗内大豆豆荚中进行的转移学习。首先,我们提出了一种新型的自动化图像生成方法,可以快速生成带有大量带注释的样品的合成体内大豆荚数据集。体外大豆豆荚样品被重叠,以模拟经常物理接触的分支大豆豆荚。然后,我们设计了两步转移学习。在第一步中,我们通过源域(MS COCO数据集)列出了一个具有合成目标域(Vitro sybean Pods数据集)的实例分割网络。在第二步中,通过在一些现实世界中成熟的大豆样品上进行填充来进行从模拟转移到现实。实验结果表明了所提出的两步传输学习方法的有效性,因此现实世界中成熟的大豆植物测试数据集的AP $ _ {50} $为0.80,该数据集高于直接适应和其AP $ _ {50} $ 0.77。此外,分支大豆豆荚的原位分割结果的可视化表明,我们的方法的性能比其他方法更好,尤其是当大豆豆荚密集地重叠时。

The mature soybean plants are of complex architecture with pods frequently touching each other, posing a challenge for in-situ segmentation of on-branch soybean pods. Deep learning-based methods can achieve accurate training and strong generalization capabilities, but it demands massive labeled data, which is often a limitation, especially for agricultural applications. As lacking the labeled data to train an in-situ segmentation model for on-branch soybean pods, we propose a transfer learning from synthetic in-vitro soybean pods. First, we present a novel automated image generation method to rapidly generate a synthetic in-vitro soybean pods dataset with plenty of annotated samples. The in-vitro soybean pods samples are overlapped to simulate the frequently physically touching of on-branch soybean pods. Then, we design a two-step transfer learning. In the first step, we finetune an instance segmentation network pretrained by a source domain (MS COCO dataset) with a synthetic target domain (in-vitro soybean pods dataset). In the second step, transferring from simulation to reality is performed by finetuning on a few real-world mature soybean plant samples. The experimental results show the effectiveness of the proposed two-step transfer learning method, such that AP$_{50}$ was 0.80 for the real-world mature soybean plant test dataset, which is higher than that of direct adaptation and its AP$_{50}$ was 0.77. Furthermore, the visualizations of in-situ segmentation results of on-branch soybean pods show that our method performs better than other methods, especially when soybean pods overlap densely.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源