diff --git a/tools/confusion_matrix.py b/tools/confusion_matrix.py index 41d308b81c..2c5b64cf4e 100644 --- a/tools/confusion_matrix.py +++ b/tools/confusion_matrix.py @@ -25,6 +25,10 @@ def parse_args(): '--color-theme', default='winter', help='theme of the matrix color map') + parser.add_argument( + '--title', + default='Normalized Confusion Matrix', + help='title of the matrix color map') parser.add_argument( '--cfg-options', nargs='+', @@ -171,7 +175,9 @@ def main(): confusion_matrix, dataset.CLASSES, save_dir=args.save_dir, - show=args.show) + show=args.show, + title=args.title, + color_theme=args.color_theme) if __name__ == '__main__':