coco集的json文件随机转换yolo集的txt文件

from pycocotools.coco import COCO
import numpy as np
import tqdm
import argparse
import random
def arg_parser():
    parser = argparse.ArgumentParser('code by rbj') #建立解析对象
    parser.add_argument('--annotation_path', type=str,
                        default=r'D:\BaiduNetdiskDownload\stuff_annotations_trainval2017\stuff_train2017.json')#存放json文件的路,需要修改
    # 给parser实例增加三个属性
    #生成的txt文件保存的目录
    parser.add_argument('--save_base_path', type=str, default='C:/Users/python_Mooc/coco_data_for_yolo/')##存放数据的文件路径需要修改
    args = parser.parse_args()
    # 把parser中设置的所有"add_argument"给返回到args子类实例当中
    # 那么parser中增加的属性内容都会在args实例中，使用即可。
    return args

if __name__ == '__main__':
    args = arg_parser()
    annotation_path = args.annotation_path
    save_base_path = args.save_base_path

    data_source = COCO(annotation_file=annotation_path)
    catIds = data_source.