www.gusucode.com > IPCV_Eval_Kit_R2019a_0ce6858工具箱matlab程序源码 > IPCV_Eval_Kit_R2019a_0ce6858/code/demo_files/I5_06_2_5_myTrainFasterRCNNObjectDetector.m

    %% Faster R-CNN偵傛傞暔懱専弌婍偺妛廗

%% 弶婜壔
clear; close all force; clc; rng('default');

%% 妛廗僨乕僞偺撉傒崬傒
data = load('fasterRCNNVehicleTrainingData.mat');
trainingData = data.vehicleTrainingData;
trainingData.imageFilename = fullfile(toolboxdir('vision'),'visiondata', ...
    trainingData.imageFilename);
layers = data.layers
analyzeNetwork(layers);

%% 妛廗僆僾僔儑儞偺巜掕
options = trainingOptions('sgdm', ...
    'MiniBatchSize', 1, ...
    'InitialLearnRate', 1e-3, ...
    'MaxEpochs', 5, ...
    'VerboseFrequency', 200);

%% Faster R-CNN暔懱専弌婍傪妛廗
% GPU偺巊梡悇彠
detector = trainFasterRCNNObjectDetector(trainingData, layers, options)

%% 妛廗偟偨Faster R-CNN暔懱専弌婍傪僥僗僩
img = imread('highway.png');
[bbox, score, label] = detect(detector, img);

%% 寢壥傪昞帵
detectedImg = insertShape(img, 'Rectangle', bbox);
figure
imshow(detectedImg)

%%
% Copyright 2018 The MathWorks, Inc.