机械机制数据集
计算机视觉与模式识别
2024-09-09 v1
摘要
本研究介绍了一个约 9,000 张机械机制图像及其对应描述的数据集，旨在支持机制设计研究。数据集包含多样化的 2D 和 3D 草图，经过严格筛选确保其相关性和质量。我们通过微调两个模型来演示该数据集的应用：1）Stable Diffusion（用于生成新的机械设计），2）BLIP-2（用于描述这些设计）。虽然 Stable Diffusion 的结果显示出前景，尤其在生成连贯的 3D 草图方面，但该模型在处理 2D 草图时表现不足，偶尔会产生毫无意义的输出。这些局限性凸显了需要进一步开发的需求，尤其是扩大数据集规模和细化模型架构。尽管如此，本工作为利用生成式 AI 在机械设计中的应用指明了方向，突显了这些方法的潜力及当前的局限性。
引用
@article{arxiv.2409.03763,
title = {A Dataset for Mechanical Mechanisms},
author = {Farshid Ghezelbash and Amir Hossein Eskandari and Amir J Bidhendi},
journal= {arXiv preprint arXiv:2409.03763},
year = {2024}
}