|
@@ -1,6 +1,4 @@
|
|
|
-
|
|
|
-
|
|
|
-%% (1) Randomly-select partial dataset
|
|
|
+%% (1) Select partial dataset randomly
|
|
|
% Getting full frame list
|
|
|
frame_directory = 'data/frames/';
|
|
|
video_names = dir( [frame_directory 'sample*' ]);
|
|
@@ -20,7 +18,7 @@ end
|
|
|
trainset_directory = 'data/trainset/';
|
|
|
if ~isdir(trainset_directory), mkdir(trainset_directory); end
|
|
|
n_all_image = length(full_frame_names);
|
|
|
-n_train_image = 101;
|
|
|
+n_train_image = 100;
|
|
|
rng(abs(010-6207-8179)) % Fix random seed
|
|
|
target_frame_number = randi( n_all_image, [1 n_train_image] );
|
|
|
for frameIdx = 1:n_train_image
|
|
@@ -63,7 +61,11 @@ for frameIdx = 1:length(img_list)
|
|
|
imagesc( mask ); colormap gray;
|
|
|
title('Result mask (binary)');
|
|
|
drawnow;
|
|
|
+
|
|
|
+ % Write result
|
|
|
imwrite( mask, [mask_directory mask_fname]);
|
|
|
+
|
|
|
+ pause(.5);
|
|
|
end
|
|
|
end
|
|
|
%open mask
|