数据集中遇到文件损坏报错,有可能是文件命名不支持中文。可用以下脚本改为数字顺序命名
代码
import os
def rename_files_in_directory(directory):
# 获取目录中的所有文件
files = [f for f in os.listdir(directory) if os.path.isfile(os.path.join(directory, f))]
# 按文件名排序
files.sort()
# 重命名文件
for index, filename in enumerate(files):
file_extension = os.path.splitext(filename)[1]
new_name = f"{index + 1}{file_extension}"
os.rename(os.path.join(directory, filename), os.path.join(directory, new_name))
print(f"Renamed {filename} to {new_name}")
# 指定要处理的目录路径
directory_path = 'zhongyao_cls/training_set/SM'
rename_files_in_directory(directory_path)
评论