[Enhance] Update Registry in MMDet3D (#1412)

* Update Registry in MMDet3D

* fix compose pipeline bug

* update registry

* fix some bugs

* fix comments

* fix comments

mmdet3d/models/roi_heads/mask_heads/pointwise_semantic_head.py

@@ -5,9 +5,8 @@ from torch import nn as nn
 from torch.nn import functional as F
 
 from mmdet3d.core.bbox.structures import rotation_3d_in_axis
-from mmdet3d.models.builder import build_loss
+from mmdet3d.models.builder import HEADS, build_loss
 from mmdet.core import multi_apply
-from mmdet.models import HEADS
 
 
 @HEADS.register_module()

mmdet3d/models/roi_heads/mask_heads/primitive_head.py

@@ -6,11 +6,10 @@ from mmcv.runner import BaseModule
 from torch import nn as nn
 from torch.nn import functional as F
 
-from mmdet3d.models.builder import build_loss
+from mmdet3d.models.builder import HEADS, build_loss
 from mmdet3d.models.model_utils import VoteModule
 from mmdet3d.ops import build_sa_module
 from mmdet.core import multi_apply
-from mmdet.models import HEADS
 
 
 @HEADS.register_module()

mmdet3d/models/roi_heads/part_aggregation_roi_head.py

@@ -6,8 +6,7 @@ from torch.nn import functional as F
 from mmdet3d.core import AssignResult
 from mmdet3d.core.bbox import bbox3d2result, bbox3d2roi
 from mmdet.core import build_assigner, build_sampler
-from mmdet.models import HEADS
-from ..builder import build_head, build_roi_extractor
+from ..builder import HEADS, build_head, build_roi_extractor
 from .base_3droi_head import Base3DRoIHead
 
 

mmdet3d/models/roi_heads/point_rcnn_roi_head.py

@@ -5,8 +5,7 @@ from torch.nn import functional as F
 from mmdet3d.core import AssignResult
 from mmdet3d.core.bbox import bbox3d2result, bbox3d2roi
 from mmdet.core import build_assigner, build_sampler
-from mmdet.models import HEADS
-from ..builder import build_head, build_roi_extractor
+from ..builder import HEADS, build_head, build_roi_extractor
 from .base_3droi_head import Base3DRoIHead
 
 

mmdet3d/models/roi_heads/roi_extractors/single_roiaware_extractor.py

@@ -3,7 +3,7 @@ import torch
 from mmcv import ops
 from mmcv.runner import BaseModule
 
-from mmdet.models.builder import ROI_EXTRACTORS
+from mmdet3d.models.builder import ROI_EXTRACTORS
 
 
 @ROI_EXTRACTORS.register_module()

mmdet3d/models/roi_heads/roi_extractors/single_roipoint_extractor.py

@@ -4,7 +4,7 @@ from mmcv import ops
 from torch import nn as nn
 
 from mmdet3d.core.bbox.structures import rotation_3d_in_axis
-from mmdet.models.builder import ROI_EXTRACTORS
+from mmdet3d.models.builder import ROI_EXTRACTORS
 
 
 @ROI_EXTRACTORS.register_module()

mmdet3d/models/segmentors/encoder_decoder.py

@@ -5,8 +5,8 @@ from torch import nn as nn
 from torch.nn import functional as F
 
 from mmseg.core import add_prefix
-from mmseg.models import SEGMENTORS
-from ..builder import build_backbone, build_head, build_loss, build_neck
+from ..builder import (SEGMENTORS, build_backbone, build_head, build_loss,
+                       build_neck)
 from .base import Base3DSegmentor
 
 

tests/test_metrics/test_losses.py

@@ -3,7 +3,7 @@ import pytest
 import torch
 from torch import nn as nn
 
-from mmdet.models import build_loss
+from mmdet3d.models.builder import build_loss
 
 
 def test_chamfer_disrance():

tests/test_models/test_necks/test_fpn.py

@@ -9,7 +9,7 @@ def test_secfpn():
         upsample_strides=[1, 2],
         out_channels=[4, 6],
     )
-    from mmdet.models.builder import build_neck
+    from mmdet3d.models.builder import build_neck
     neck = build_neck(neck_cfg)
     assert neck.deblocks[0][0].in_channels == 2
     assert neck.deblocks[1][0].in_channels == 3