Fix topdown multi-class model

talmolab · Dec 16, 2024 · 058a861 · 058a861
1 parent a003f29
commit 058a861
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 0 deletions.
diff --git a/sleap/nn/data/pipelines.py b/sleap/nn/data/pipelines.py
@@ -1251,6 +1251,7 @@ def make_viz_pipeline(self, data_provider: Provider) -> Pipeline:
                 provider=data_provider,
             )
         pipeline += Normalizer.from_config(self.data_config.preprocessing)
+        pipeline += Resizer.from_config(self.data_config.preprocessing)
         pipeline += InstanceCentroidFinder.from_config(
             self.data_config.instance_cropping,
             skeletons=self.data_config.labels.skeletons,

diff --git a/sleap/nn/inference.py b/sleap/nn/inference.py
@@ -4280,6 +4280,7 @@ def _initialize_inference_model(self):
             refinement="integral" if self.integral_refinement else "local",
             integral_patch_size=self.integral_patch_size,
             return_confmaps=False,
+            resize_input_image=False,
         )
         if use_gt_centroid:
             centroid_crop_layer.input_scale = cfg.data.preprocessing.input_scaling

diff --git a/sleap/nn/training.py b/sleap/nn/training.py
@@ -1319,6 +1319,7 @@ def _setup_visualization(self):
             peak_threshold=0.2,
             refinement="local",
             return_confmaps=True,
+            resize_input_image=False,
         )
 
         def visualize_example(example):
@@ -1759,6 +1760,7 @@ def _setup_visualization(self):
             peak_threshold=0.2,
             refinement="local",
             return_confmaps=True,
+            resize_input_image=False,
         )
 
         def visualize_example(example):