From fd793bab5a7c52987a2f4df3a3573d9003abb4cc Mon Sep 17 00:00:00 2001 From: "yechenzhi@kuaishou.com" <136920488@qq.com> Date: Sat, 25 Mar 2023 19:11:22 +0800 Subject: [PATCH] centerpoint coordinates correspondence of gt box length and width --- mmdet3d/models/dense_heads/centerpoint_head.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/mmdet3d/models/dense_heads/centerpoint_head.py b/mmdet3d/models/dense_heads/centerpoint_head.py index ad58e5876..72f8f747d 100644 --- a/mmdet3d/models/dense_heads/centerpoint_head.py +++ b/mmdet3d/models/dense_heads/centerpoint_head.py @@ -519,16 +519,16 @@ def get_targets_single(self, for k in range(num_objs): cls_id = task_classes[idx][k] - 1 - width = task_boxes[idx][k][3] - length = task_boxes[idx][k][4] - width = width / voxel_size[0] / self.train_cfg[ + length = task_boxes[idx][k][3] + width = task_boxes[idx][k][4] + length = length / voxel_size[0] / self.train_cfg[ 'out_size_factor'] - length = length / voxel_size[1] / self.train_cfg[ + width = width / voxel_size[1] / self.train_cfg[ 'out_size_factor'] if width > 0 and length > 0: radius = gaussian_radius( - (length, width), + (width, length), min_overlap=self.train_cfg['gaussian_overlap']) radius = max(self.train_cfg['min_radius'], int(radius))