PSROI Pooling输出预测

来源:5-5 HyperNet、RFCN介绍

Super波神

2020-01-09

老师您好,对于PSROI Pooling的输出,再经过一个mean pooling后就不再进行分类和回归计算了吗?怎么就直接得到了它的类别和位置信息呢?

写回答

1回答

会写代码的好厨师

2020-01-31

不是不进行分类和回归计算了。实际上,我们在使用网络进行分类和回归时,使用的层要么是FC层,要么是用卷积层,实现方式都是相同的。那这个层到底是完成分类还是回归,主要看我们如何去对这个层计算loss,比如,预测的结果,我们认为是一个类别的概率分布,用交叉熵来算loss,那这个层就是分类层。如果是对应的坐标值,然后用smoothl1或者l1/l2这样的loss来计算,那就是一个回归层。

在roi pooling之后,在经过卷积或者其他层操作以后,最终输出的几个维度分别是概率分布+坐标信息。在计算loss的时候,对相应的维度分别计算loss,那就能够实现类别的预测和位置信息的回归

1
0

深度学习之目标检测常用算法原理+实践精讲

从原理到场景实战,掌握目标检测核心技术

882 学习 · 221 问题

查看课程