matconvnet分步注释(三)运行matconvnet

运行matconvnet勒

function [net, info] = cnn_plate()
%初始化神经网络
run('C:\Users\TAT\Desktop\matconvnet-1.0-beta22\matlab\vl_setupnn.m') ;
%datadir='E:\MachineLearning\caffe\caffe-windows-master\platerecognition\data\platerecognition\chars2';
%导入原始数据地址
datadir='C:\Users\TAT\Desktop\matconvnet-1.0-beta22\mycode\plate\ann';
%导入数据的mat格式
opts.expDir = fullfile(vl_rootnn, 'data', 'mycodplate') ;
opts.imdbPath = fullfile(opts.expDir, 'imdb.mat');
%判断mat数据是否存在,不存在则重建
if exist(opts.imdbPath,'file')
    imdb=load(opts.imdbPath);%加载mat数据
else
    imdb=cnn_plate_setup_data(datadir);%重建mat数据
    mkdir(opts.expDir) ;%创建文件夹
    save(opts.imdbPath, '-struct', 'imdb') ;%保存文件
end
%设置网络结构
net=cnn_plate_init();
%载入原始数据
net.meta.normalization.averageImage =imdb.images.data_mean ;
%是否使用gpu加速
opts.train.gpus=1;
%训练网络
[net, info] = cnn_train(net, ...%网络
  imdb, ...%输入数据
  getBatch(opts), ...%批处理
  ...%---参数---
  'expDir', ...
  opts.expDir, ...
  net.meta.trainOpts, ...
  opts.train, ...
  'val',...
  find(imdb.images.set == 3)...
  ...%---------
  );

function fn = getBatch(opts)
% --------------------------------------------------------------------
    fn = @(x,y) getSimpleNNBatch(x,y) ;
end
function [images, labels]  = getSimpleNNBatch(imdb, batch)
    images = imdb.images.data(:,:,:,batch) ;
    labels = imdb.images.labels(1,batch) ;
    if opts.train.gpus > 0
        images = gpuArray(images) ;
    end
end
end


  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

RtZero

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值