在detectron2中，如何将文本放在图像上？

Question

我正在学习如何很好地使用detectron2。我可以预测边界框。但是，我还想在图像上放置边界框坐标。为此，我使用cv2.putext库。但它没有起作用。 你能让下面的代码在图片上显示边界框坐标吗

from detectron2.utils.visualizer import ColorMode
import glob
for imageName in glob.glob(os.path.join(test_path, '*jpg')):
  im = cv2.imread(imageName)
  outputs = predictor(im)
  v = Visualizer(im[:, :, ::-1],
            metadata=train_metadata,
            scale=0.8)
  out = v.draw_instance_predictions(outputs["instances"].to("cpu"))
  cv2_imshow(out.get_image()[:, :, ::-1])

Answer 1

v = Visualizer(im[:, :, ::-1],
            metadata=train_metadata,
            scale=0.8)
out = v.draw_instance_predictions(outputs["instances"].to("cpu"))
boxes = v._convert_boxes(outputs["instances"].pred_boxes.to('cpu')).squeeze()
for box in boxes:
    out = v.draw_text(f"{box}", (box[0], box[1]))
cv2_imshow(out.get_image()[:, :, ::-1])

对于draw_text函数，第一个参数是文本，第二个参数是位置。有关更多信息，请查看以下链接

https://detectron2.readthedocs.io/modules/utils.html?highlight=draw_text#detectron2.utils.visualizer.Visualizer.draw_text