Minor clean up

DeepDeformationMapRegistration/utils/nifty_utils.py

@@ -1,40 +0,0 @@
-import os
-import nibabel as nb
-import numpy as np
-import zipfile
-
-
-TEMP_UNZIP_PATH = '/mnt/EncryptedData1/Users/javier/ext_datasets/LITS17/temp'
-NII_EXTENSION = '.nii'
-
-
-def save_nifti(data, save_path, verbose=True):
-    data_nifti = nb.Nifti1Image(data, affine=np.eye(4))
-
-    data_nifti.header.get_xyzt_units()
-    try:
-        data_nifti.to_filename(save_path)  # Save as NiBabel file
-        if verbose:
-            print('Saved {}'.format(save_path))
-    except ValueError:
-        print('Could not save {}'.format(save_path))
-
-
-def unzip_nii_file(file_path):
-    file_dir, file_name = os.path.split(file_path)
-    file_name = file_name.split('.zip')[0]
-
-    dest_path = os.path.join(TEMP_UNZIP_PATH, file_name)
-    zipfile.ZipFile(file_path).extractall(TEMP_UNZIP_PATH)
-
-    if not os.path.exists(dest_path):
-        print('ERR: File {} not unzip-ed!'.format(file_path))
-        dest_path = None
-    return dest_path
-
-
-def delete_temp(file_path, verbose=False):
-    assert NII_EXTENSION in file_path
-    os.remove(file_path)
-    if verbose:
-        print('Deleted file: ', file_path)

EvaluationScripts/Evaluate_3d_weaklySupervised.py

@@ -14,7 +14,7 @@ from datetime import datetime
 import DeepDeformationMapRegistration.utils.constants as C
 from DeepDeformationMapRegistration.data_generator import DataGeneratorManager
 from DeepDeformationMapRegistration.utils.misc import try_mkdir
-from DeepDeformationMapRegistration.utils.nifty_utils import save_nifti
+from DeepDeformationMapRegistration.utils.nifti_utils import save_nifti
 from DeepDeformationMapRegistration.networks import WeaklySupervised
 from DeepDeformationMapRegistration.losses import HausdorffDistanceErosion
 from DeepDeformationMapRegistration.layers import UncertaintyWeighting