如何解决在detectron2中,如何将文本放在图像上?
我正在学习如何正确使用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])
解决方法
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
函数,第一个参数是文本,第二个参数是位置。有关更多信息,请检查以下链接。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。