我尝试使用我自己的图像数据库训练我自己的神经网络,如
中所述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")在内部转换为正确的大小。
有谁知道可能是什么错误?
答案 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"用于培训的文件。