SortScrews: A Dataset and Baseline for Real-time Screw Classification
SortScrews:一种用于实时螺钉分类的数据集和基线
机构 * University of Toronto(多伦多大学) ; Vector Institute(向量研究所) ; Project Neura ; UTMIST ; Amplimit
AI总结 本文提出SortScrews数据集,用于解决工业自动化中螺钉分类问题,通过标准化采集设置获取6种螺钉类型和背景类图像,采用EfficientNet-B0和ResNet-18进行迁移学习,验证了在小数据集下可控采集条件下的有效学习能力。