数据集批量更新分类OR删除分类行数据
import osdef remove_class_from_file(file_path, class_to_remove):"""从YOLO格式的标注文件中删除指定类别的行记录，并去除空行。:param file_path: YOLO标注文件路径:param class_to_remove: 需要删除的类别ID:return: 删除的行数"""deleted_lines = 0with open(file_path, 'r') as file:lines = file.readlines()filtered_lines = [line for line in lines if not line.startswith(f"{class_to_remove} ")]cleaned_lines = [line for line in filtered_lines if line.strip()]deleted_lines = len(lines) - len(cleaned_lines)with open(file_path, 'w') as file:file.writelines(cleaned_lines)return deleted_linesdef update_class_in_file(file_path, old_class, new_class):"""将YOLO格式的标注文件中指定类别的数字更新为新的类别ID。:param file_path: YOLO标注文件路径:param old_class: 需要更新的旧类别ID:param new_class: 更新后的新类别ID:return: 更新的行数"""updated_lines = 0with open(file_path, 'r') as file:lines = file.readlines()updated_lines = 0new_lines = []for line in lines:if line.startswith(f"{old_class} "):new_lines.append(line.replace(f"{old_class} ", f"{new_class} ", 1))updated_lines += 1else:new_lines.append(line)with open(file_path, 'w') as file:file.writelines(new_lines)return updated_linesdef process_directory(directory_path, class_to_remove=None, old_class=None, new_class=None):"""处理指定目录下的所有YOLO标注文件，可以选择删除指定类别的记录或更新分类数字。:param directory_path: 目录路径:param class_to_remove: 需要删除的类别ID（如果需要删除指定类别的记录）:param old_class: 需要更新的旧类别ID（如果需要更新分类数字）:param new_class: 更新后的新类别ID（如果需要更新分类数字）"""total_deleted_lines = 0total_updated_lines = 0for filename in os.listdir(directory_path):if filename.endswith('.txt'): file_path = os.path.join(directory_path, filename)if class_to_remove is not None:deleted_lines = remove_class_from_file(file_path, class_to_remove)print(f"已从文件 {filename} 中删除类别 {class_to_remove} 的记录，共删除 {deleted_lines} 行。")total_deleted_lines += deleted_linesif old_class is not None and new_class is not None:updated_lines = update_class_in_file(file_path, old_class, new_class)print(f"已从文件 {filename} 中将类别 {old_class} 更新为 {new_class}，共更新 {updated_lines} 行。")total_updated_lines += updated_linesif class_to_remove is not None:print(f"总共删除了 {total_deleted_lines} 行记录。")if old_class is not None and new_class is not None:print(f"总共更新了 {total_updated_lines} 行记录。")
directory_path = 'C:\\Users\\合格的牛马\\Desktop\\datasets\\test-delete\\labels'
class_to_remove = 5
process_directory(directory_path, class_to_remove=class_to_remove)
old_class = 1
new_class = 100
process_directory(directory_path, old_class=old_class, new_class=new_class)
)
 ＜详解＞)
)
