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 efficient test bug (#702)

Browse Source
This commit is contained in:
Junjun2016 2021-07-15 11:42:33 +08:00 committed by GitHub
parent 813222a116
commit 259eeb08c8
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
mmseg/core/evaluation/eval_hooks.py
View File

@ -31,7 +31,11 @@ class EvalHook(_EvalHook):
return return
from mmseg.apis import single_gpu_test from mmseg.apis import single_gpu_test
results = single_gpu_test(runner.model, self.dataloader, show=False) results = single_gpu_test(
runner.model,
self.dataloader,
show=False,
efficient_test=self.efficient_test)
runner.log_buffer.output['eval_iter_num'] = len(self.dataloader) runner.log_buffer.output['eval_iter_num'] = len(self.dataloader)
key_score = self.evaluate(runner, results) key_score = self.evaluate(runner, results)
if self.save_best: if self.save_best:
@ -84,7 +88,8 @@ class DistEvalHook(_DistEvalHook):
runner.model, runner.model,
self.dataloader, self.dataloader,
tmpdir=tmpdir, tmpdir=tmpdir,
gpu_collect=self.gpu_collect) gpu_collect=self.gpu_collect,
efficient_test=self.efficient_test)
if runner.rank == 0: if runner.rank == 0:
print('\n') print('\n')
runner.log_buffer.output['eval_iter_num'] = len(self.dataloader) runner.log_buffer.output['eval_iter_num'] = len(self.dataloader)