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.