您的位置:首页 > 编程语言 > MATLAB

用caffe训练完模型之后调用matlab接口直接进行测试,不用重新编写模型程序

2018-03-28 19:54 821 查看
    当我们用caffe训练完自己的模型之后,可以直接读取训练完的模型,调用matlab接口进行网络的前向传播,不用重新用matlab或者matconvnet编写测试程序。

第一步:模型和mat文件
weights= '1.caffemodel'
model   = '1_mat.prototxt'
第二步:提取模型参数和读取网络结构
net = caffe.Net(model, weights, 'test')

第三步:前向传播
res = net.forward({data})

outputdata = res{1}

这样就可以直接得到输出,而且可以gpu加速,不用再用matcovnet进行测试程序的编写。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐