From eeb757108eeb650296b3a8b567ea0d30a14a6e68 Mon Sep 17 00:00:00 2001 From: Liang Yang Date: Tue, 8 Mar 2022 10:23:37 +0800 Subject: [PATCH] [Fix] single gpu test bug --- mmfewshot/detection/apis/test.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mmfewshot/detection/apis/test.py b/mmfewshot/detection/apis/test.py index 37b218e..9efcdd9 100644 --- a/mmfewshot/detection/apis/test.py +++ b/mmfewshot/detection/apis/test.py @@ -69,14 +69,14 @@ def single_gpu_test(model: nn.Module, if is_module_wrapper(model): model.module.show_result( img_show, - result[i], + result[j], show=show, out_file=out_file, score_thr=show_score_thr) else: model.show_result( img_show, - result[i], + result[j], show=show, out_file=out_file, score_thr=show_score_thr)