问题描述
我正在学习如何正确使用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
函数,第一个参数是文本,第二个参数是位置。有关更多信息,请检查以下链接。