训练中的一个错误

Question

大家，我想用caffe训练一个5类的检测任务，用"SSD:单发MultiBox检测器“，所以我把num_classes从21改为6，但是我得到了以下错误：

检查失败: num_priors_ * num_classes_ ==底部1->通道() (52392比183372)优先级必须与置信度预测数相匹配。

我可以理解这个错误，我发现52392/6= 183372 /21，也就是为什么我将num_classes改为6，但是置信度预测的数量仍然是183372。那么如何解决这个问题呢。非常感谢!

Answer 1

由于SSD不仅取决于分类输出的标签数量，还取决于BB预测，因此您需要更改模型中其他几个地方的num_output。

我强烈建议您不要手工操作，而是使用'examples/ssd'文件夹中提供的python脚本。例如，可以将'examples/ssd/ssd_pascal_speed.py'更改为：

num_classes = 5 # instead of 21

然后使用此脚本提供的模型文件。