minor bug fix

Author: hellock

mmdet/core/loss/losses.py

@@ -92,6 +92,8 @@ def accuracy(pred, target, topk=1):
     if isinstance(topk, int):
         topk = (topk, )
         return_single = True
+    else:
+        return_single = False
 
     maxk = max(topk)
     _, pred_label = pred.topk(maxk, 1, True, True)

tools/test.py

@@ -23,7 +23,7 @@ def single_test(model, data_loader, show=False):
 
         if show:
             model.module.show_result(data, result, dataset.img_norm_cfg,
-                                     dataset.CLASSES)
+                                     dataset=dataset.CLASSES)
 
         batch_size = data['img'][0].size(0)
         for _ in range(batch_size):