本文主要是实现了Matlab Documentation中的一个例程
https://ww2.mathworks.cn/help/vision/ref/trainrcnnobjectdetector.html
看了很多篇有关于深度学习入门的教程,觉得一头雾水,但这个例程很容易跑通,可以让我们这些新手玩家大概知道如何训练一个神经网络,怎样准备一个数据集,整体代码大致是怎么样的,要包括哪些内容。因为我也是新手,下边的内容只是我个人的理解,若有差错,烦请大神们不吝赐教!
一、电脑环境
系统:macOS Mojave 10.14
Matlab版本:2018b(校园版license)
注意这里一定要是正版Matlab才可以安装各种数据包
CPU:Intel Core i5
显卡:Intel HD Graphics 5000
二、代码
Talk is cheap, show your the code.
load('rcnnStopSigns.mat','stopSigns','layers');
imDir = fullfile(matlabroot, 'toolbox', 'vision', 'visiondata',...
'stopSignImages');
addpath(imDir);
options = trainingOptions('sgdm', ...
'MiniBatchSize', 32, ...
'InitialLearnRate', 1e-6, ...
'MaxEpochs', 10);
rcnn = trainRCNNObjectDetector(stopSigns, layers, options,...
'NegativeOverlapRange', [0 0.3]);
img = imread('stopSignTest.jpg');