diff --git a/mmseg/apis/test.py b/mmseg/apis/test.py index 5b4b2da40..7f98abf29 100644 --- a/mmseg/apis/test.py +++ b/mmseg/apis/test.py @@ -31,7 +31,7 @@ def single_gpu_test(model, data_loader, show=False, out_dir=None): for i, data in enumerate(data_loader): with torch.no_grad(): result = model(return_loss=False, **data) - if isinstance(results, list): + if isinstance(result, list): results.extend(result) else: results.append(result) @@ -96,7 +96,7 @@ def multi_gpu_test(model, data_loader, tmpdir=None, gpu_collect=False): for i, data in enumerate(data_loader): with torch.no_grad(): result = model(return_loss=False, rescale=True, **data) - if isinstance(results, list): + if isinstance(result, list): results.extend(result) else: results.append(result)