用python批量修改文件名称和修改图片分辨率

用python批量修改文件名称和修改图片分辨率

修改文件名称

import os
import sys


# 修改生成的文件名字
def rename():
    path = input("请输入路径(例如D:\\\\picture)：")
    name = input("请输入开头名:")
    startNumber = input("请输入开始数:")
    fileType = input("请输入后缀名（如 .jpg、.txt等等）:")
    print("正在生成以" + name + startNumber + fileType + "迭代的文件名")
    count = 0
    filelist = os.listdir(path)
    for files in filelist:
        Olddir = os.path.join(path, files)
        if os.path.isdir(Olddir):
            continue
        Newdir = os.path.join(path, name + str(count + int(startNumber)) + fileType)
        os.rename(Olddir, Newdir)
        count += 1
    print("一共修改了" + str(count) + "个文件")


rename()

修改图片分辨率（下方代码是.png改为分辨率512*512）也可以改为.jpg

import cv2
import os
def get_file_path_by_name(file_dir):
    '''
    获取指定路径下所有文件的绝对路径
    :param file_dir:
    :return:
    '''
    L = []
    for root, dirs, files in os.walk(file_dir):  # 获取所有文件
        for file in files:  # 遍历所有文件名
            if os.path.splitext(file)[1] == '.png':   # 指定尾缀  ***重要***
                L.append(os.path.join(root, file))  # 拼接处绝对路径并放入列表
    print('总文件数目：', len(L))
    return L

list_dir = get_file_path_by_name(r'D:\data\datainternet\train\masks')
# 1. 遍历所有png文件路径
for item in list_dir:
    # 2. 读取图片
    img_raw = cv2.imread(item)
    # 3. reszize (512,512)
    img_resize  = cv2.resize(img_raw,(512,512),interpolation=cv2.INTER_AREA)
    print(img_resize.shape) # 打印resize大小
    # 4. 获取图片名字，用于另存
    photo_name = os.path.split(item)[1]
    # 5. 获取图片的类别名词，下面代码用的 “\\”是windows系统下的路径分割符，在linux下不同
    class_name = item.split('\\')[-2]
    saved_file_dir = os.path.join('..','dataset_samesize',class_name)
    saved_file_path = os.path.join('..','dataset_samesize',class_name,photo_name)
    # 6.创建文件夹，否则cv2.imwrite找不到文件路径，无法保存
    if not os.path.exists(saved_file_dir):
       os.makedirs(saved_file_dir)
       print("目录:"+saved_file_dir+"创建成功！")
    print(saved_file_path)
    # 7.保存
    cv2.imwrite(saved_file_path, img_resize)