平衡coco格式json文件中的空目标图像和包含目标图像比例为1:1

import json
import random


def balance_empty_targets(coco_json_file, output_file):
    with open(coco_json_file, 'r') as file:
        data = json.load(file)

    image_ids_with_annotations = set(anno['image_id'] for anno in data['annotations'])

    images_with_targets = [image for image in data['images'] if image['id'] in image_ids_with_annotations]
    images_without_targets = [image for image in data['images'] if image['id'] not in image_ids_with_annotations]

    num_to_remove = len(images_without_targets) - len(images_with_targets)
    if num_to_remove > 0:
        images_to_remove = random.sample(images_without_targets, num_to_remove)
        images_without_targets = [image for image in images_without_targets if image not in images_to_remove]

    balanced_images = images_with_targets + images_without_targets

    # Update 'images' key with the new list of balanced images
    data['images'] = balanced_images

    with open(output_file, 'w') as file:
        json.dump(data, file)


if __name__ == '__main__':
    coco_json_file = './train.json'
    output_file = './balanced_coco.json'
    balance_empty_targets(coco_json_file, output_file)

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

打赏作者

大鹏要高飞

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

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

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

打赏作者

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

抵扣说明:

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

余额充值