fix test tools bugs

xvjiarui · Oct 10, 2018 · b07edd6 · b07edd6
1 parent 839c74b
commit b07edd6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tools/test.py b/tools/test.py
@@ -3,11 +3,11 @@
 import torch
 import mmcv
 from mmcv.runner import load_checkpoint, parallel_test, obj_from_dict
-from mmcv.parallel import scatter, MMDataParallel
+from mmcv.parallel import scatter, collate, MMDataParallel
 
 from mmdet import datasets
 from mmdet.core import results2json, coco_eval
-from mmdet.datasets import collate, build_dataloader
+from mmdet.datasets import build_dataloader
 from mmdet.models import build_detector, detectors