Fix incorrect behavior to access train pipeline from ConcatDataset …

…in analyze_results.py (#11004)
open-mmlab · Oct 8, 2023 · 4f26a9e · 4f26a9e
1 parent bbfa179
commit 4f26a9e
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tools/analysis_tools/analyze_results.py b/tools/analysis_tools/analyze_results.py
@@ -375,9 +375,12 @@ def main():
  cfg.test_dataloader.pop('batch_size', 0)
  if cfg.train_dataloader.dataset.type in ('MultiImageMixDataset',
  'ClassBalancedDataset',
- 'RepeatDataset', 'ConcatDataset'):
+ 'RepeatDataset'):
  cfg.test_dataloader.dataset.pipeline = get_loading_pipeline(
  cfg.train_dataloader.dataset.dataset.pipeline)
+ elif cfg.train_dataloader.dataset.type in ('ConcatDataset', ):
+ cfg.test_dataloader.dataset.pipeline = get_loading_pipeline(
+ cfg.train_dataloader.dataset.datasets[0].pipeline)
  else:
  cfg.test_dataloader.dataset.pipeline = get_loading_pipeline(
  cfg.train_dataloader.dataset.pipeline)