Skip to content
Snippets Groups Projects
Unverified Commit f9cf4daa authored by Kai Chen's avatar Kai Chen Committed by GitHub
Browse files

Merge pull request #203 from hellock/hotfix 

minor bug fix
parents c5d8f002 46a0de62
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
mmdet/core/loss/losses.py
+ 2
− 0
View file @ f9cf4daa
...@@ -92,6 +92,8 @@ def accuracy(pred, target, topk=1): ...@@ -92,6 +92,8 @@ def accuracy(pred, target, topk=1):
if isinstance(topk, int): if isinstance(topk, int):
topk = (topk, ) topk = (topk, )
return_single = True return_single = True
else:
return_single = False
maxk = max(topk) maxk = max(topk)
_, pred_label = pred.topk(maxk, 1, True, True) _, pred_label = pred.topk(maxk, 1, True, True)
......
tools/test.py
+ 1
− 1
View file @ f9cf4daa
...@@ -23,7 +23,7 @@ def single_test(model, data_loader, show=False): ...@@ -23,7 +23,7 @@ def single_test(model, data_loader, show=False):
if show: if show:
model.module.show_result(data, result, dataset.img_norm_cfg, model.module.show_result(data, result, dataset.img_norm_cfg,
dataset.CLASSES) dataset=dataset.CLASSES)
batch_size = data['img'][0].size(0) batch_size = data['img'][0].size(0)
for _ in range(batch_size): for _ in range(batch_size):
......
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