From a6adf8f019f816f3f9f7f5ae26db05ef4429a562 Mon Sep 17 00:00:00 2001
From: Kai Chen <chenkaidev@gmail.com>
Date: Wed, 10 Oct 2018 20:37:06 +0800
Subject: [PATCH] minor fix
---
mmdet/models/detectors/base.py | 1 -
tools/test.py | 2 +-
2 files changed, 1 insertion(+), 2 deletions(-)
diff --git a/mmdet/models/detectors/base.py b/mmdet/models/detectors/base.py
index e617b0e..6d26dc3 100644
--- a/mmdet/models/detectors/base.py
+++ b/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
diff --git a/tools/test.py b/tools/test.py
index e1552e5..b322bb2 100644
--- a/tools/test.py
+++ b/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():
--
GitLab