arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2212.04790cs.CV
收藏

工业应用中用于对象分类的合成数据

Synthetic Data for Object Classification in Industrial Applications

HMS Industrial Networks AB(HMS工业网络公司)

August Baaz, Yonan Yonan, Kevin Hernandez-Diaz, Fernando Alonso-Fernandez, Felix Nilsson

摘要:

机器学习最大的挑战之一是数据汇集。训练数据是一个重要的部分,因为它决定了模型的行为。在对象分类中,在不同条件下捕获每个对象的大量图像并不总是可能的,并且可能非常耗时和繁琐。因此,这项工作探索了使用游戏引擎来创建人工图像,以应对训练数据集中的有限数据。我们结合真实数据和合成数据来训练对象分类引擎,这一策略已被证明有利于提高分类器决策的信心,这在工业设置中往往至关重要。为了结合真实数据和合成数据,我们首先在大量合成数据上训练分类器,然后在真实图像上对其进行微调。另一个重要的结果是,微调所需的真实图像数量不是很高,达到最高精度时,每个类别只有12或24张图像。这大大降低了捕获大量真实数据的要求。

英文摘要:

One of the biggest challenges in machine learning is data collection. Training data is an important part since it determines how the model will behave. In object classification, capturing a large number of images per object and in different conditions is not always possible and can be very time-consuming and tedious. Accordingly, this work explores the creation of artificial images using a game engine to cope with limited data in the training dataset. We combine real and synthetic data to train the object classification engine, a strategy that has shown to be beneficial to increase confidence in the decisions made by the classifier, which is often critical in industrial setups. To combine real and synthetic data, we first train the classifier on a massive amount of synthetic data, and then we fine-tune it on real images. Another important result is that the amount of real images needed for fine-tuning is not very high, reaching top accuracy with just 12 or 24 images per class. This substantially reduces the requirements of capturing a great amount of real data.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑