【数据集】批量命名YOLO数据集
【代码】【数据集】批量命名YOLO数据集。
·
记得修改 folder_path从1开始排号
批量命名图片
import os
def batch_rename_images(folder_path):
# 获取指定文件夹中的所有文件名
files = os.listdir(folder_path)
# 筛选出图片文件(可以根据需要扩展支持的图片格式)
image_extensions = ['.jpg', '.jpeg', '.png', '.bmp', '.gif', '.tiff']
image_files = [file for file in files if os.path.splitext(file)[1].lower() in image_extensions]
# 对图片文件按原始文件名排序(可选,根据需要调整排序方式)
image_files.sort()
# 初始化序号
index = 1
# 遍历图片文件并重命名
for image_file in image_files:
# 获取文件的扩展名
file_extension = os.path.splitext(image_file)[1]
# 构造新的文件名
new_file_name = f"{index}{file_extension}"
# 构造完整的旧文件路径和新文件路径
old_file_path = os.path.join(folder_path, image_file)
new_file_path = os.path.join(folder_path, new_file_name)
# 重命名文件
os.rename(old_file_path, new_file_path)
print(f"文件 {image_file} 已重命名为 {new_file_name}")
# 序号加1
index += 1
# 指定图片路径
folder_path = r"C:\Users\AnK\Desktop\seaclear\images\train"
batch_rename_images(folder_path)
批量命名.txt
import os
def batch_rename_txt_files(folder_path):
# 获取指定文件夹中的所有文件和文件夹名称
files_and_folders = os.listdir(folder_path)
# 筛选出所有的.txt文件
txt_files = [file for file in files_and_folders if file.endswith('.txt')]
# 按照文件名排序(如果需要按照文件创建时间或其他方式排序,可以修改排序方式)
txt_files.sort()
# 遍历.txt文件并重命名
for index, txt_file in enumerate(txt_files, start=1):
# 构造新的文件名
new_file_name = f"{index}.txt"
# 构造完整的旧文件路径和新文件路径
old_file_path = os.path.join(folder_path, txt_file)
new_file_path = os.path.join(folder_path, new_file_name)
# 重命名文件
os.rename(old_file_path, new_file_path)
print(f"文件 {txt_file} 已重命名为 {new_file_name}")
# 示例用法
# 替换为你的目标文件夹路径
folder_path = r"C:\Users\AnK\Desktop\seaclear\labels\train"
batch_rename_txt_files(folder_path)
魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。
更多推荐

所有评论(0)