fix test tools bugs

b07edd6b · pangjm · 839c74b7 · b07edd6b
Commit b07edd6b authored Oct 10, 2018 by pangjm
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

tools/test.py tools/test.py +2 -2

No files found.
--- a/tools/test.py
+++ b/tools/test.py
@@ -3,11 +3,11 @@ import argparse
 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