我正在尝试使用多类lmdb数据训练AlexNet。输入图层是:
层{ 名称:“数据” 类型:“数据” 顶部:“数据” data_param { 来源:“/ home / gal /caffe-1.0/models/mixDCNN/hdf5 / …
你设定 num_output: 8 为您 "fc8_3_color" 图层,也就是说,您试图用8个离散类0..7预测“颜色”。但是,查看数据时,颜色标签的范围是17到24 - 这不起作用。如果你使用的话,标签必须是0..7 "SoftmaxWithLoss" 和 "Accuracy" 层。
num_output: 8
"fc8_3_color"
"SoftmaxWithLoss"
"Accuracy"