论文标题
一种用于自动生成的标记植物图像的嵌入式系统,以启用机器学习应用程序
An embedded system for the automated generation of labeled plant images to enable machine learning applications in agriculture
论文作者
论文摘要
在各种和数量方面,缺乏足够的培训数据通常是任何领域中机器学习(ML)应用程序开发的瓶颈。对于农业应用,基于ML的模型旨在执行诸如自主植物分类之类的任务,通常仅将其耦合到一种或可能只有几种植物物种。结果,每个特定于作物的任务很可能需要其自己的专业培训数据,以及如何满足对数据需求的问题,现在通常会掩盖更常规的实际训练此类模型。为了解决这个问题,我们开发了一个嵌入式的机器人系统,以自动生成和标记大型植物图像,以用于农业中的ML应用。该系统几乎可以从任何角度对植物进行成像,从而确保各种数据。并且,每秒最多可达一个图像的成像速率,它可以在每天达到数千至数万张图像的规模上产生带长的数据集。因此,该系统为手动生成和标签的时间和成本密集的方法提供了重要的选择。此外,使用由蓝色钥匙织物制成的均匀背景可以实现其他图像处理技术,例如背景置换和植物分割。它还有助于训练过程,从本质上讲,迫使模型专注于植物特征并消除随机相关性。为了展示系统的功能,我们生成了一个超过34,000个标记图像的数据集,我们通过该数据集训练了ML模型,以将草与测试数据中的非花草区分开,并将其与各种来源区分开。现在,我们计划生成更大的加拿大农作物和杂草数据集,以便在农业领域进一步实现ML应用。
A lack of sufficient training data, both in terms of variety and quantity, is often the bottleneck in the development of machine learning (ML) applications in any domain. For agricultural applications, ML-based models designed to perform tasks such as autonomous plant classification will typically be coupled to just one or perhaps a few plant species. As a consequence, each crop-specific task is very likely to require its own specialized training data, and the question of how to serve this need for data now often overshadows the more routine exercise of actually training such models. To tackle this problem, we have developed an embedded robotic system to automatically generate and label large datasets of plant images for ML applications in agriculture. The system can image plants from virtually any angle, thereby ensuring a wide variety of data; and with an imaging rate of up to one image per second, it can produce lableled datasets on the scale of thousands to tens of thousands of images per day. As such, this system offers an important alternative to time- and cost-intensive methods of manual generation and labeling. Furthermore, the use of a uniform background made of blue keying fabric enables additional image processing techniques such as background replacement and plant segmentation. It also helps in the training process, essentially forcing the model to focus on the plant features and eliminating random correlations. To demonstrate the capabilities of our system, we generated a dataset of over 34,000 labeled images, with which we trained an ML-model to distinguish grasses from non-grasses in test data from a variety of sources. We now plan to generate much larger datasets of Canadian crop plants and weeds that will be made publicly available in the hope of further enabling ML applications in the agriculture sector.