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

[Fix] Fix test error when using --show-dir (#1091)

Browse Source 
* Fix test error when using --show-dir

* Move show before test

Co-authored-by: Jiarui XU <xvjiarui0826@gmail.com>
This commit is contained in:
Zhaolun Zou 2021-12-07 00:10:18 +08:00 committed by GitHub
parent 3d58196e62
commit 079c520e58
1 changed files with 14 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
mmseg/apis/test.py
View File

@ -90,19 +90,6 @@ def single_gpu_test(model,
with torch.no_grad(): with torch.no_grad():
result = model(return_loss=False, **data) result = model(return_loss=False, **data)
if efficient_test:
result = [np2tmp(_, tmpdir='.efficient_test') for _ in result]
if format_only:
result = dataset.format_results(
result, indices=batch_indices, **format_args)
if pre_eval:
# TODO: adapt samples_per_gpu > 1.
# only samples_per_gpu=1 valid now
result = dataset.pre_eval(result, indices=batch_indices)
results.extend(result)
if show or out_dir: if show or out_dir:
img_tensor = data['img'][0] img_tensor = data['img'][0]
img_metas = data['img_metas'][0].data[0] img_metas = data['img_metas'][0].data[0]
@ -129,6 +116,20 @@ def single_gpu_test(model,
out_file=out_file, out_file=out_file,
opacity=opacity) opacity=opacity)
if efficient_test:
result = [np2tmp(_, tmpdir='.efficient_test') for _ in result]
if format_only:
result = dataset.format_results(
result, indices=batch_indices, **format_args)
if pre_eval:
# TODO: adapt samples_per_gpu > 1.
# only samples_per_gpu=1 valid now
result = dataset.pre_eval(result, indices=batch_indices)
results.extend(result)
else:
results.extend(result)
batch_size = len(result) batch_size = len(result)
for _ in range(batch_size): for _ in range(batch_size):
prog_bar.update() prog_bar.update()