构建了560张螺钉图像数据集,支持工业级实时螺钉分类。
SortScrews: A Dataset and Baseline for Real-time Screw Classification
- 采集560张512×512螺钉图像,含6类螺钉与背景
- 轻量模型在小数据下准确率达90%以上,验证可控采集有效性
- 开源数据集+采集脚本,适合自动化与机器人研发人员
自动识别螺钉类型对工业自动化、机器人和库存管理至关重要。然而,公开的螺钉分类数据集稀缺,尤其缺乏在自动化分拣系统中常见的单对象控制场景数据。本文提出SortScrews数据集,用于螺钉的逐例视觉分类。该数据集包含560张512×512分辨率的RGB图像,涵盖六种螺钉类型及背景类。图像通过标准化采集流程获取,包含光照和相机视角的轻微变化,共四种拍摄设置。为支持可复现研究与数据扩展,我们提供可复用的数据采集脚本,用户可用低成本摄像头设备构建自定义硬件组件数据集。基于ImageNet预训练的EfficientNet-B0和ResNet-18模型进行迁移学习,建立基线结果。尽管数据集规模有限,这些轻量模型仍取得优异分类精度,表明在可控采集条件下,小数据也能实现有效学习。数据集、采集流程与基线代码已公开于https://github.com/ATATC/SortScrews。
原文摘要 · Abstract (English)
Automatic identification of screw types is important for industrial automation, robotics, and inventory management. However, publicly available datasets for screw classification are scarce, particularly for controlled single-object scenarios commonly encountered in automated sorting systems. In this work, we introduce $\textbf{SortScrews}$, a dataset for casewise visual classification of screws. The dataset contains 560 RGB images at $512\times512$ resolution covering six screw types and a background class. Images are captured using a standardized acquisition setup and include mild variations in lighting and camera perspective across four capture settings. To facilitate reproducible research and dataset expansion, we also provide a reusable data collection script that allows users to easily construct similar datasets for custom hardware components using inexpensive camera setups. We establish baseline results using transfer learning with EfficientNet-B0 and ResNet-18 classifiers pretrained on ImageNet. In addition, we conduct a well-explored failure analysis. Despite the limited dataset size, these lightweight models achieve strong classification accuracy, demonstrating that controlled acquisition conditions enable effective learning even with relatively small datasets. The dataset, collection pipeline, and baseline training code are publicly available at https://github.com/ATATC/SortScrews.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。