Reverse indices to what it should be

Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
2019-07-02 11:44:43 +02:00
parent 7ad1789e24
commit 3d7d97355c

View File

@ -385,10 +385,10 @@ def visualise(args: argparse.Namespace) -> None:
for instance in labels:
bbox = instance['bbox']
# Transform the predicted bounding boxes for the 300x300 image to the original image dimensions.
xmin = bbox[1]
ymin = bbox[0]
xmax = bbox[3]
ymax = bbox[2]
xmin = bbox[0]
ymin = bbox[1]
xmax = bbox[2]
ymax = bbox[3]
color = colors[cats_to_classes[int(instance['coco_id'])]]
label = f"{cats_to_names[int(instance['coco_id'])]}"
current_axis.add_patch(