3 个月前

MixTeacher:基于多尺度教师模型挖掘有潜力的标签用于半监督目标检测

MixTeacher:基于多尺度教师模型挖掘有潜力的标签用于半监督目标检测

摘要

目标实例间的尺度变化仍是目标检测任务中的一个关键挑战。尽管现代检测模型取得了显著进展,但在半监督学习场景下,这一问题尤为突出。现有半监督目标检测方法通常依赖严格条件从网络预测中筛选高质量的伪标签,然而我们观察到,尺度极端的目标往往具有较低的置信度,导致这些目标缺乏有效的正样本监督。为此,本文提出一种新颖的框架,通过引入混合尺度教师(mixed scale teacher)来提升伪标签生成质量,并实现尺度不变的学习。此外,我们提出一种基于多尺度预测置信度提升的伪标签挖掘策略,该策略充分利用了混合尺度特征带来的更优预测结果。在MS COCO和PASCAL VOC基准数据集上,针对多种半监督设置的大量实验表明,所提方法取得了新的最先进性能。代码与模型已开源,地址为:\url{https://github.com/lliuz/MixTeacher}。

代码仓库

lliuz/mixteacher
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
semi-supervised-object-detection-on-coco-1MixTeacher-FRCNN
mAP: 25.16
semi-supervised-object-detection-on-coco-1MixTeacher-FCOS
mAP: 23.83
semi-supervised-object-detection-on-coco-10MixTeacher-FCOS
detector: FCOS-Res50
mAP: 36.95
semi-supervised-object-detection-on-coco-10MixTeacher-FRCNN
detector: FRCNN-Res50
mAP: 36.72
semi-supervised-object-detection-on-coco-2MixTeacher-FCOS
mAP: 27.88
semi-supervised-object-detection-on-coco-2MixTeacher-FRCNN
mAP: 29.11
semi-supervised-object-detection-on-coco-5MixTeacher-FCOS
mAP: 33.42
semi-supervised-object-detection-on-coco-5MixTeacher-FRCNN
mAP: 34.06

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
MixTeacher:基于多尺度教师模型挖掘有潜力的标签用于半监督目标检测 | 论文 | HyperAI超神经