File size: 1,970 Bytes
944cdc2 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 | import os
import json
import argparse
import tqdm
def preprocess_annotations(data_path, annotation_path, split, output_path):
# load split
with open("/home/yuqian_fu/Projects/ego-exo4d-relation/correspondence/split.json", 'r') as fp:
splits = json.load(fp)
with open(annotation_path, "r") as fp:
gt_annotations = json.load(fp)['annotations']
takes = splits[split]
annotations = {"version": "xx",
"challenge": "xx",
"annotations": {}}
for take in tqdm.tqdm(takes):
with open(f'{data_path}/{take}/annotation.json', 'r') as fp:
vid_anno = json.load(fp)
annotations["annotations"][take] = vid_anno
annotations["annotations"][take]["annotated_frames"] = {}
for obj in vid_anno['masks'].keys():
annotations["annotations"][take]["annotated_frames"][obj] = {}
for cam in vid_anno['masks'][obj].keys():
annotations["annotations"][take]["annotated_frames"][obj][cam] = gt_annotations[take]['object_masks'][obj][cam]['annotated_frames']
# breakpoint()
with open(output_path, 'w') as fp:
json.dump(annotations, fp)
if __name__ == '__main__':
parser = argparse.ArgumentParser()
parser.add_argument("--data_path", type=str, required=True,
help="Path to directory with processed dataset")
parser.add_argument("--annotations_path", type=str, required=True,
help="Path to annotations file from EgoExo4D e.g. `relations_test.json`")
parser.add_argument("--split", type=str, required=True,
help="Split on which to evaluate. train/test/val")
parser.add_argument("--output_path", type=str, required=True,
help="Path to the output processed annotations")
args = parser.parse_args()
preprocess_annotations(args.data_path, args.annotations_path, args.split, args.output_path) |