%% 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);