18
18
LGR = logging .getLogger (__name__ )
19
19
20
20
21
- def transform_images (images_df , target , masker , metadata_df = None , out_dir = None ):
21
+ def transform_images (images_df , target , masker , metadata_df = None , out_dir = None , overwrite = False ):
22
22
"""Generate images of a given type from other image types and write out to files.
23
23
24
24
Parameters
@@ -37,6 +37,8 @@ def transform_images(images_df, target, masker, metadata_df=None, out_dir=None):
37
37
out_dir : :obj:`str` or :obj:`None`, optional
38
38
Path to output directory. If None, use folder containing first image
39
39
for each study in ``images_df``.
40
+ overwrite : :obj:`bool`, optional
41
+ Whether to overwrite existing files or not. Default is False.
40
42
41
43
Returns
42
44
-------
@@ -84,7 +86,11 @@ def transform_images(images_df, target, masker, metadata_df=None, out_dir=None):
84
86
# Get converted data
85
87
img = resolve_transforms (target , available_data , new_masker )
86
88
if img is not None :
87
- img .to_filename (new_file )
89
+ if overwrite or not op .isfile (new_file ):
90
+ img .to_filename (new_file )
91
+ else :
92
+ LGR .debug ("Image already exists. Not overwriting." )
93
+
88
94
images_df .loc [images_df ["id" ] == id_ , target ] = new_file
89
95
else :
90
96
images_df .loc [images_df ["id" ] == id_ , target ] = None
0 commit comments