From 32e371536ea1b1d1711914938daee85c9028108e Mon Sep 17 00:00:00 2001 From: wyhallenwu Date: Wed, 3 Jan 2024 16:44:33 +0900 Subject: [PATCH] fix the default value of Anchor3DHead to match with the default Anchor3DRangeGenerator --- mmdet3d/models/dense_heads/anchor3d_head.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/mmdet3d/models/dense_heads/anchor3d_head.py b/mmdet3d/models/dense_heads/anchor3d_head.py index 053a8ca46e..580e2f1763 100644 --- a/mmdet3d/models/dense_heads/anchor3d_head.py +++ b/mmdet3d/models/dense_heads/anchor3d_head.py @@ -54,12 +54,13 @@ def __init__(self, use_direction_classifier: bool = True, anchor_generator: ConfigType = dict( type='Anchor3DRangeGenerator', - range=[0, -39.68, -1.78, 69.12, 39.68, -1.78], - strides=[2], + ranges=[[0, -39.68, -1.78, 69.12, 39.68, -1.78]], sizes=[[3.9, 1.6, 1.56]], + scales=[1], rotations=[0, 1.57], - custom_values=[], - reshape_out=False), + custom_values=(), + reshape_out=True, + size_per_range=True), assigner_per_size: bool = False, assign_per_class: bool = False, diff_rad_by_sin: bool = True,