Skip to content

GitLab

Commit a6adf8f0 authored by Kai Chen's avatar Kai Chen
Browse files

minor fix

parent 1229095f
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
mmdet/models/detectors/base.py
View file @ a6adf8f0
...@@ -3,7 +3,6 @@ from abc import ABCMeta, abstractmethod ...@@ -3,7 +3,6 @@ from abc import ABCMeta, abstractmethod
import mmcv import mmcv
import numpy as np import numpy as np
import torch
import torch.nn as nn import torch.nn as nn
from mmdet.core import tensor2imgs, get_classes from mmdet.core import tensor2imgs, get_classes
......
tools/test.py
View file @ a6adf8f0
...@@ -32,7 +32,7 @@ def single_test(model, data_loader, show=False): ...@@ -32,7 +32,7 @@ def single_test(model, data_loader, show=False):
def _data_func(data, device_id): def _data_func(data, device_id):
data = scatter(collate([data], samples_per_gpu=1), [device_id])[0] 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(): def parse_args():
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment