【发布时间】：2015-06-30 13:27:01
【问题描述】：
我尝试使用我自己的图像数据库来训练我自己的神经网络，如
中所述http://caffe.berkeleyvision.org/gathered/examples/imagenet.html
但是，当我想在使用 matlab 包装器对一些标准图像进行训练后检查神经网络时，我得到以下输出/错误：
Done with init
Using GPU Mode
Done with set_mode
Elapsed time is 3.215971 seconds.
Error using caffe
Invalid input size
我之前使用 matlab 包装器基于预训练模型提取 cnn 特征。有效。所以我不认为我的图像的输入大小是问题（它们通过函数“prepare_image”在内部转换为正确的大小）。
有谁知道可能是什么错误？
【问题讨论】：
-
请发布 Matlab 代码以重现问题
-
我可以稍后通过 Matlab-Code。但原则上，它只是标准的“matcaffe_demo.m”文件，我只将 2 条路径（1）调整为我自己训练的 caffemodel 文件，（2）调整为相应的 prototxt 文件。我必须承认我没有更改图像均值文件的路径，该文件由“prepare_image”调用。这是因为我自己计算的图像均值是“binaryproto”格式，而不是 Matlab 要求的“.mat”格式。在训练网络时，由于内存问题，我不得不减少“批量大小”。我希望这不会导致问题。
-
matcaffe_demo 的代码在尝试计算图像分数时完全中断（“scores = caffe('forward', input_data);”）。之前的一切似乎都工作正常（由“matcaffe_init（use_gpu，myOwnModel_def_file，myOwnModel_file）”初始化）