一个边界框上的多个标签用于Tensorflow SSD Mobilenet

Question

我已经配置了SSD mobilenet v1，并且之前也对模型进行了训练。但是，在我的数据集中，每个边界框都有多个类标签。我的数据集包含面孔，每张面孔都有2个标签：年龄和性别。这两个标签都具有相同的边界框坐标。

在对该数据集进行训练之后，我遇到的问题是该模型仅标记了面部的性别而不是年龄。但是在yolo中，可以显示性别和年龄。

是否可以使用SSD移动网在单个边界框上获得多个标签？

Answer 1

这取决于实现方式，但是SSD使用softmax层来预测每个边界框一个类，而YOLO可以预测每个类的单个S型置信度得分。因此，在SSD中，会选择带有argmax的单个类，但在YOLO中，您可以接受阈值以上的多个类。

但是，您实际上正在使用两种类型的输出来完成多任务学习问题，因此您应该扩展这些模型以共同预测两种类型的类。