首页 文章

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

提问于
浏览
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”在内部转换为正确的大小) .

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

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

相关问题