From dd619179c57a3fe33b6d80b34a39c4519377ffef Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Tue, 2 Jul 2019 11:25:01 +0200 Subject: [PATCH] Fixed classes variable assignment Signed-off-by: Jim Martens --- src/twomartens/masterthesis/cli.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/twomartens/masterthesis/cli.py b/src/twomartens/masterthesis/cli.py index 86cbd19..038843b 100644 --- a/src/twomartens/masterthesis/cli.py +++ b/src/twomartens/masterthesis/cli.py @@ -367,7 +367,8 @@ def visualise(args: argparse.Namespace) -> None: _, _, cats_to_names, _ = coco_utils.get_coco_category_maps(annotation_file_train) colors = pyplot.cm.hsv(np.linspace(0, 1, 81)).tolist() - classes = ['background'].extend(cats_to_names) + classes = ['background'] + classes.extend(cats_to_names) i = 0 nr_images = len(file_names[args.trajectory]) @@ -384,7 +385,6 @@ def visualise(args: argparse.Namespace) -> None: current_axis = pyplot.gca() for instance in labels: - print(instance) bbox = instance['bbox'] # Transform the predicted bounding boxes for the 300x300 image to the original image dimensions. xmin = bbox[0]