diff --git a/mmseg/apis/test.py b/mmseg/apis/test.py
index 2b9cc1703..1597df6aa 100644
--- a/mmseg/apis/test.py
+++ b/mmseg/apis/test.py
@@ -97,7 +97,7 @@ def single_gpu_test(model,
                 result = np2tmp(result)
             results.append(result)
 
-        batch_size = data['img'][0].size(0)
+        batch_size = len(result)
         for _ in range(batch_size):
             prog_bar.update()
     return results
diff --git a/mmseg/models/segmentors/base.py b/mmseg/models/segmentors/base.py
index 58c31887f..7b5375753 100644
--- a/mmseg/models/segmentors/base.py
+++ b/mmseg/models/segmentors/base.py
@@ -155,7 +155,7 @@ class BaseSegmentor(nn.Module):
         outputs = dict(
             loss=loss,
             log_vars=log_vars,
-            num_samples=len(data_batch['img'].data))
+            num_samples=len(data_batch['img_metas']))
 
         return outputs