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

Attempt to switch axes

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

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

@ -385,10 +385,10 @@ def visualise(args: argparse.Namespace) -> None:
for instance in labels: for instance in labels:
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[1]
ymin = bbox[1] ymin = bbox[0]
xmax = bbox[2] xmax = bbox[3]
ymax = bbox[3] ymax = bbox[2]
color = colors[cats_to_classes[int(instance['coco_id'])]] color = colors[cats_to_classes[int(instance['coco_id'])]]
label = f"{cats_to_names[int(instance['coco_id'])]}" label = f"{cats_to_names[int(instance['coco_id'])]}"
current_axis.add_patch( current_axis.add_patch(