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

Added wordnet class name to visualised ground truth bboxes

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-07-02 11:52:17 +02:00
parent 8bddf320f8
commit 92d60ad4e3
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -390,7 +390,7 @@ def visualise(args: argparse.Namespace) -> None:
xmax = bbox[2]
ymax = bbox[3]
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'])]}: {instance['wordnet_class_name']}"
current_axis.add_patch(
pyplot.Rectangle((xmin, ymin), xmax - xmin, ymax - ymin, color=color, fill=False, linewidth=2))
current_axis.text(xmin, ymin, label, size='x-large', color='white',