jpdefrutos commited on
Commit
ca1d395
·
1 Parent(s): a7b71d6

Renamed the folder to SoA_methods to hold the evaluation script of other methods

Browse files
{ANTs → SoA_methods}/eval_ants.py RENAMED
@@ -102,8 +102,8 @@ if __name__ == '__main__':
102
  mov_centroids = vol_file['mov_centroids'][:]
103
 
104
  # ndarray to ANTsImage
105
- fix_img_ants = ants.make_image(fix_img.shape[:-1], np.squeeze(fix_img)) # ANTs doesn't work fine with 1-ch images
106
- mov_img_ants = ants.make_image(mov_img.shape[:-1], np.squeeze(mov_img)) # ANTs doesn't work fine with 1-ch images
107
 
108
  t0_syn = time.time()
109
  reg_output_syn = ants.registration(fix_img_ants, mov_img_ants, 'SyN')
@@ -121,13 +121,13 @@ if __name__ == '__main__':
121
  for reg_output in [reg_output_syn, reg_output_syncc]:
122
  mov_to_fix_trf_list = reg_output[FWD_TRFS]
123
  pred_img = reg_output[WARPED_MOV].numpy()
124
- pred_img = pred_img[..., np.newaxis] # ANTs doesn't work fine with 1-ch images
125
 
126
  fix_seg_ants = ants.make_image(fix_seg.shape, np.squeeze(fix_seg))
127
  mov_seg_ants = ants.make_image(mov_seg.shape, np.squeeze(mov_seg))
128
  pred_seg = ants.apply_transforms(fixed=fix_seg_ants, moving=mov_seg_ants,
129
  transformlist=mov_to_fix_trf_list).numpy()
130
- pred_seg = np.squeeze(pred_seg) # ANTs adds an extra axis which shouldn't be there
131
  with sess.as_default():
132
  dice, hd, dice_macro = sess.run([dice_tf, hd_tf, dice_macro_tf],
133
  {'fix_seg:0': fix_seg[np.newaxis, ...], # Batch axis
 
102
  mov_centroids = vol_file['mov_centroids'][:]
103
 
104
  # ndarray to ANTsImage
105
+ fix_img_ants = ants.make_image(fix_img.shape[:-1], np.squeeze(fix_img)) # SoA doesn't work fine with 1-ch images
106
+ mov_img_ants = ants.make_image(mov_img.shape[:-1], np.squeeze(mov_img)) # SoA doesn't work fine with 1-ch images
107
 
108
  t0_syn = time.time()
109
  reg_output_syn = ants.registration(fix_img_ants, mov_img_ants, 'SyN')
 
121
  for reg_output in [reg_output_syn, reg_output_syncc]:
122
  mov_to_fix_trf_list = reg_output[FWD_TRFS]
123
  pred_img = reg_output[WARPED_MOV].numpy()
124
+ pred_img = pred_img[..., np.newaxis] # SoA doesn't work fine with 1-ch images
125
 
126
  fix_seg_ants = ants.make_image(fix_seg.shape, np.squeeze(fix_seg))
127
  mov_seg_ants = ants.make_image(mov_seg.shape, np.squeeze(mov_seg))
128
  pred_seg = ants.apply_transforms(fixed=fix_seg_ants, moving=mov_seg_ants,
129
  transformlist=mov_to_fix_trf_list).numpy()
130
+ pred_seg = np.squeeze(pred_seg) # SoA adds an extra axis which shouldn't be there
131
  with sess.as_default():
132
  dice, hd, dice_macro = sess.run([dice_tf, hd_tf, dice_macro_tf],
133
  {'fix_seg:0': fix_seg[np.newaxis, ...], # Batch axis