Skip to content

GitLab

Commit 102611ad authored by Ronny Votel's avatar Ronny Votel Committed by TF Object Detection Team
Browse files

Pass boxes_strided to multi-class centernet postprocessing 

PiperOrigin-RevId: 381366438
parent 6264aa72
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
research/object_detection/meta_architectures/center_net_meta_arch.py
View file @ 102611ad
...@@ -3759,7 +3759,7 @@ class CenterNetMetaArch(model.DetectionModel): ...@@ -3759,7 +3759,7 @@ class CenterNetMetaArch(model.DetectionModel):
]) ])
keypoints, keypoint_scores = self._postprocess_keypoints_multi_class( keypoints, keypoint_scores = self._postprocess_keypoints_multi_class(
prediction_dict, channel_indices, y_indices, x_indices, prediction_dict, channel_indices, y_indices, x_indices,
None, num_detections) boxes_strided, num_detections)
keypoints, keypoint_scores = ( keypoints, keypoint_scores = (
convert_strided_predictions_to_normalized_keypoints( convert_strided_predictions_to_normalized_keypoints(
keypoints, keypoint_scores, self._stride, true_image_shapes, keypoints, keypoint_scores, self._stride, true_image_shapes,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment