| import json, copy, ipdb, os |
|
|
| def process(): |
| cnt = 1228428 |
| for i in range(64, 65): |
| f = open('./ubody' + str(i) + '.json', 'r') |
| data = json.load(f) |
| all = [] |
| for item in data: |
| cur = copy.deepcopy(item) |
| cur['id'] = str(cnt) |
| cur['image'] = "/data7tb/gzy/PoseLLM/Data/" + cur['image'].replace('Ubody/images', 'UBody_images').replace("ConferenceVideoSegmentation/", "").replace("YT/VideoConference", "VideoConference").replace('LifestyleVLOG/', '').replace('YT/LiveVlog_', 'LiveVlog_').replace('SoundOfPixels/', '').replace('YT/ConductMusic_', 'ConductMusic_').replace('Speech_S1/', '').replace('Speech_S2/', '').replace('Speech_S3/', '').replace('Speech_S4/', '').replace('TVShow_S1/TVShow_S1/', 'TVShow_S1/').replace('TVShow_S2/TVShow_S2/', 'TVShow_S2/').replace('TVShow_S3/TVShow_S3/', 'TVShow_S3/').replace('TVShow_S4/TVShow_S4/', 'TVShow_S4/').replace('TVShow_S5/TVShow_S5/', 'TVShow_S5/').replace('TVShow_S6/TVShow_S6/', 'TVShow_S6/').replace('TVShow_S7/TVShow_S7/', 'TVShow_S7/').replace('TVShow_S8/TVShow_S8/', 'TVShow_S8/').replace('TVShow_S9/TVShow_S9/', 'TVShow_S9/').replace('TVShow_S10/TVShow_S10/', 'TVShow_S10/').replace('TVShow_S11/TVShow_S11/', 'TVShow_S11/').replace('TVShow_S12/TVShow_S12/', 'TVShow_S12/') |
| if not os.path.exists(cur['image']): |
| print("不存在:", cur['image']) |
| print(i, cnt) |
| exit(1) |
| cur['conversations'][0]['value'] = "Please provide the SMPL parameters of the single person in the image." |
| root_pose = cur['conversations'][1]['root_pose'] |
| body_pose = cur['conversations'][1]['body_pose'] |
| shape = cur['conversations'][1]['shape'] |
| answer = "Sure. Here are the SMPL parameters of this person:\n" |
| answer += "root_pose:" + str(root_pose) + '\n' |
| answer += "body_pose:" + str(body_pose) + '\n' |
| answer += "shape:" + str(shape) |
| cur['conversations'][1]['value'] = answer |
| all.append(cur) |
| cnt += 1 |
| g = open('./text/UBody-' + str(i) + '.json', 'w+') |
| json.dump(all, g, indent=4) |
| print("finish", i) |
| i += 1 |
| print(cnt) |
|
|
| def spl(): |
| f = open('/data7tb/gzy/PoseLLM/Data/Ubody.json', 'r') |
| data = json.load(f) |
| i = 4 |
| while(i * 20000 < len(data)): |
| cur = data[i * 20000: (i + 1) * 20000] |
| g = open("/data7tb/gzy/PoseLLM/Data/UBody_json/ubody" + str(i + 1) + ".json", 'w') |
| json.dump(cur, g, indent=4) |
| print("Finish File", i + 1) |
| i += 1 |
|
|
| process() |
