Browse Source

업데이트 'step2_gen_trainset.m'

Hio-Been Han 3 years ago
parent
commit
cf46ca3cee
1 changed files with 6 additions and 4 deletions
  1. 6 4
      step2_gen_trainset.m

+ 6 - 4
step2_gen_trainset.m

@@ -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