faster rcnn matlab,Faster rcnn 模型的数据标定的知识求助,matlab2017b自带的函数

%构建CNN网络

%输入层,最小检测对象设置为540*960

inputLayer = imageInputLayer([10 10 3]);

%中间层

%定义卷基层参数

filterSize = [3 3];

numFilters =32;

middleLayers = [

%第一轮,只包含CNN和ReLU

convolution2dLayer(filterSize,numFilters,'Padding',1)

reluLayer()

%第二轮,包含CNN、ReLU和Pooling

convolution2dLayer(filterSize,numFilters,'Padding',1)

reluLayer()

maxPooling2dLayer(3,'Stride',2)

];

%输出层

finalLayers = [

%新增一个包含64个输出的全连接层

fullyConnectedLayer(64)

%新增一个非线性ReLU层

reluLayer()

%新增一个2个输出的全连接层,用于判断图像是否半酣检测对象

fullyConnectedLayer(2)

%添加softmax和classification层

softmaxLayer()

classificationLayer()

];

%组合所有层

layers=[

inputLayer

middleLayers

finalLayers

];

%%训练CNN网络

%将数据划分两部分

%前60%数据用于训练,后面40%用于测试

ind = round(size(data1,1)*0.6);

trainData = data1(1:ind,:);

testData = data1(ind+1:end,:);

%训练过程包括4步,每步可以使用单独的参数,与可以使用同一个参数

options=[

%第1步,Training a Region Proposal Network (RPN)

trainingOptions('sgdm','Maxepochs',10,'InitialLearnRate',1e-5,'CheckpointPath',tempdir)

%第2步,Training a Fast R-CNN Network using the RPN from step1

trainingOptions('sgdm','Maxepochs',10,'InitialLearnRate',1e-5,'CheckpointPath',tempdir)

%第3步,Re-training RPN using weight sharing with Fast R-CNN

trainingOptions('sgdm','Maxepochs',10,'InitialLearnRate',1e-6,'CheckpointPath',tempdir)

%第4步,Re-training Fast R-CNN using updated RPN

trainingOptions('sgdm','Maxepochs',10,'InitialLearnRate',1e-6,'CheckpointPath',tempdir)

];

%设置模型的本地存储

doTrainingAndEval = 1;

if doTrainingAndEval

%训练R-CNN神经网络,神经网络工具箱提供了3个函数

%(1)trainRCNNObjectDetector,训练快且检测满,允许指定proposaFcn

%(2)trainFastRCNNObjectDetector,速度较快,允许指定proposaFcn

%(3)trainFasterRCNNObjectDetector,优化运行性能,不需要指定proposaFcn

detector = trainFasterRCNNObjectDetector(trainData,layers,options,...

'NegativeOverlapRange',[0 0.3],...

'PositiveOverlapRange',[0.6 1],...

'BoxPyramidScale',1.2);

else

%加载已经训练好的神经网络

detector = data.detector;

end

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值