小目标检测2_OHEM

主要参考:

有参考很多文章与博客,有的可能忘记放出处了,侵权请联系,马上修改~

Online Hard-Example Mining(在线难例挖掘)

RPN网络(两阶段网络中,使用区域提议网络RPN在产生目标区域建议框)中目标可能被很多框标记,这些框中有的包含了整个目标;有的框可能只包含不完全的目标,这类框是错的,可以将其判定为难例。
通过不断被难例训练就可以使网络具有更好的鲁棒性。

通常,每个图像随机选择256个ROI(这个应该是可以指定的),一些作为前景(目标),一些作为背景。

在OHEM方法中,将传统的ROI的随机选择替换为基于其损失值的选择。在排序前要进行 NMS非极大抑制,区域根据其loss损失值进行排序,只有足够高的区域才会传递给分类学习模块。

这确保了对网络错误最多的样本(困难的样本)的学习,分别对背景和前景对象进行选择,以确保在每个梯度下降步骤中有足够数量的正样本和负样本。

网络

在这里插入图片描述
OHEM中有两组ROI Network,其中
(a)是绿色
是一个可读(read-only)的ROI Network,每次只读取权重后进行forward࿰

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值