我正在学习detectron2并使用口袋妖怪数据进行练习。 因此,我遵循detectron2教程(这里:https://colab.research.google.com/drive/16jcaJoc6bCFAQ96jDe2HwtXj7BMD_-m5)
此外,还开发了一个检测器,可以检测到下面的图像
但是,在这张图中,红色圆圈是类名。我只想绘制边界框(绿色),红色圆圈(类名)除外。 下面是我的可视化代码。应该修改什么代码? 多谢各位
import glob
for imageName in sorted(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"))
boxes = v._convert_boxes(outputs["instances"].pred_boxes.to('cpu'))
for box in boxes:
box = (round(box[0]), round(box[1]), round(box[2]) - round(box[0]), round(box[3] - box[1]))
out = v.draw_text(f"{box[2:4]}", (box[0], box[1]))
cv2_imshow(out.get_image()[:, :, ::-1])
有一个函数可以绘制二值遮罩,但没有函数可以在彩色图像上绘制遮罩。如何在原始输入的彩色图像上绘制遮罩
多谢各位
类^{} 包含包含^{} 的所有方法。您希望将所有框绘制到
Visualizer
对象,最后返回一个VisImage
相关问题 更多 >
编程相关推荐