[Fix] Fix init_weight problem by reverting PR#495 and requiring mmdet<=2.11.0 (#501)

.github/workflows/build.yml

@@ -91,7 +91,7 @@ jobs:
       - name: Install mmdet3d dependencies
         run: |
           pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu101/${{matrix.torch_version}}/index.html
-          pip install -q git+https://github.com/open-mmlab/mmdetection.git
+          pip install mmdet==2.11.0
           pip install -r requirements.txt
       - name: Build and install
         run: |

docs/getting_started.md

@@ -12,15 +12,15 @@ The required versions of MMCV and MMDetection for different versions of MMDetect
 
 | MMDetection3D version | MMDetection version |    MMCV version     |
 |:-------------------:|:-------------------:|:-------------------:|
-| master              | mmdet>=2.10.0       | mmcv-full>=1.3.2, <=1.4|
-| 0.12.0              | mmdet>=2.5.0        | mmcv-full>=1.2.4, <=1.4|
-| 0.11.0              | mmdet>=2.5.0        | mmcv-full>=1.2.4, <=1.4|
-| 0.10.0              | mmdet>=2.5.0        | mmcv-full>=1.2.4, <=1.4|
-| 0.9.0               | mmdet>=2.5.0        | mmcv-full>=1.2.4, <=1.4|
-| 0.8.0               | mmdet>=2.5.0        | mmcv-full>=1.1.5, <=1.4|
-| 0.7.0               | mmdet>=2.5.0        | mmcv-full>=1.1.5, <=1.4|
-| 0.6.0               | mmdet>=2.4.0        | mmcv-full>=1.1.3, <=1.2|
-| 0.5.0               | 2.3.0               | mmcv-full==1.0.5|
+| master              | mmdet>=2.10.0, <=2.11.0| mmcv-full>=1.2.4, <=1.4|
+| 0.12.0              | mmdet>=2.5.0, <=2.11.0 | mmcv-full>=1.2.4, <=1.4|
+| 0.11.0              | mmdet>=2.5.0, <=2.11.0 | mmcv-full>=1.2.4, <=1.4|
+| 0.10.0              | mmdet>=2.5.0, <=2.11.0 | mmcv-full>=1.2.4, <=1.4|
+| 0.9.0               | mmdet>=2.5.0, <=2.11.0 | mmcv-full>=1.2.4, <=1.4|
+| 0.8.0               | mmdet>=2.5.0, <=2.11.0 | mmcv-full>=1.1.5, <=1.4|
+| 0.7.0               | mmdet>=2.5.0, <=2.11.0 | mmcv-full>=1.1.5, <=1.4|
+| 0.6.0               | mmdet>=2.4.0, <=2.11.0 | mmcv-full>=1.1.3, <=1.2|
+| 0.5.0               | 2.3.0                  | mmcv-full==1.0.5|
 
 # Installation
 

mmdet3d/__init__.py

@@ -16,7 +16,7 @@ def digit_version(version_str):
     return digit_version
 
 
-mmcv_minimum_version = '1.3.2'
+mmcv_minimum_version = '1.2.4'
 mmcv_maximum_version = '1.4.0'
 mmcv_version = digit_version(mmcv.__version__)
 
@@ -27,7 +27,7 @@ assert (mmcv_version >= digit_version(mmcv_minimum_version)
     f'Please install mmcv>={mmcv_minimum_version}, <={mmcv_maximum_version}.'
 
 mmdet_minimum_version = '2.10.0'
-mmdet_maximum_version = '3.0.0'
+mmdet_maximum_version = '2.11.0'
 mmdet_version = digit_version(mmdet.__version__)
 assert (mmdet_version >= digit_version(mmdet_minimum_version)
         and mmdet_version <= digit_version(mmdet_maximum_version)), \

mmdet3d/models/builder.py

 import warnings
-from mmcv.cnn import MODELS as MMCV_MODELS
 from mmcv.utils import Registry
 
 from mmdet.models.builder import (BACKBONES, DETECTORS, HEADS, LOSSES, NECKS,
-                                  ROI_EXTRACTORS, SHARED_HEADS)
+                                  ROI_EXTRACTORS, SHARED_HEADS, build)
 
-MODELS = Registry('models', parent=MMCV_MODELS)
-
-VOXEL_ENCODERS = MODELS
-MIDDLE_ENCODERS = MODELS
-FUSION_LAYERS = MODELS
+VOXEL_ENCODERS = Registry('voxel_encoder')
+MIDDLE_ENCODERS = Registry('middle_encoder')
+FUSION_LAYERS = Registry('fusion_layer')
 
 
 def build_backbone(cfg):
     """Build backbone."""
-    return BACKBONES.build(cfg)
+    return build(cfg, BACKBONES)
 
 
 def build_neck(cfg):
     """Build neck."""
-    return NECKS.build(cfg)
+    return build(cfg, NECKS)
 
 
 def build_roi_extractor(cfg):
     """Build RoI feature extractor."""
-    return ROI_EXTRACTORS.build(cfg)
+    return build(cfg, ROI_EXTRACTORS)
 
 
 def build_shared_head(cfg):
     """Build shared head of detector."""
-    return SHARED_HEADS.build(cfg)
+    return build(cfg, SHARED_HEADS)
 
 
 def build_head(cfg):
     """Build head."""
-    return HEADS.build(cfg)
+    return build(cfg, HEADS)
 
 
 def build_loss(cfg):
     """Build loss function."""
-    return LOSSES.build(cfg)
+    return build(cfg, LOSSES)
 
 
 def build_detector(cfg, train_cfg=None, test_cfg=None):
@@ -52,20 +49,19 @@ def build_detector(cfg, train_cfg=None, test_cfg=None):
         'train_cfg specified in both outer field and model field '
     assert cfg.get('test_cfg') is None or test_cfg is None, \
         'test_cfg specified in both outer field and model field '
-    return DETECTORS.build(
-        cfg, default_args=dict(train_cfg=train_cfg, test_cfg=test_cfg))
+    return build(cfg, DETECTORS, dict(train_cfg=train_cfg, test_cfg=test_cfg))
 
 
 def build_voxel_encoder(cfg):
     """Build voxel encoder."""
-    return VOXEL_ENCODERS.build(cfg)
+    return build(cfg, VOXEL_ENCODERS)
 
 
 def build_middle_encoder(cfg):
     """Build middle level encoder."""
-    return MIDDLE_ENCODERS.build(cfg)
+    return build(cfg, MIDDLE_ENCODERS)
 
 
 def build_fusion_layer(cfg):
     """Build fusion layer."""
-    return FUSION_LAYERS.build(cfg)
+    return build(cfg, FUSION_LAYERS)