Как в детекторроне2 я могу разместить имена ключевых точек на изображении? ⇐ Python

[Anonymous]

Во-первых, я объясню постановку задачи. Я использую Detectron2 для обнаружения ключевых точек для моего пользовательского набора данных. Я использую приведенный ниже код для визуализации ограничивающей рамки и ключевых точек.

Код: Выделить всё

outputs = predictor(im[..., ::-1])
v = Visualizer(im[:, :, ::-1], MetadataCatalog.get(cfg.DATASETS.TRAIN[0]), scale=1.2)
out = v.draw_instance_predictions(outputs["instances"].to("cpu"))
plt.figure(figsize=(60,30))
plt.imshow(out.get_image()[..., ::-1][..., ::-1])

Я получаю результат в виде изображения ниже.
[img]https://i. sstatic.net/TIucH.png[/img]

Как видите, метка класса и две ключевые точки нарисованы на изображении визуализатором детектора 2. Имена двух ключевых точек — [keypoint_up, keypoint_down].
Я также хочу напечатать имена двух ключевых точек на изображении. Что-то вроде примера ниже.

Я знаю, что детектор предоставляет функцию draw_text. Если у кого-нибудь есть идея, как я мог бы использовать его, чтобы поместить имя ключевой точки на изображение, это было бы очень полезно. Пример кодирования был бы очень полезен. Любое другое решение, кроме функции draw_text, также приветствуется. Спасибо, что уделили время

Подробнее здесь: https://stackoverflow.com/questions/746 ... -the-image