[Fix] single gpu test bug

2022-03-08 10:23:37 +08:00 · 2022-03-08 10:23:37 +08:00 · eeb757108e
parent 217a3ae43d
commit eeb757108e
1 changed files with 2 additions and 2 deletions
--- 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)