datasets/fix_handal_videoname.py

import json
from pycocotools.coco import COCO
from tqdm import tqdm
import string



text_pth  = "/data/work-gcp-europe-west4-a/yuqian_fu/datasets/HANDAL/test_json/handal_dataset_mugs_test.json"
save_path = "/data/work-gcp-europe-west4-a/yuqian_fu/datasets/HANDAL/test_json/handal_dataset_mugs_test_videoname.json"

new_data = []

with open(text_pth, "r") as fp:
   datas = json.load(fp)



# data是一帧帧图片
for data in datas:
    video_name = data["image"].split("/")[0]
    data["video_name"] = video_name
    new_data.append(data)


print(len(new_data))
with open(save_path, "w") as fp:
    json.dump(new_data, fp)