python 读取多个文件夹中的图片,将图片翻转后再保存到原来文件夹中

小白一个,很简单的问题自己折腾了很久,记录一下:

import cv2
import os
import glob

root_path=r'E:\YAO\94.1\facenet-master\train_data\new3d_data'
list_path=os.listdir(root_path)
sum=0
for k,names in enumerate(list_path):

    bmp_path=glob.glob(r'E:\YAO\94.1\facenet-master\train_data\new3d_data\{}\*bmp'.format(names))

    for i,path_bmp in enumerate(bmp_path):

        # (filename,extension) = os.path.splitext(path_bmp)
        a = os.path.split(path_bmp)
        name = os.path.basename(a[0])  # 人名
        # print(name)
        img = cv2.imread(path_bmp)
        flip=cv2.flip(img,1)
        cv2.imwrite(r'E:\YAO\94.1\facenet-master\train_data\new3d_data\{}\flip_{}{}.jpg'.format(names,name,i), flip)

        print(name,i)
        sum+=1

print('total pic',sum)

猜你喜欢

转载自blog.csdn.net/xingtianyao/article/details/88662387