目标检测TXT标签文件转成json格式的代码
时间: 2023-05-24 09:06:00 浏览: 55
以下是一个示例代码,可将目标检测的TXT标签文件转换为JSON格式:
```python
import json
def txt_to_json(txt_file_path):
# 打开 txt 文件并读取内容
with open(txt_file_path, 'r') as txt_file:
lines = txt_file.readlines()
# 解析每一行数据
labels = []
for line in lines:
line = line.strip().split()
label = {
'class_name': line[0],
'xmin': int(line[1]),
'ymin': int(line[2]),
'xmax': int(line[3]),
'ymax': int(line[4])
}
labels.append(label)
# 将标签数据转换为 JSON 格式
json_data = json.dumps(labels)
return json_data
```
使用示例:
```python
json_data = txt_to_json('labels.txt')
print(json_data)
```
输出:
```json
[
{
"class_name": "person",
"xmin": 10,
"ymin": 20,
"xmax": 100,
"ymax": 200
},
{
"class_name": "car",
"xmin": 50,
"ymin": 100,
"xmax": 200,
"ymax": 300
}
]
```
注意:该代码假定输入的TXT文件格式为以下格式:每一行数据表示一个目标,每行数据由五个部分组成,分别是目标的类别、最左侧点的x坐标、最上侧点的y坐标、最右侧点的x坐标和最下侧点的y坐标。例如:person 10 20 100 200。如果你的TXT文件格式不同,你需要在代码中做出相应的修改。