我想将输出图像保存为 json 以供将来注释。 以下代码能够显示输出预测。想寻求您的帮助,以帮助我如何将输出保存为 json 格式。非常感谢。
from detectron2.utils.visualizer import ColorMode
path = '/content/drive/MyDrive/Detectron2/test/*.png'
for file in glob.glob(path):
print(file)
im = cv2.imread(file, 1)
outputs = predictor(im)
print(outputs)
v = Visualizer(im[:, :, ::-1],
metadata=train_metadata,
scale=0.5,
instance_mode=ColorMode.SEGMENTATION
)
out = v.draw_instance_predictions(outputs["instances"].to("cpu"))
cv2_imshow(out.get_image()[:, :, ::-1])