使用caffe时出错输入大小无效

时间:2015-04-22 17:43:17

标签: matlab caffe matcaffe

我尝试使用我自己的图像数据库训练我自己的神经网络,如

中所述

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")在内部转换为正确的大小。

有谁知道可能是什么错误?

1 个答案:

答案 0 :(得分:2)

找到解决方案:我引用了错误的" .prototxt"文件(它有点令人困惑,因为文件非常相似。 因此,对于使用matlab包装器计算功能,需要将以下内容引用到" matcaffe_demo.m"中的文件:

models/bvlc_reference_caffenet/deploy.prototxt
models/bvlc_reference_caffenet/MyModel_caffenet_train_iter_450000.caffemodel

其中" MyModel_caffenet_train_iter_450000.caffemodel"是培训期间创建的唯一文件。

一开始我无意中引用了

models/bvlc_reference_caffenet/MyModel_train_val.prototxt

是" .prototxt"用于培训的文件。