Несоответствие индекса для пакетной обработки изображений - PullRequest
Новая
       30

Несоответствие индекса для пакетной обработки изображений

1 голос
/ 10 ноября 2019

Для вычисления Precision Recall набора изображений я получил программу . но я столкнулся с некоторыми проблемами при настройке. Так как в файлах программы нет изображения, я не смог реализовать исходный шаблон именования. В моем случае изображения в Method_Path и GT_Path имеют имена от 1 до N (например, 500). 

%% initialization
clear all
close all;clc;
method = 'mine'; % name of the salient object method you want to evaluate, you need to change it
dataset = 'Pascal'; % name of dataset, you need to change this
method_path='./AMC_AE_Pascal/';
GT_path='./Pascal_GT/';
resultpath = [method_path '*' '.png']; % path to saliency maps, you need to change this
truthpath = [GT_path '*' '.png']; % path to ground-truth masks, yoiu need to change this
savepath = './result/PRcurve/'; % save path of the 256 combinations of precision-recall values
if ~exist(savepath,'dir')
mkdir(savepath);
end
dir_im = dir(resultpath);
assert(~isempty(dir_im),'No saliency map found, please check the path!');
dir_tr= dir(truthpath);
assert(~isempty(dir_tr),'No ground-truth image found, please check the path!');
assert(length(dir_im)==length(dir_tr),'The number of saliency maps and ground-truth images are not equal!')
imNum = length(dir_tr);
precision = zeros(256,1);
recall = zeros(256,1);
%% compute pr curve
for i = 1:imNum
imName = dir_tr(i).name;
input_im = imread([imName(1:end-4),resultpath(end-3:end)]);
truth_im = imread([truthpath(1:end-4),imName]);
truth_im = truth_im(:,:,1);
input_im = input_im(:,:,1);
if max(max(truth_im))==255
    truth_im = truth_im./255;
end
for threshold = 0:255
index1 = (input_im>=threshold);
truePositive = length(find(index1 & truth_im));
groundTruth = length(find(truth_im));
detected = length(find(index1));
if truePositive~=0
 precision(threshold+1) = precision(threshold+1)+truePositive/detected;
 recall(threshold+1) = recall(threshold+1)+truePositive/groundTruth;
end
end
display(num2str(i));
end
precision = precision./imNum;
recall = recall./imNum;
pr = [precision'; recall'];
fid = fopen([savepath dataset, '_', method, '_PRCurve.txt'],'at');
fprintf(fid,'%f %f\n',pr);
fclose(fid);
disp('Done!');

При выполнении кода я получаю следующие ошибки:

1 Ответ

2 голосов
/ 10 ноября 2019

Использование fullfile лучше, чем объединять части имени файла.

Использование кода, подобного [imName(1:end-4),resultpath(end-3:end)], не является надежным, поскольку предполагает слишком много длины расширений.

Я внес несколько изменений в ваш код: 

%% initialization
clear all
close all;clc;
method = 'mine'; % name of the salient object method you want to evaluate, you need to change it
dataset = 'Pascal'; % name of dataset, you need to change this
method_path='./AMC_AE_Pascal/';
GT_path='./Pascal_GT/';
im_pattern = '*.png'; %Image pattern (all PNG file in the folder).
resultpath = method_path;%resultpath = [method_path '*' '.png']; % path to saliency maps, you need to change this
truthpath = GT_path;%truthpath = [GT_path '*' '.png']; % path to ground-truth masks, yoiu need to change this
savepath = './result/PRcurve/'; % save path of the 256 combinations of precision-recall values
if ~exist(savepath,'dir')
    mkdir(savepath);
end
dir_im = dir(fullfile(resultpath, im_pattern));%dir_im = dir(resultpath);
assert(~isempty(dir_im),'No saliency map found, please check the path!');
dir_tr= dir(fullfile(truthpath, im_pattern));%dir_tr= dir(truthpath);
assert(~isempty(dir_tr),'No ground-truth image found, please check the path!');
assert(length(dir_im)==length(dir_tr),'The number of saliency maps and ground-truth images are not equal!')
imNum = length(dir_tr);
precision = zeros(256,1);
recall = zeros(256,1);
%% compute pr curve
for i = 1:imNum
    imName = dir_tr(i).name;
    input_im = imread(fullfile(resultpath, imName));%input_im = imread([imName(1:end-4),resultpath(end-3:end)]);
    truth_im = imread(fullfile(truthpath, imName));%truth_im = imread([truthpath(1:end-5),imName]);%truth_im = imread([truthpath(1:end-4),imName]);
    truth_im = truth_im(:,:,1);
    input_im = input_im(:,:,1);
    if max(max(truth_im))==255
        truth_im = truth_im./255;
    end
    for threshold = 0:255
        index1 = (input_im>=threshold);
        truePositive = length(find(index1 & truth_im));
        groundTruth = length(find(truth_im));
        detected = length(find(index1));
        if truePositive~=0
            precision(threshold+1) = precision(threshold+1)+truePositive/detected;
            recall(threshold+1) = recall(threshold+1)+truePositive/groundTruth;
        end
    end
    display(num2str(i));
end
precision = precision./imNum;
recall = recall./imNum;
pr = [precision'; recall'];
fid = fopen([savepath dataset, '_', method, '_PRCurve.txt'],'at');
fprintf(fid,'%f %f\n',pr);
fclose(fid);
disp('Done!');

Если вы столкнулись с новыми проблемами, вы можете просто использовать отладчик ...

Я не вижу чувствительности к имени файла.
Я попробовал следующий список файлов PNG: 

1.png
12345.png
12346.png
2.png
3 - Copy (2).png
3 - Copy (3).png
3 - Copy (4).png
3 - Copy (5).png
3 - Copy (6).png
3 - Copy (7).png
3 - Copy (8).png
3 - Copy (9).png
3 - Copy.png
3.png
...