| 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) |