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