Caffe vgg-SSD训练错误【Check failed: 0 == bottom[0]->count() % explicit_count (0 vs. 8) bottom count (74088) must be 】
问题描述:
VGG-SSD训练自己的数据时出现如下错误:
解决方案:
检查test.prototxt中mbox_conf_reshape层
layer {
name: "mbox_conf_reshape"
type: "Reshape"
bottom: "mbox_conf"
top: "mbox_conf_reshape"
reshape_param {
shape {
dim: 0
dim: -1
dim: 3 #####该处的值等于num_classes,若超过类别数,则标题类似的错误。
}
}
}