将回归层添加到caffe

Question

我实现了一个基于深度学习的微笑检测系统。最底层是系统的输出，根据人的微笑程度有10个输出。

我想用一个回归层将这10个输出转换为1到10范围内的数字输出。

我怎样才能在caffe中做到这一点？

谢谢

Answer 1

为了将您拥有的10个输出转换为单个输出，您需要一个包含10个输入和单个输出的"InnerProduct"层。要训练这一层，您还需要在已有的10个输出层之上添加一个适合回归的损失层。

这种损耗层可以是例如Euclidean loss layer或Ross Girshick的smooth L1 loss layer。