mirrors/mmsegmentation
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmsegmentation.git synced 2025-06-03 22:03:48 +08:00
Code Issues Projects Releases Wiki Activity

Update test.py (#93)

Browse Source
This commit is contained in:
MY_ 2020-08-25 21:12:39 +08:00 committed by GitHub
parent 8c0e093b31
commit 429c8b3fbf
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mmseg/apis/test.py
View File

@ -31,7 +31,7 @@ def single_gpu_test(model, data_loader, show=False, out_dir=None):
for i, data in enumerate(data_loader): for i, data in enumerate(data_loader):
with torch.no_grad(): with torch.no_grad():
result = model(return_loss=False, **data) result = model(return_loss=False, **data)
if isinstance(results, list): if isinstance(result, list):
results.extend(result) results.extend(result)
else: else:
results.append(result) 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): for i, data in enumerate(data_loader):
with torch.no_grad(): with torch.no_grad():
result = model(return_loss=False, rescale=True, **data) result = model(return_loss=False, rescale=True, **data)
if isinstance(results, list): if isinstance(result, list):
results.extend(result) results.extend(result)
else: else:
results.append(result) results.append(result)