Commit e3b5253b authored by ZCMax's avatar ZCMax Committed by ChaimZhu
Browse files

Update all registries and fix some ut problems

parent 8dd8da12
......@@ -4,9 +4,9 @@ import torch
from mmcv.cnn import Scale
from torch import nn as nn
from mmdet3d.core import build_bbox_coder
from mmdet3d.core.bbox import (CameraInstance3DBoxes, DepthInstance3DBoxes,
LiDARInstance3DBoxes)
from mmdet.core import build_bbox_coder
def test_partial_bin_based_box_coder():
......
......@@ -12,14 +12,14 @@ from mmdet3d.utils import register_all_modules, setup_multi_processes
def test_register_all_modules():
from mmdet3d.registry import TRANSFORMS
from mmdet3d.registry import DATASETS
sys.modules.pop('mmdet3d.datasets', None)
sys.modules.pop('mmdet3d.datasets.pipelines', None)
TRANSFORMS._module_dict.pop('PointSample', None)
assert 'PointSample' not in TRANSFORMS.module_dict
sys.modules.pop('mmdet3d.datasets.kitti_dataset', None)
DATASETS._module_dict.pop('KittiDataset', None)
assert 'KittiDataset' not in DATASETS.module_dict
register_all_modules(init_default_scope=True)
assert 'PointSample' in TRANSFORMS.module_dict
assert 'KittiDataset' in DATASETS.module_dict
assert DefaultScope.get_current_instance().scope_name == 'mmdet3d'
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment