json转txt yolo
时间: 2023-09-09 21:08:56 浏览: 154
你可以使用Python中的`json`和`txt`库来实现将JSON格式转换为Yolo格式的TXT文件。以下是一个示例代码:
```python
import json
def convert_json_to_yolo(json_file, txt_file):
with open(json_file, 'r') as f:
data = json.load(f)
with open(txt_file, 'w') as f:
for obj in data['objects']:
label = obj['label']
x, y, w, h = obj['bbox']
# 转换坐标为Yolo格式
x_center = x + w / 2
y_center = y + h / 2
width = w
height = h
# 将转换后的信息写入TXT文件
line = f"{label} {x_center} {y_center} {width} {height}\n"
f.write(line)
# 示例调用
json_file = 'input.json'
txt_file = 'output.txt'
convert_json_to_yolo(json_file, txt_file)
```
请将上述代码中的`json_file`替换为你的JSON文件路径,将`txt_file`替换为你想要保存Yolo格式结果的TXT文件路径。运行代码后,它将从JSON文件中提取对象的标签和边界框信息,并将它们转换为Yolo格式的TXT文件。
阅读全文