Error: Out of memory

Question

Monika Zikmundová 2021-6-6

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/849205-error-out-of-memory

回答： Srivardhan Gadila 2021-6-9

Hello,

I am writing a faster r-cnn network script to search for craters. When you run the script, matlab generates an out-of-memory error message. Can someone please advise me how to solve this problem?

My matlab code:

clear all
close all
clc
format long g
data=load('matlab_V21_ROI_11_3_2021.mat')
gTruth=data.gTruth;
gTruth.DataSource
gTruth.LabelDefinitions;
replaceimds=imageDatastore('C:\monika\vse\kratery')
replaceDataSource=groundTruthDataSource(replaceimds)
gTruth.DataSource=replaceDataSource
%objectdetector
trainingDataTable=objectDetectorTrainingData(gTruth);
% [imds,blds]=objectDetectorTrainingData(gTruth);
% cds=combine(imds,blds)
% read(cds)
% counts=countEachLabel(cds)
filenamesImages=trainingDataTable.imageFilename
tblBoxes=trainingDataTable(:,'krater')
imds=imageDatastore(filenamesImages, 'LabelSource','foldernames')
blds=boxLabelDatastore(tblBoxes)
cds=combine(imds,blds)
%prevod na RGB
numberofImages=length(imds.Files)
for k=1:numberofImages
    inputFileName=imds.Files{k};
    fprintf('Checking %s\n', inputFileName);
    grayImage=imread(inputFileName);
    [rows,columns,numberofColorChannels]=size(grayImage);
    if numberofColorChannels == 1
        rgbImage=cat(3,grayImage, grayImage, grayImage);
        imshow(rgbImage);
    else
        imshow(rgbImage);
    end
    axis('on','image');
    %pause(0.1);
end
% anchor boxes I
network = resnet50;%squeezenet; % resnet50
inputImageSize =[224 224 3]; %network.Layers(1).InputSize;
% image augmenter
imageAugmenter=imageDataAugmenter(...
   'RandRotation',[0 90],...
   'RandScale',[0.5 1]);
augmentedImdsTrain=augmentedImageDatastore(inputImageSize,imds,'DataAugmentation',imageAugmenter,'OutputSizeMode','randcrop');  % ořiznutí z náhodného umístění ve vstupním obrázku
%preprocessedTrainingData=transform(cds,@(data)preprocessData(data,inputImageSize));
numClasses = 1;
featureLayer = 'activation_40_relu';%'fire5-concat' %'activation_40_relu';
%estimate anchor boxes
numAnchors = 3
[anchorBoxes,meanIoU] = estimateAnchorBoxes(cds,numAnchors);
anchorBoxes
meanIoU
maxNumAnchors = 3;
meanIoU = zeros([maxNumAnchors,1]);
anchorBoxes = cell(maxNumAnchors, 1);
for k = 1:maxNumAnchors
    % Estimate anchors and mean IoU.
    [anchorBoxes{k},meanIoU(k)] = estimateAnchorBoxes(cds,k);    
end
figure
plot(1:maxNumAnchors,meanIoU,'-o')
ylabel("Mean IoU")
xlabel("Number of Anchors")
title("Number of Anchors vs. Mean IoU")
% anchor boxes - pokracovani
anchorBoxes = [72,70; 99,96; 53,51];
lgraph = fasterRCNNLayers(inputImageSize,numClasses,anchorBoxes, ...
                          network,featureLayer);
analyzeNetwork(lgraph);
options = trainingOptions ( 'sgdm' , ... 
      'MiniBatchSize' , 1, ... 
      'InitialLearnRate' , 1e-3, ... 
      'MaxEpochs' , 5);
detector = trainFasterRCNNObjectDetector (cds, lgraph, options, ... 
        'NegativeOverlapRange' , [0 0.3], ... 
        'PositiveOverlapRange' , [0.6 1]);

0 个评论
显示 -2更早的评论隐藏 -2更早的评论

请先登录，再进行评论。

请先登录，再回答此问题。

Answer 1

Srivardhan Gadila 2021-6-9

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/849205-error-out-of-memory#answer_720940

If your network is being trained on GPU then it seems that the gpu memory is not sufficient and you can set the 'ExecutionEnvironment' to 'cpu' in the trainingOptions and try training the network. Alternatively you can also check the Tips section of the trainFasterRCNNObjectDetector about the possible workarounds suggested for "out-of-memory" issues.

You can refer to the following MATLAB Answer as well: Why do I run into 'Out of memory' error using Faster R-CNN method?

If neither of the above suggestions work then you can contact Mathworks Support.