Unverified Commit d8425466 authored by VVsssssk's avatar VVsssssk Committed by GitHub
Browse files

[Bug]fix mmdet bc breaking (#1434)

* fix mmdet bc breaking

* fix version
parent 276b2648
...@@ -11,17 +11,18 @@ from mmcv.parallel import MMDataParallel, MMDistributedDataParallel ...@@ -11,17 +11,18 @@ from mmcv.parallel import MMDataParallel, MMDistributedDataParallel
from mmcv.runner import (get_dist_info, init_dist, load_checkpoint, from mmcv.runner import (get_dist_info, init_dist, load_checkpoint,
wrap_fp16_model) wrap_fp16_model)
import mmdet
from mmdet3d.apis import single_gpu_test from mmdet3d.apis import single_gpu_test
from mmdet3d.datasets import build_dataloader, build_dataset from mmdet3d.datasets import build_dataloader, build_dataset
from mmdet3d.models import build_model from mmdet3d.models import build_model
from mmdet.apis import multi_gpu_test, set_random_seed from mmdet.apis import multi_gpu_test, set_random_seed
from mmdet.datasets import replace_ImageToTensor from mmdet.datasets import replace_ImageToTensor
try: if mmdet.__version__ > '2.23.0':
# If mmdet version > 2.20.0, setup_multi_processes would be imported and # If mmdet version > 2.23.0, setup_multi_processes would be imported and
# used from mmdet instead of mmdet3d. # used from mmdet instead of mmdet3d.
from mmdet.utils import setup_multi_processes from mmdet.utils import setup_multi_processes
except ImportError: else:
from mmdet3d.utils import setup_multi_processes from mmdet3d.utils import setup_multi_processes
......
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