2martens/masterthesis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Fixed classes variable assignment

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-07-02 11:25:01 +02:00
parent 58d85a7efc
commit dd619179c5
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/twomartens/masterthesis/cli.py
View File

@ -367,7 +367,8 @@ def visualise(args: argparse.Namespace) -> None:
_, _, cats_to_names, _ = coco_utils.get_coco_category_maps(annotation_file_train) _, _, cats_to_names, _ = coco_utils.get_coco_category_maps(annotation_file_train)
colors = pyplot.cm.hsv(np.linspace(0, 1, 81)).tolist() 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 i = 0
nr_images = len(file_names[args.trajectory]) nr_images = len(file_names[args.trajectory])
@ -384,7 +385,6 @@ def visualise(args: argparse.Namespace) -> None:
current_axis = pyplot.gca() current_axis = pyplot.gca()
for instance in labels: for instance in labels:
print(instance)
bbox = instance['bbox'] bbox = instance['bbox']
# Transform the predicted bounding boxes for the 300x300 image to the original image dimensions. # Transform the predicted bounding boxes for the 300x300 image to the original image dimensions.
xmin = bbox[0] xmin = bbox[0]