|
import json |
|
input_path = "/root/huggingface/FoodieQA/FoodieQA/sivqa_tidy.json" |
|
output_path = "/root/huggingface/FoodieQA/FoodieQA/sivqa_llava.json" |
|
|
|
with open(input_path, 'r', encoding='utf-8') as f: |
|
foodqa = json.load(f) |
|
|
|
llava_format = [] |
|
for data in foodqa: |
|
llava_format.append({ |
|
"image": data['food_meta']['food_file'], |
|
"conversations": [ |
|
{ |
|
"from": "human", |
|
"value": data['question']+"\n<image>" |
|
}, |
|
{ |
|
"from": "gpt", |
|
"value": data['choices'][int(data['answer'])] + ",图中的菜是"+ data['food_meta']['food_name'] |
|
} |
|
] |
|
}) |
|
|
|
with open(output_path, 'w', encoding='utf-8') as f: |
|
json.dump(llava_format, f, indent=4, ensure_ascii=False) |