Detectron2 将输出文件保存为 json

问题描述 投票:0回答:0

我想将输出图像保存为 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])
python json image-segmentation semantic-segmentation detectron
© www.soinside.com 2019 - 2024. All rights reserved.