bugfix

tlpss · Oct 10, 2023 · 2acddc5 · 2acddc5
1 parent 9ffa9a2
commit 2acddc5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/keypoint_detection/models/detector.py b/keypoint_detection/models/detector.py
@@ -315,7 +315,7 @@ def visualize_predicted_keypoints(self, result_dict):
         # get the keypoints from the heatmaps
         predicted_heatmaps = predicted_heatmaps.detach().float()
         predicted_keypoints = get_keypoints_from_heatmap_batch_maxpool(
-            predicted_heatmaps, self.max_keypoints, self.minimal_keypoint_pixel_distance, abs_max_threshold=0.4
+            predicted_heatmaps, self.max_keypoints, self.minimal_keypoint_pixel_distance, abs_max_threshold=0.2
         )
         # overlay the images with the keypoints
         grid = visualize_predicted_keypoints(images, predicted_keypoints, self.keypoint_channel_configuration)
@@ -355,7 +355,7 @@ def test_step(self, test_batch, batch_idx):
             self.log_channel_predictions_grids(image_grids, mode="test")
 
             keypoint_grids = self.visualize_predicted_keypoints(result_dict)
-            self.log_predicted_keypoints(keypoint_grids, mode="validation")
+            self.log_predicted_keypoints(keypoint_grids, mode="test")
 
         self.log("test/epoch_loss", result_dict["loss"])
         self.log("test/gt_loss", result_dict["gt_loss"])