[英]In detectron2, how could I plot only bounding on images without label (class name) in object detection?
我正在学习detectron2并使用pokemonster数据练习它。 所以,我按照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
类包含所有方法,包括draw_box()
。 你想把所有的盒子都画到你的Visualizer
对象上,最后返回一个VisImage
v = Visualizer(
im[:, :, ::-1],
metadata=train_metadata,
scale=0.8,
)
for box in outputs["instances"].pred_boxes.to('cpu'):
v.draw_box(box)
v.draw_text(str(box[:2].numpy()), tuple(box[:2].numpy()))
v = v.get_output()
img = v.get_image()[:, :, ::-1]
cv2_imshow(img)
有绘制二进制蒙版的功能,但没有在彩色图像上绘制蒙版的功能。 如何在原始输入彩色图像上绘制蒙版。
output = predictor(input_image)
V = Visualizer(input_image[:, :, ::-1],
MetadataCatalog.get(cfg.DATASETS.TRAIN[0]), scale = 1.2)
mask=output["instances"].get("pred_masks")
mask=mask.to("cpu")
V.draw_instance_predictions(output["instances"].to("cpu"))
for m in mask:
V.draw_mask(m)
V = V.get_output()
cv2.imwrite("frame%d.jpg" % count, V.get_image()[:, :, ::-1])
谢谢
这是盒子和面具的工作版本:
v = Visualizer(im[:, :, ::-1],
metadata=train_metadata,
scale=0.8,
)
# Boxes
for box in outputs["instances"].pred_boxes.to('cpu'):
v.draw_box(box)
# Masks
for mask in outputs["instances"].pred_masks.to('cpu'):
v.draw_soft_mask(mask)
v = v.get_output()
img = v.get_image()[:, :, ::-1]
cv2_imshow(img)
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.