目标检测数据集半自动生成技术研究

作者：孙晓璇^*; 张磊; 李健

来源：计算机系统应用, 2019, 28(10): 8-14.

DOI：10.15888/j.cnki.csa.007101

• YOLOv3
• SSD
• 差异值哈希
• 半自动生成训练集

目标检测广泛使用于计算机视觉领域.在不同的场景中,我们需要使用不同的数据集训练模型.但是,人工生成数据集标签非常耗时.本文提出一种半自动的方法生成数据集标签,然后按照图像相似度设置的阈值自动筛选,最后保留符合要求的图像和对应的标签作为最终的数据集.实验表明,该方法可以提高数据集生成标签的速度,同时确保了准确率.

• 单位
  中国科学院计算机网络信息中心; 中国科学院大学