OHEM网络

大家好，又见面了，我是你们的朋友全栈君。

该网络就是解决fast r-cnn、sppnet等网络在训练过程中，训练样本不均衡的问题。比如可能前景少，背景多。

网络结构

在这里插入图片描述

这个网络相比于fast r-cnn就增加了红色的部分，同时绿色部分最终计算出来的loss不再是用于反向传播，而是寻找hard negative，下边红色区域计算出来的loss用于反向传播。注意下边红色区域计算loss是利用在绿色区域寻找的hard negative进行计算的。

发布者：全栈程序员栈长，转载请注明出处：https://javaforall.cn/142004.html原文链接：https://javaforall.cn