| import json | |
| input_path = "/root/huggingface/FoodieQA/FoodieQA/sivqa_tidy.json" # 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) |