### 使用Python读取并修改YOLO格式的标签 对于处理YOLO格式的标签文件，通常这些文件是以`.txt`形式存在，每一行代表一个边界框，其内容由五个数值组成：类别ID、中心点X坐标比例、中心点Y坐标比例、宽度比例以及高度比例。为了实现对这类文件的操作，可以编写一段简单的Python程序来完成读取和修改的任务。 #### 代码示例 下面是一个具体的例子，展示了如何通过Python脚本来加载YOLO格式的TXT文件，并对其进行必要的更改： ```python def read_yolo_labels(file_path): with open(file_path, 'r') as file: lines = file.readlines() labels = [] for line in lines: values = list(map(float, line.strip().split())) class_id = int(values[0]) center_x = float(values[1]) center_y = float(values[2]) width = float(values[3]) height = float(values[4]) labels.append((class_id, center_x, center_y, width, height)) return labels def modify_and_save_yolo_labels(labels, output_file_path, modification_function): modified_lines = [] for label in labels: new_label = modification_function(label) formatted_line = " ".join([str(x) for x in new_label]) + "\n" modified_lines.append(formatted_line) with open(output_file_path, 'w') as file: file.writelines(modified_lines) # 修改函数的例子：增加所有物体的高度和宽度各5% def example_modification(label_tuple): class_id, cx, cy, w, h = label_tuple adjusted_w = min(w * 1.05, 1.0) # 确保不超过图像的最大尺寸 adjusted_h = min(h * 1.05, 1.0) return (class_id, cx, cy, adjusted_w, adjusted_h) input_file = './labels/input.txt' output_file = './labels/output_modified.txt' original_labels = read_yolo_labels(input_file) modify_and_save_yolo_labels(original_labels, output_file, example_modification) ``` 上述代码首先定义了一个`read_yolo_labels()`函数用来解析YOLO格式的TXT文件[^1]；接着是另一个名为`modify_and_save_yolo_labels()`的辅助函数负责应用给定的调整逻辑并将更新后的数据保存回新的TXT文件中。最后提供了一个简单的`example_modification()`作为示范性的修改操作——这里只是简单地增加了每个矩形框大小的比例。