color_theme error fixed

open-mmlab · Feb 5, 2025 · cd491ea · cd491ea
1 parent c7adc52
commit cd491ea
Showing 1 changed file with 5 additions and 1 deletion.
diff --git 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()
+