taskswithcode/DeticChatGPT

1# Copyright (c) Facebook, Inc. and its affiliates.2import argparse3import json4import path5import os6 7if __name__ == '__main__':8    parser = argparse.ArgumentParser()9    parser.add_argument("--ann", default='datasets/objects365/annotations/zhiyuan_objv2_train_fixname.json')10    parser.add_argument("--img_dir", default='datasets/objects365/train/')11    args = parser.parse_args()12 13    print('Loading', args.ann)14    data = json.load(open(args.ann, 'r'))15    images = []16    count = 017    for x in data['images']:18        path = '{}/{}'.format(args.img_dir, x['file_name'])19        if os.path.exists(path):20            images.append(x)21        else:22            print(path)23            count = count + 124    print('Missing', count, 'images')25    data['images'] = images26    out_name = args.ann[:-5] + '_fixmiss.json'27    print('Saving to', out_name)28    json.dump(data, open(out_name, 'w'))29