Browse Source

Update 'automatic_segmentation_program/preprocess.py'

Frederik Filip Stæger 4 years ago
parent
commit
72da9199da
1 changed files with 3 additions and 3 deletions
  1. 3 3
      automatic_segmentation_program/preprocess.py

+ 3 - 3
automatic_segmentation_program/preprocess.py

@@ -30,6 +30,7 @@ def image_data_to_nii(pixdim, image_data, shrink, out_dir, file_path, save_unsca
     scale = 1/shrink
     new_dim = (round(image_dim[1] * scale), round(image_dim[0] * scale))
     new_arr = np.ndarray(new_dim + (image_data.shape[2],))
+    print(f'Pixel dimensions: {pixdim} um')
 
     for i in range(0,image_data.shape[2]):
         cur_channel = image_data[:, :, i]
@@ -37,7 +38,6 @@ def image_data_to_nii(pixdim, image_data, shrink, out_dir, file_path, save_unsca
         new_arr[:, :, i] = resized
     path_scaled, path_unscaled = get_out_paths(out_dir, Path(file_path).stem)
     nii_scaled = nib.Nifti1Image(new_arr, np.eye(4))
-    nii_scaled.header['xyzt_units'] = 3
     nii_scaled.header['pixdim'][1:3] = pixdim * shrink, pixdim * shrink
     nib.save(nii_scaled, str(path_scaled))
 
@@ -56,14 +56,14 @@ def tiff_to_nii(tif_path, out_dir, pixdim=None, shrink=10):
     tif_header = dict(tif_image.tag)
     output = np.empty(np.array(tif_image).shape + (0,))
     if not pixdim:
-        pixdim = 10e6/tif_header[282][0][0]
+        pixdim = 10e2/tif_header[282][0][0]
     for i, page in enumerate(ImageSequence.Iterator(tif_image)):
         page_data = np.expand_dims(np.array(page), 2)
         output = np.concatenate((output, page_data), 2)
     return image_data_to_nii(pixdim, output, shrink, out_dir, tif_path)
 
 
-def split_nii_channels(nii_path, out_dir=None, flip=False, mask_index=-1, bias=False):
+def split_nii_channels(nii_path, out_dir=None, flip=False, mask_index=-1, bias=True):
     if out_dir is None:
         out_dir = nii_path.parent
     nii = nib.load(str(nii_path))