
刚开始学计算机视觉要给数据集标注,一个一个改名字有点麻烦,所有用一下Python快速修改一下。
只需要修改一下图片路径,命名格式就可以了
python
import os
#图片路径
image_dir = "D:/Pycharm/yolo8_project/haican_dataset/labels/train"
#文本路径
label_dir = "D:/Pycharm/yolo8_project/haican_dataset/labels/train"
# 获取文件
image_files = [f for f in os.listdir(image_dir) if f.endswith(('.txt', '.jpg', '.png'))]#文件格式
# 批量重命名
for i, old_name in enumerate(image_files, 1):
# 新文件名
new_name = f"haican_{i:03d}"#要改成什么名字
# 重命名图片
old_image_path = os.path.join(image_dir, old_name)
new_image_path = os.path.join(image_dir, new_name + os.path.splitext(old_name)[1])
os.rename(old_image_path, new_image_path)
# 重命名对应的标注文件
old_label_name = os.path.splitext(old_name)[0] + ".txt"
old_label_path = os.path.join(label_dir, old_label_name)
new_label_path = os.path.join(label_dir, new_name + ".txt")
if os.path.exists(old_label_path):
os.rename(old_label_path, new_label_path)
print(f"重命名成功: {old_name} → {new_name}")
print("所有文件重命名完成!")