
摘要
迄今为止,性能最强的半监督目标检测器(SS-OD)均基于伪框(pseudo-boxes)方法,但这类方法需经过一系列后处理操作,且依赖于经过微调的超参数。本文提出用密集预测(dense prediction)作为统一而直接的伪标签形式,替代稀疏的伪框。与伪框相比,所提出的密集伪标签(Dense Pseudo-Label, DPL)无需任何后处理步骤,因而能够保留更丰富的信息。此外,我们引入了一种区域选择技术,以突出关键信息并抑制密集标签中携带的噪声。基于DPL构建的半监督目标检测算法被命名为Dense Teacher。在COCO和VOC数据集上,Dense Teacher在多种实验设置下均展现出优于基于伪框方法的性能。
代码仓库
megvii-basedetection/denseteacher
官方
pytorch
GitHub 中提及
基准测试
| 基准 | 方法 | 指标 |
|---|---|---|
| semi-supervised-object-detection-on-coco-10 | Dense Teacher | detector: FCOS-Res50 mAP: 37.13 |
| semi-supervised-object-detection-on-coco-100 | Dense Teacher | mAP: 46.2 |
| semi-supervised-object-detection-on-coco-5 | Dense Teacher | mAP: 33.01 |