🏗️ Extend package to 3D images and heatmaps.

predict-idlab · Jun 13, 2024 · fac09b8 · fac09b8
1 parent fcbc211
commit fac09b8
Show file tree

Hide file tree

Showing 12 changed files with 2,195 additions and 782 deletions.
diff --git a/src/landmarker/data/landmark_dataset.py b/src/landmarker/data/landmark_dataset.py
diff --git a/src/landmarker/heatmap/__init__.py b/src/landmarker/heatmap/__init__.py
@@ -1,26 +1,26 @@
 """
 Heatmap module.
 """
-from .generator import LaplacianHeatmapGenerator, GaussianHeatmapGenerator
+
 from .decoder import (
     coord_argmax,
+    coord_cov_from_gaussian_ls,
+    coord_cov_windowed_weigthed_sample_cov,
     coord_local_soft_argmax,
-    coord_weighted_spatial_mean,
     coord_soft_argmax,
-    heatmap_to_coord,
-    heatmap_to_coord_enlarge,
-    coord_soft_argmax_cov,
+    coord_weighted_spatial_mean,
     coord_weighted_spatial_mean_cov,
-    heatmap_to_coord_cov,
-    coord_cov_from_gaussian_ls,
     cov_from_gaussian_ls,
     heatmap_coord_to_weighted_sample_cov,
-    weighted_sample_cov,
-    coord_cov_windowed_weigthed_sample_cov,
+    heatmap_to_coord,
+    heatmap_to_coord_cov,
+    heatmap_to_coord_enlarge,
     heatmap_to_multiple_coord,
     non_maximum_surpression,
     non_maximum_surpression_local_soft_argmax,
+    weighted_sample_cov,
 )
+from .generator import GaussianHeatmapGenerator, LaplacianHeatmapGenerator
 
 __all__ = [
     "LaplacianHeatmapGenerator",