提交 9cf24bdd 编写于 作者: A A. Unique TensorFlower 提交者: TF Object Detection Team

Fix assign_center_targets_from_keypoints parameter error.

PiperOrigin-RevId: 552854244
上级 bae839cf
......@@ -3069,8 +3069,7 @@ class CenterNetMetaArch(model.DetectionModel):
width=input_width,
gt_classes_list=gt_classes_list,
gt_keypoints_list=gt_keypoints_list,
gt_weights_list=gt_weights_list,
maximum_normalized_coordinate=maximum_normalized_coordinate)
gt_weights_list=gt_weights_list)
else:
gt_boxes_list = self.groundtruth_lists(fields.BoxListFields.boxes)
heatmap_targets = assigner.assign_center_targets_from_boxes(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册