import json


test_data = json.load(open("reformat_test.json", 'r'))

new_data = {}

for d in test_data:
    # image1 = d["img0"]
    # image2 = d["img1"]
    # image_id = "%s_%s" % (image1, image2)
    image_id = d["uid"]
    new_data[image_id] = d["sents"]

json.dump(new_data, open("reformat_test_all.json", "w"))