YOLO同时对图片和标签进行重命名

2023-12-23 17:43:35
import os

image_folder = '/path/to/image/folder'  # 图像文件夹路径
label_folder = '/path/to/label/folder'  # 标签文件夹路径

image_files = sorted(os.listdir(image_folder))  # 获取图像文件列表并排序
label_files = sorted(os.listdir(label_folder))  # 获取标签文件列表并排序

# 确保图像和标签文件数量相同
assert len(image_files) == len(label_files)

# 遍历文件列表
for i in range(len(image_files)):
    old_image_path = os.path.join(image_folder, image_files[i])  # 原始图像文件路径
    old_label_path = os.path.join(label_folder, label_files[i])  # 原始标签文件路径

    new_image_name = f'image_{i}.jpg'  # 新图像文件名
    new_label_name = f'label_{i}.txt'  # 新标签文件名

    new_image_path = os.path.join(image_folder, new_image_name)  # 新图像文件路径
    new_label_path = os.path.join(label_folder, new_label_name)  # 新标签文件路径

    # 重命名图像文件
    os.rename(old_image_path, new_image_path)

    # 重命名标签文件
    os.rename(old_label_path, new_label_path)

文章来源:https://blog.csdn.net/weixin_43722052/article/details/135170709
本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。