难例挖掘在太赫兹成像目标检测中的应用  

Application of Hard Example Mining to Terahertz Imaging Object Detection

作　　者：薛飞 梁栋[1,2] 喻洋 XUE Fei;LIANG Dong;YU Yang(MIIT Key Laboratory of Pattern Analysis and Machine Intelligence,School of Computer Science and Technology,Nanjing University of Aeronautics and Astronautics,Nanjing 211106,China;Collaborative Innovation Center of Novel Software Technology and Industrialization,Nanjing 210093,China;Institute of Electronic Engineering,China Academy of Engineering Physics,Mianyang 621900,China)

机构地区：[1]南京航空航天大学计算机科学与技术学院模式分析与机器智能工业和信息化部重点实验室,江苏南京211106 [2]软件新技术与产业化协同创新中心,江苏南京210093 [3]中国工程物理研究院电子工程研究所,四川绵阳621900

出　　处：《计算机技术与发展》2021年第9期124-130,136,共8页Computer Technology and Development

基　　金：国家重点研发计划(2017YFB0802300);国家自然科学基金(61601223)。

摘　　要：太赫兹成像中的隐蔽物体检测是公共安全和反恐的迫切需要。由于太赫兹成像质量差,在太赫兹图像上的目标检测比在计算机视觉领域常用的公共目标检测数据集上要困难得多。文中收集了一个多目标的主动太赫兹成像数据集。针对样本不平衡问题,对比了RetinaNet使用交叉熵和Focal Loss作为损失函数时的检测性能。针对那些检测效果较差的目标,利用难例挖掘技术来增强训练模型。由于传统的难例挖掘技术是在二阶段目标检测器基础上设计的,无法直接应用在一阶段检测器上,文章以RetinaNet为基础设计了一种以图像为单位的难例挖掘方案。实验也验证了YOLOv3、YOLOv4、FRCN-OHEM和基础的RetinaNet在该数据集上的性能。实验结果表明,Focal Loss的使用提高了平均检测精度,难例挖掘技术的应用也提高了检测器对小目标等难例的检测率。Hidden object detection in terahertz imaging is an urgent need for public safety and counter-terrorism.Due to the poor quality of terahertz imaging,object detection on terahertz images is much more difficult than that on public object detection datasets commonly used in computer vision.A multi-object active terahertz imaging dataset is collected.The detection performance of RetinaNet when using cross entropy and Focal Loss as a loss function is compared for the sample imbalance problem.For those samples with poor detection performance,the training model is augmented using the hard example mining technique.Since the traditional hard example mining technique is designed on two-stage object detector and cannot be directly applied to a one-stage detector,a hard example mining scheme is designed based on RetinaNet with image as unit.The experiments also verify the performance of YOLOv3,YOLOv4,FRCN-OHEM and the base RetinaNet on this dataset.The experimental results show that the use of Focal Loss improves the average detection accuracy,and the application of hard example mining technique also improves the detection accuracy of the detector for hard examples such as small objects.

关 键 词：太赫兹图像 太赫兹成像 目标检测 难例挖掘 样本不平衡 RetinaNet 

分 类 号：TP183[自动化与计算机技术—控制理论与控制工程]