From ad03905d5310fbaec7a9f4f9c4fa57ba21219493 Mon Sep 17 00:00:00 2001 From: gitttt-1234 Date: Mon, 16 Dec 2024 14:12:51 -0800 Subject: [PATCH] Adjust centroids --- sleap/nn/inference.py | 1 + 1 file changed, 1 insertion(+) diff --git a/sleap/nn/inference.py b/sleap/nn/inference.py index 426c28b36..5b65c825d 100644 --- a/sleap/nn/inference.py +++ b/sleap/nn/inference.py @@ -767,6 +767,7 @@ def call(self, example_gt: Dict[str, tf.Tensor]) -> Dict[str, tf.Tensor]: full_imgs = example_gt["image"] if self.input_scale != 1.0: full_imgs = sleap.nn.data.resizing.resize_image(full_imgs, self.input_scale) + example_gt["centroids"] *= self.input_scale crop_sample_inds = example_gt["centroids"].value_rowids() # (n_peaks,) n_peaks = tf.shape(crop_sample_inds)[0] # total number of peaks in the batch centroid_points = example_gt["centroids"].flat_values # (n_peaks, 2)