1 # -*- coding: utf-8 -*-
2 """
3 Created on Sun Feb 4 12:15:38 2018
4
5 @author: markli
6 """
7 from PIL import Image;
8 import os;
9
10 class Mirror:
11 def __init__(self):
12 #self.path = path;
13 self.formats = ['.png','.jpg','.jpeg','.bmp','.gif'];
14
15 def ImageMirror(self,fp,savepath):
16 filepath = fp;
17 f_format = '';
18 if(os.path.exists(filepath) == False):
19 print("路径不存在");
20 return;
21 f_format = os.path.splitext(filepath)[1].lower();
22 if(f_format not in self.formats):
23 print("图片格式不正确");
24 return;
25
26 img = Image.open(filepath);
27 img_pixel = img.load();
28 mirror = Image.new(img.mode,img.size,"white");
29
30 width, height = img.size;
31 """水平镜像转换，遍历每个像素点，将后列变前列"""
32 for y in range(height):
33 for x in range(width):
34 pixel = img_pixel[width-1-x,y];
35 mirror.putpixel((x,y),pixel);
36
37 sp,f = os.path.splitext(savepath);
38 if(f != f_format):
39 savepath = sp + f_format;
40 mirror.save(savepath);
41
42 def TranslateAll(self,filedir,savedir):
43 """
44 将目标图片集全部进行镜像处理
45 filedir 目标图片集所在的文件夹路径
46 savedir 镜像图片保存的文件夹路径
47 """
48 filelist = self.Getfile(filedir);
49 if(os.path.exists(savedir) == True):
50 print("保存路径已存在，请重新设定路径");
51 return;
52 os.mkdir(savedir);
53 for f in filelist:
54 fn,ext = os.path.splitext(os.path.split(f)[1]);
55 fn = fn + "mirror"; #给定镜像图片的名称
56 filename = fn + ext; #镜像图片与原图具有相同的扩展名
57 savefile = os.path.join(savedir,filename); #构造出完整的保存路径
58 self.ImageMirror(f,savefile);
59
60 def Getfile(self,filedir):
61 """获得文件夹filedir目录下所有的文件路径"""
62 filepath = [];
63 if(os.path.exists(filedir) == False):
64 print("路径不存在");
65 return filepath;
66 if(os.path.isdir(filedir) == False):
67 print("该路径不是文件夹");
68 return filepath;
69 filelist = os.listdir(filedir);
70
71 for f in filelist:
72 f = os.path.join(filedir,f);
73 if(os.path.isfile(f) == True):
74 filepath.append(f);
75 elif(os.path.isdir(f) == True):
76 filepath.extend(self.Getfile(f));
77 else:
78 continue;
79 return filepath;
80
81 #fp = "C:\\Users\\yangp\\Desktop\\python_b_blue.jpg";
82 #m = Mirror();
83 #savep = "C:\\Users\\yangp\\Desktop\\python_b_blue_m.jpg";
84 #m.ImageMirror(fp,savep);
85
86 filedir = "C:\\Users\\yangp\\Desktop\\mirror";
87 savedir = "C:\\Users\\yangp\\Desktop\\mirror2";
88 m = Mirror();
89 m.TranslateAll(filedir,savedir);
