前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >Scale Match:国科大提出小目标检测的尺度匹配方法,用于预训练数据集处理

Scale Match:国科大提出小目标检测的尺度匹配方法,用于预训练数据集处理

作者头像
CV君
发布2019-12-27 15:30:55
3K0
发布2019-12-27 15:30:55
举报
文章被收录于专栏:我爱计算机视觉

圣诞快乐~

今天分享一篇新出的论文 Scale Match for Tiny Person Detection,作者贡献了一个细小人物目标检测的数据集 TinyPerson,同时提出一种对预训练数据进行尺度调整的Scale Match(尺度匹配)的方法,显著改进了小目标检测。

该文作者信息:

作者均来自中国科学院大学。

众所周知,小目标检测一直是比较棘手的技术问题,现有目标检测算法往往难以给出理想的结果,但在某些场合,小目标检测却非常有意义。

比如,长距离的视频监控,画面中的人物可能都比较小。

作者构建的 TinyPerson 细小人物目标数据库与其他知名的目标检测数据库的视觉比较:

TinyPerson 样本局部放大图

可见,如果大家不点击查看大图的话,基本上就只能看到 TinyPerson上有一些黑点而已。

下面图表和表格是对几大数据集的尺度的统计:

TinyPerson 中人物平均绝对大小才 18 个像素。能较好的反应长距离视频监控场景中的人物。

TInyPerson数据集统计数据

当目标仅有十几个像素大小的时候,算法如何有效捕捉目标呢?事实上,作者没有发明新的检测算法,而是给算法增加训练样本,并对这些训练样本进行尺度调整,使其在尺度上和目标数据集(TinyPerson)统计属性上相似。

嗯,思路就是这么直接!?

作者将这个尺度调整的过程称之为尺度匹配(Scale Match),请看下图:

Psize(s;E)代表用于增加样本的外部数据集中目标尺度的统计直方图;

Psize(s;Dtrain)则代表目标检测任务数据集中目标尺度的统计直方图;

尺度匹配(Scale Match)的过程就是让这两个直方图分布相似。然后把尺度调整后的外部数据集加入训练集,训练算法模型。

熟悉数字图像处理的朋友,肯定学过直方图匹配(将一幅图像的直方图调整为与另一幅图像直方图相似),尺度匹配与其类似,不同的是,这里直方图每个分箱代表的是不同的尺度的目标的统计。

尺度匹配的算法流程如下:

图像直方图匹配中要保持像素值映射的单调性,目标检测中尺度匹配不需要严格单调(毕竟尺度调整之后该是什么目标还是什么目标),但作者也尝试了保持尺度调整单调性的Monotone Scale Match (MSM),这在某种程度上保留了预训练数据集中目标出现的自然尺度的相对大小一致性。

实验结果

作者实验了一阶段、二阶段、anchor-free 几种有代表性的目标检测算法,以下是在使用MS COCO做预训练然后在TinyPerson上的结果比较:

文中使用 Faster RCNN-FPN 作为比较的基线算法,下表为添加Scale Match、Monotone Scale Match之后在TinyPerson、Tiny Citypersons数据集上的实验结果:

可见,对于训练数据集进行尺度匹配带来了显著的精度提升,AP上升约3-4个点。

论文地址: https://arxiv.org/abs/1912.10664v1

数据代码地址:

https://github.com/ucas-vg/TinyBenchmark

(目前还未开源)

本文参与 腾讯云自媒体同步曝光计划,分享自微信公众号。
原始发表:2019-12-25,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 我爱计算机视觉 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
相关产品与服务
图像识别
腾讯云图像识别基于深度学习等人工智能技术,提供车辆,物体及场景等检测和识别服务, 已上线产品子功能包含车辆识别,商品识别,宠物识别,文件封识别等,更多功能接口敬请期待。
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档