基于coco 格式的数据集，将coco json标签绘制在原图上，并输出到output文件夹

在一个目录中，包含train和val文件夹，以及train.json和val.json文件，将此程序放在该目录下即可
其中train和val文件夹包含jpg图像

import json
import os
from PIL import Image, ImageDraw

def draw_bbox_on_image(image_path, bbox_list, output_path):
    image = Image.open(image_path)
    draw = ImageDraw.Draw(image)

    for bbox in bbox_list:
        x, y, w, h = bbox
        draw.rectangle([x, y, x + w, y + h], outline="red", width=2)

    image.save(output_path)

def process_coco_json(json_file, image_folder, output_folder):
    with open(json_file, "r") as f:
        data = json.load(f)

    if not os.path.exists(output_folder):
        os.makedirs(output_folder)

    image_id_to_bbox_list = {}
    for annotation in data["annotations"]:
        image_id = annotation["image_id"]
        bbox = annotation["bbox"]
        if image_id not in image_id_to_bbox_list:
            image_id_to_bbox_list[image_id] = []
        image_id_to_bbox_list[image_id].append(bbox)

    for image in data["images"]:
        image_id = image["id"]
        image_path = os.path.join(image_folder, image["file_name"])
        output_path = os.path.join(output_folder, image["file_name"])

        if image_id in image_id_to_bbox_list:
            draw_bbox_on_image(image_path, image_id_to_bbox_list[image_id], output_path)

def main():
    train_json = "train.json"
    train_folder = "train"
    train_output_folder = "output/train"

    val_json = "val.json"
    val_folder = "val"
    val_output_folder = "output/val"

    process_coco_json(train_json, train_folder, train_output_folder)
    process_coco_json(val_json, val_folder, val_output_folder)

if __name__ == "__main__":
    main()