correct scripts

main
v-h 1 year ago
parent ea7f34f023
commit d2b5aeebf5

@ -147,7 +147,7 @@ options = trainingOptions('sgdm',...
'CheckpointPath',tempdir,... 'CheckpointPath',tempdir,...
'ValidationData',validationData); 'ValidationData',validationData);
netname = "netDetectorResNet50.mat" netname = "netDetectorResNet50.mat";
if doAugmentation if doAugmentation
netname = "netDetectorResNet50_2.mat" netname = "netDetectorResNet50_2.mat"
@ -164,7 +164,7 @@ if doTraining
save netname detector; save netname detector;
else else
% Load pretrained detector for the example. % Load pretrained detector for the example.
load netname detector; load (netname, detector);
end end
% ----- quick check/test % ----- quick check/test

@ -39,7 +39,7 @@
close all; close all;
clear; clear;
doTraining = true; doTraining = false;
% first we need the data... % first we need the data...
dataDir = 'Picturedata'; % Destination-Folder for provided (img) Data dataDir = 'Picturedata'; % Destination-Folder for provided (img) Data
@ -141,7 +141,7 @@ options = trainingOptions('sgdm',...
'CheckpointPath',tempdir,... 'CheckpointPath',tempdir,...
'ValidationData',validationData); 'ValidationData',validationData);
netname = "netDetectorResNet50_stepthree.mat" netname = "netDetectorResNet50_stepthree.mat";
@ -155,7 +155,7 @@ if doTraining
save netname detector; save netname detector;
else else
% Load pretrained detector for the example. % Load pretrained detector for the example.
load netname detector; load (netname, 'detector');
end end
% ----- quick check/test % ----- quick check/test

Loading…
Cancel
Save