解析train.json和val.json文件,删除json文件中空目标图像的信息,并删除对应文件夹下的空目标图像文件

import json
import os

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

    # Collect image IDs with annotations
    image_ids_with_annotations = set()
    for annotation in data["annotations"]:
        image_ids_with_annotations.add(annotation["image_id"])

    # Filter images with annotations
    new_images = [image for image in data["images"] if image["id"] in image_ids_with_annotations]

    # Collect empty image filenames
    empty_image_filenames = [image["file_name"] for image in data["images"] if image["id"] not in image_ids_with_annotations]

    # Update images in the JSON data
    data["images"] = new_images

    # Save the updated JSON data
    with open(json_file, "w") as f:
        json.dump(data, f)

    # Remove empty images from the image folder
    for filename in empty_image_filenames:
        file_path = os.path.join(image_folder, filename)
        if os.path.exists(file_path):
            os.remove(file_path)

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

    val_json = "val.json"
    val_folder = "val"

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

if __name__ == "__main__":
    main()

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

大鹏要高飞

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值