upgrade to build_anchor_generator (#941)

* visuliaztion typos

* fix to visualization

* fixed lint errors by pre-commit

* upgrade to build_anchor_generator

* fixed  isort message

mmdet3d/core/anchor/__init__.py

 # Copyright (c) OpenMMLab. All rights reserved.
-from mmdet.core.anchor import build_anchor_generator
+from mmdet.core.anchor import build_prior_generator
 from .anchor_3d_generator import (AlignedAnchor3DRangeGenerator,
                                   AlignedAnchor3DRangeGeneratorPerCls,
                                   Anchor3DRangeGenerator)
 
 __all__ = [
     'AlignedAnchor3DRangeGenerator', 'Anchor3DRangeGenerator',
-    'build_anchor_generator', 'AlignedAnchor3DRangeGeneratorPerCls'
+    'build_prior_generator', 'AlignedAnchor3DRangeGeneratorPerCls'
 ]

mmdet3d/models/dense_heads/anchor3d_head.py

@@ -6,8 +6,8 @@ from torch import nn as nn
 
 from mmdet3d.core import (PseudoSampler, box3d_multiclass_nms, limit_period,
                           xywhr2xyxyr)
-from mmdet.core import (build_anchor_generator, build_assigner,
-                        build_bbox_coder, build_sampler, multi_apply)
+from mmdet.core import (build_assigner, build_bbox_coder,
+                        build_prior_generator, build_sampler, multi_apply)
 from mmdet.models import HEADS
 from ..builder import build_loss
 from .train_mixins import AnchorTrainMixin
@@ -84,7 +84,7 @@ class Anchor3DHead(BaseModule, AnchorTrainMixin):
         self.fp16_enabled = False
 
         # build anchor generator
-        self.anchor_generator = build_anchor_generator(anchor_generator)
+        self.anchor_generator = build_prior_generator(anchor_generator)
         # In 3D detection, the anchor stride is connected with anchor size
         self.num_anchors = self.anchor_generator.num_base_anchors
         # build box coder

mmdet3d/models/detectors/imvoxelnet.py

 # Copyright (c) OpenMMLab. All rights reserved.
 import torch
 
-from mmdet3d.core import bbox3d2result, build_anchor_generator
+from mmdet3d.core import bbox3d2result, build_prior_generator
 from mmdet3d.models.fusion_layers.point_fusion import point_sample
 from mmdet.models import DETECTORS, build_backbone, build_head, build_neck
 from mmdet.models.detectors import BaseDetector
@@ -30,7 +30,7 @@ class ImVoxelNet(BaseDetector):
         bbox_head.update(test_cfg=test_cfg)
         self.bbox_head = build_head(bbox_head)
         self.n_voxels = n_voxels
-        self.anchor_generator = build_anchor_generator(anchor_generator)
+        self.anchor_generator = build_prior_generator(anchor_generator)
         self.train_cfg = train_cfg
         self.test_cfg = test_cfg
 

tests/test_utils/test_anchors.py

@@ -7,7 +7,7 @@ CommandLine:
 """
 import torch
 
-from mmdet3d.core.anchor import build_anchor_generator
+from mmdet3d.core.anchor import build_prior_generator
 
 
 def test_anchor_3d_range_generator():
@@ -26,7 +26,7 @@ def test_anchor_3d_range_generator():
         rotations=[0, 1.57],
         reshape_out=False)
 
-    anchor_generator = build_anchor_generator(anchor_generator_cfg)
+    anchor_generator = build_prior_generator(anchor_generator_cfg)
     repr_str = repr(anchor_generator)
     expected_repr_str = 'Anchor3DRangeGenerator(anchor_range=' \
                         '[[0, -39.68, -0.6, 70.4, 39.68, -0.6], ' \
@@ -65,7 +65,7 @@ def test_aligned_anchor_generator():
         reshape_out=True)
 
     featmap_sizes = [(256, 256), (128, 128), (64, 64)]
-    anchor_generator = build_anchor_generator(anchor_generator_cfg)
+    anchor_generator = build_prior_generator(anchor_generator_cfg)
     assert anchor_generator.num_base_anchors == 8
 
     # check base anchors
@@ -200,7 +200,7 @@ def test_aligned_anchor_generator_per_cls():
         reshape_out=False)
 
     featmap_sizes = [(100, 100), (50, 50)]
-    anchor_generator = build_anchor_generator(anchor_generator_cfg)
+    anchor_generator = build_prior_generator(anchor_generator_cfg)
 
     # check base anchors
     expected_grid_anchors = [[