infer batch size using len(result) in test function (open-mmlab#532)

bowenroom · May 6, 2021 · f253451 · f253451
1 parent aa14be1
commit f253451
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmseg/apis/test.py b/mmseg/apis/test.py
@@ -149,7 +149,7 @@ def multi_gpu_test(model,
             results.append(result)
 
         if rank == 0:
-            batch_size = data['img'][0].size(0)
+            batch_size = len(result)
             for _ in range(batch_size * world_size):
                 prog_bar.update()