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

Research on Semi-Automatic Generation Technology of Object Detection Datasets

作　　者：孙晓璇 张磊[1] 李健[1] SUN Xiao-Xuan;ZHANG Lei;LI Jian(Computer Network Information Center,Chinese Academy of Sciences,Beijing 100190,China;School of Computer and Control Engineering,University of Chinese Academy of Sciences,Beijing 100049,China)

机构地区：[1]中国科学院计算机网络信息中心,北京100190 [2]中国科学院大学计算机与控制学院,北京100049

出　　处：《计算机系统应用》2019年第10期8-14,共7页Computer Systems & Applications

基　　金：中国科学院科技服务网络计划(KFJ-STS-QYZD-058)~~

摘　　要：目标检测广泛使用于计算机视觉领域.在不同的场景中,我们需要使用不同的数据集训练模型.但是,人工生成数据集标签非常耗时.本文提出一种半自动的方法生成数据集标签,然后按照图像相似度设置的阈值自动筛选,最后保留符合要求的图像和对应的标签作为最终的数据集.实验表明,该方法可以提高数据集生成标签的速度,同时确保了准确率.Object detection is widely used in the field of computer vision.In different occasions,we need to use different training set to train the model.However,manually generating label is very time consuming.This study proposed a semi-automatic method to generate labels for dataset,then automatically filter them according to the threshold set by image similarity,lastly retain the required images and corresponding labels as the final dataset.Experiments show that the method can both improve the speed and ensure accuracy rate of generating labels for dataset.

关 键 词：YOLOv3 SSD 差异值哈希 半自动生成训练集 

分 类 号：TP3[自动化与计算机技术—计算机科学与技术]