caffe.net matlab,windows-matlab环境下,生成的caffe模型无法运行

我的环境如下:windows10,matlab2014a

在配置完MATLAB的接口之后,可以运行MATLAB自带的测试程序:

[scores, maxlabel] = classification_demo(im, use_gpu)

然后我准备了mnist例程的相关文件,如下:

deploy.prototxt和lenet_iter_10000.caffemodel

我试图输入一张图片,然后运行得到分类结果,但运行的结果是matlab卡死,程序自动退出,不知道哪里出了问题,代码如下:

if exist('../+caffe', 'dir')

addpath('..');

else

error('Please run this demo from caffe/matlab/demo');

end

% Set caffe mode

if exist('use_gpu', 'var') && use_gpu

caffe.set_mode_gpu();

gpu_id = 0;  % we will use the first gpu in this demo

caffe.set_device(gpu_id);

else

caffe.set_mode_cpu();

end

% Initialize the network using BVLC CaffeNet for image classification

% Weights (parameter) file needs to be downloaded from Model Zoo.

model_dir = '../../models/mnist/';

net_model = [model_dir 'deploy.prototxt'];

net_weights = [model_dir 'lenet_iter_10000.caffemodel'];

phase = 'test'; % run with phase test (so that dropout isn't applied)

if ~exist(net_weights, 'file')

error('Please download CaffeNet from Model Zoo before you run this demo');

end

% Initialize a network

net = caffe.Net(net_model, net_weights, phase);

每次都是运行到net = caffe.Net(net_model, net_weights, phase);就卡死 了

高手请指教

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值