Pass boxes_strided to multi-class centernet postprocessing

PiperOrigin-RevId: 381366438

Pass boxes_strided to multi-class centernet postprocessing
PiperOrigin-RevId: 381366438
102611ad · Ronny Votel · TF Object Detection Team · 6264aa72 · 102611ad
Commit 102611ad authored Jun 24, 2021 by Ronny Votel Committed by TF Object Detection Team Jun 24, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

research/object_detection/meta_architectures/center_net_meta_arch.py ...ject_detection/meta_architectures/center_net_meta_arch.py +1 -1

No files found.
--- a/research/object_detection/meta_architectures/center_net_meta_arch.py
+++ b/research/object_detection/meta_architectures/center_net_meta_arch.py
@@ -3759,7 +3759,7 @@ class CenterNetMetaArch(model.DetectionModel):
        ])
        keypoints, keypoint_scores = self._postprocess_keypoints_multi_class(
            prediction_dict, channel_indices, y_indices, x_indices,
-            None, num_detections)
+            boxes_strided, num_detections)
        keypoints, keypoint_scores = (
            convert_strided_predictions_to_normalized_keypoints(
                keypoints, keypoint_scores, self._stride, true_image_shapes,