From b07edd6b1d29530c50855e193b45f21d40187348 Mon Sep 17 00:00:00 2001 From: pangjm <pjmzju@gmail.com> Date: Wed, 10 Oct 2018 14:06:27 +0800 Subject: [PATCH] fix test tools bugs --- tools/test.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tools/test.py b/tools/test.py index 3b1ce2d..2552e7a 100644 --- 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 -- GitLab