Commit 1229095f authored by Kai Chen's avatar Kai Chen
Browse files

fix flake8 error in python 2

parent ba927c9e
...@@ -74,7 +74,7 @@ class DistEvalHook(Hook): ...@@ -74,7 +74,7 @@ class DistEvalHook(Hook):
# compute output # compute output
with torch.no_grad(): with torch.no_grad():
result = runner.model( result = runner.model(
**data_gpu, return_loss=False, rescale=True) return_loss=False, rescale=True, **data_gpu)
results[idx] = result results[idx] = result
batch_size = runner.world_size batch_size = runner.world_size
......
...@@ -34,9 +34,7 @@ class BaseDetector(nn.Module): ...@@ -34,9 +34,7 @@ class BaseDetector(nn.Module):
pass pass
def extract_feats(self, imgs): def extract_feats(self, imgs):
if isinstance(imgs, torch.Tensor): assert isinstance(imgs, list)
return self.extract_feat(imgs)
elif isinstance(imgs, list):
for img in imgs: for img in imgs:
yield self.extract_feat(img) yield self.extract_feat(img)
......
...@@ -17,7 +17,7 @@ def single_test(model, data_loader, show=False): ...@@ -17,7 +17,7 @@ def single_test(model, data_loader, show=False):
prog_bar = mmcv.ProgressBar(len(data_loader.dataset)) prog_bar = mmcv.ProgressBar(len(data_loader.dataset))
for i, data in enumerate(data_loader): for i, data in enumerate(data_loader):
with torch.no_grad(): with torch.no_grad():
result = model(**data, return_loss=False, rescale=not show) result = model(return_loss=False, rescale=not show, **data)
results.append(result) results.append(result)
if show: if show:
......
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