%% Create traning & testing list for the Pascal-S images using 2-fold cross validation

%%

clear all; clc;

 

folder = '/mnt/vana/amirul/datasets/Pascal-S/images/';

image_files=dir([folder '*.jpg']);

 

train_file = fopen('/mnt/vana/amirul/projects/salient_instance_ranking/data/pascal-s/train_cv_1.txt', 'w');

test_file = fopen('/mnt/vana/amirul/projects/salient_instance_ranking/data/pascal-s/test_cv_1.txt', 'w');

 

cv=cvpartition(850,'kfold',2);

idx=training(cv,1);

 

for i = 1 : length(image_files)

   

    img_file = image_files(i).name;

   

    gt_file = [img_file(1:end-4) '.png'];

   

    if idx(i) == 1

        fprintf(train_file, '/mnt/vana/amirul/datasets/Pascal-S/images/%s /mnt/vana/amirul/datasets/Pascal-S/gt_original/%s\n', img_file, gt_file);  

   

    else

        fprintf(test_file, '/mnt/vana/amirul/datasets/Pascal-S/images/%s /mnt/vana/amirul/datasets/Pascal-S/gt_original/%s\n', img_file, gt_file);  

   

    end

   

end

 

fclose(train_file);

fclose(test_file);