minor fix

mmdet/models/detectors/base.py

@@ -3,7 +3,6 @@ from abc import ABCMeta, abstractmethod
 
 import mmcv
 import numpy as np
-import torch
 import torch.nn as nn
 
 from mmdet.core import tensor2imgs, get_classes

tools/test.py

@@ -32,7 +32,7 @@ def single_test(model, data_loader, show=False):
 
 def _data_func(data, device_id):
     data = scatter(collate([data], samples_per_gpu=1), [device_id])[0]
-    return dict(**data, return_loss=False, rescale=True)
+    return dict(return_loss=False, rescale=True, **data)
 
 
 def parse_args():