Added COMET zoom factors

COMET/Evaluate_network.py

@@ -216,9 +216,9 @@ if __name__ == '__main__':
                 # mov_centroids_isotropic = mov_centroids * upsample_scale
                 pred_centroids_isotropic = pred_centroids * upsample_scale
 
-                fix_centroids_isotropic = np.divide(fix_centroids_isotropic, C.IXI_DATASET_iso_to_cubic_scales)
-                # mov_centroids_isotropic = np.divide(mov_centroids_isotropic, C.IXI_DATASET_iso_to_cubic_scales)
-                pred_centroids_isotropic = np.divide(pred_centroids_isotropic, C.IXI_DATASET_iso_to_cubic_scales)
+                fix_centroids_isotropic = np.divide(fix_centroids_isotropic, C.COMET_DATASET_iso_to_cubic_scales)
+                # mov_centroids_isotropic = np.divide(mov_centroids_isotropic, C.COMET_DATASET_iso_to_cubic_scales)
+                pred_centroids_isotropic = np.divide(pred_centroids_isotropic, C.COMET_DATASET_iso_to_cubic_scales)
                 # Now we can measure the TRE in mm
                 tre_array = target_registration_error(fix_centroids_isotropic, pred_centroids_isotropic, False).eval()
                 tre = np.mean([v for v in tre_array if not np.isnan(v)])

DeepDeformationMapRegistration/utils/constants.py

@@ -509,6 +509,8 @@ REG_MANUAL_W = [1.] * len(REG_PRIOR_W)
 # .../T1/training/zoom_factors.csv contain the scale factors of all the training samples from isotropic to 128x128x128
 #   The augmentation values will be scaled using the average+std
 IXI_DATASET_iso_to_cubic_scales = np.asarray([0.655491 + 0.039223, 0.496783 + 0.029349, 0.499691 + 0.028155])
+# ...OSLO_COMET_CT/Formatted_128x128x128/zoom_factors.csv contain the scale factors of all the training samples from isotropic to 128x128x128
+COMET_DATASET_iso_to_cubic_scales = np.asarray([0.455259 + 0.048027, 0.492012 + 0.044298, 0.577552 + 0.051708])
 MAX_AUG_DISP_ISOT = 30
 MAX_AUG_DEF_ISOT = 6
 MAX_AUG_DISP = np.max(MAX_AUG_DISP_ISOT * IXI_DATASET_iso_to_cubic_scales)  # Scaled displacements