1

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

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

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

4

1 回答 1

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”文件。

于 2015-04-23T13:25:51.700 回答