D:\360MoveData\Users\Administrator\Desktop\2023JYSwork\YOLO\smoke-yolov5-2\data\dataset\labels
时间: 2023-07-06 17:43:34 浏览: 215
好的,我已经明白了。以下是生成所有 ".txt" 文件中出现过的 label,并写入 labels.txt 文件中的 Python 代码:
```python
import os
labels_set = set()
# 获取所有 ".txt" 文件
txt_files = [f for f in os.listdir('D:/360MoveData/Users/Administrator/Desktop/2023JYSwork/YOLO/smoke-yolov5-2/data/dataset/labels') if f.endswith('.txt')]
# 遍历每个 ".txt" 文件,获取其中的 label
for txt_file in txt_files:
with open(os.path.join('D:/360MoveData/Users/Administrator/Desktop/2023JYSwork/YOLO/smoke-yolov5-2/data/dataset/labels', txt_file)) as f:
lines = f.readlines()
for line in lines:
label = line.split()[0]
labels_set.add(label)
# 将获取到的 label 写入 labels.txt 文件中
with open('labels.txt', 'w') as f:
for label in sorted(labels_set):
f.write(label + '\n')
print('labels.txt 文件已生成!')
```
请将以上代码保存为 `generate_labels.py` 文件,并将其放在与 `.txt` 文件相同的目录下,然后运行该文件即可生成 `labels.txt` 文件。
阅读全文
相关推荐

















