feat: REVERIE to R2R format

This commit is contained in:
Ting-Jun Wang 2023-11-06 15:52:54 +08:00
parent ab5010d32d
commit a5db597de5
Signed by: snsd0805
GPG Key ID: 48D331A3D6160354

37
adversarial_summary.py Normal file
View File

@ -0,0 +1,37 @@
import json
import os
for file in ['train', 'val_unseen', 'val_seen', 'train_seen', 'test', 'val_train_seen']:
print(file)
if os.path.isfile('data/adversarial/reverie_{}_fnf.json'.format(file)):
with open('data/adversarial/reverie_{}_fnf.json'.format(file)) as fp:
data = json.load(fp)
result = {}
for i in data:
instruction_id = i['path_id']
if instruction_id not in result:
result[instruction_id] = {
'distance': float(i['distance']),
'scan': i['scan'],
'path_id': int(i['path_id']),
'path': i['path'],
'heading': float(i['heading']),
'instructions': [ i['instruction'] ],
'swap': [ True if i['found'] else False ],
'id': i['id'],
'objId': i['objId']
}
else:
result[instruction_id]['instructions'].append(i['instruction'])
result[instruction_id]['swap'].append( True if i['found'] else False)
output = []
for k, item in result.items():
output.append(item)
else:
output = []
with open('data/adversarial/R2R_{}.json'.format(file), 'w') as fp:
json.dump(output, fp)