From cd491ea3e15665f56904d3c78b32e7d0754f9aac Mon Sep 17 00:00:00 2001 From: unique-chan Date: Wed, 5 Feb 2025 22:25:38 +0900 Subject: [PATCH] color_theme error fixed --- tools/analysis_tools/confusion_matrix.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/tools/analysis_tools/confusion_matrix.py b/tools/analysis_tools/confusion_matrix.py index df5f88fe2..9a2350ea7 100644 --- a/tools/analysis_tools/confusion_matrix.py +++ b/tools/analysis_tools/confusion_matrix.py @@ -226,6 +226,8 @@ def plot_confusion_matrix(confusion_matrix, if save_dir is not None: plt.savefig( os.path.join(save_dir, 'confusion_matrix.png'), format='png') + plt.savefig( + os.path.join(save_dir, 'confusion_matrix.svg'), format='svg') if show: plt.show() @@ -261,8 +263,10 @@ def main(): confusion_matrix, dataset.CLASSES + ('background', ), save_dir=args.save_dir, - show=args.show) + show=args.show, + color_theme=args.color_theme) if __name__ == '__main__': main() +