diff --git a/tools/analysis_tools/confusion_matrix.py b/tools/analysis_tools/confusion_matrix.py index 224b8364bc..696537ea4a 100644 --- a/tools/analysis_tools/confusion_matrix.py +++ b/tools/analysis_tools/confusion_matrix.py @@ -1,5 +1,6 @@ # Copyright (c) OpenMMLab. All rights reserved. import argparse +import logging import tempfile import torch @@ -83,6 +84,7 @@ def main(): classes = runner.test_loop.dataloader.dataset.metainfo.get( 'classes') cm = runner.test()['confusion_matrix/result'] + logging.shutdown() else: predictions = load(args.ckpt_or_result) evaluator = Evaluator(ConfusionMatrix())