fix metric's dataset metainfo is none bug

Seeed-Studio · Jul 10, 2023 · e00ff76 · e00ff76
1 parent 7a869bb
commit e00ff76
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tools/utils/inference.py b/tools/utils/inference.py
@@ -280,6 +280,7 @@ def __init__(
 
  def init(self, cfg):
  self.evaluator: Evaluator = self.runner.build_evaluator(self.cfg.get('val_evaluator'))
+ self.evaluator.dataset_meta = self.dataloader.dataset.METAINFO
  if hasattr(cfg.model, 'data_preprocessor'):
  self.data_preprocess = MODELS.build(cfg.model.data_preprocessor)