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

用于深度学习的合成图像数据

Synthetic Image Data for Deep Learning

BMW IT Research Center(宝马IT研究中心);Clemson University(克莱姆森大学)

Jason W. Anderson, Marcin Ziolkowski, Ken Kennedy, Amy W. Apon

摘要:

从3D模型渲染的真实合成图像数据可用于增强图像集和训练图像分类语义分割模型。在这项工作中,我们将探讨如何高质量的基于物理的渲染和域随机化可以有效地创建一个大型的合成数据集的基础上生产的三维CAD模型的真实车辆。我们使用这个数据集来量化使用U网和双U网模型的合成增强的有效性。我们发现,在这个领域,合成图像是一种有效的技术,用于增强有限的真实训练数据集。我们观察到,在纯合成图像上训练的模型在真实验证图像上的平均预测IoU非常低。我们还观察到,即使在合成数据集中添加非常少量的真实图像,也会大大提高准确性,并且在使用合成图像增强的数据集上训练的模型比仅在真实图像上训练的模型更准确。最后,我们发现,在受益于增量训练或模型专业化的用例中,在合成图像上预训练基础模型可以大大降低迁移学习的训练成本,允许高达90%的模型训练被预先加载。

英文摘要:

Realistic synthetic image data rendered from 3D models can be used to augment image sets and train image classification semantic segmentation models. In this work, we explore how high quality physically-based rendering and domain randomization can efficiently create a large synthetic dataset based on production 3D CAD models of a real vehicle. We use this dataset to quantify the effectiveness of synthetic augmentation using U-net and Double-U-net models. We found that, for this domain, synthetic images were an effective technique for augmenting limited sets of real training data. We observed that models trained on purely synthetic images had a very low mean prediction IoU on real validation images. We also observed that adding even very small amounts of real images to a synthetic dataset greatly improved accuracy, and that models trained on datasets augmented with synthetic images were more accurate than those trained on real images alone. Finally, we found that in use cases that benefit from incremental training or model specialization, pretraining a base model on synthetic images provided a sizeable reduction in the training cost of transfer learning, allowing up to 90\% of the model training to be front-loaded.

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

↑