split set for training data only

Author: solivr

exps/cbad/make_cbad.py

@@ -39,10 +39,11 @@ def generate_cbad_dataset(downloading_dir: str, masks_dir: str):
                            draw_baselines=True,
                            draw_endpoints=False)
 
-    print('Make eval set from the given training data (0.15/0.85 eval/train)')
-    csv_filename = os.path.join(output_dir, 'set_data.csv')
-    split_set_for_eval(csv_filename)
-    print('Done!')
+        if 'train' in output_dir:
+            print('Make eval set from the given training data (0.15/0.85 eval/train)')
+            csv_filename = os.path.join(output_dir, 'set_data.csv')
+            split_set_for_eval(csv_filename)
+            print('Done!')
 
 
 if __name__ == '__main__':