贴些图片处理的代码

ghostwwl

好久没来CU贴代码了  贴一段  买了相机后  一直写给自己用的

呵呵  用的最多的 还是批量加水印

1. 
   
2. #!/usr/bin/env python
   
3. #-*- coding:UTF-8 -*-
   
4. 
   
5. #****************************
   
6. #  Author: ghostwwl
   
7. #  Email: [email]ghostwwl@gmail.com[/email]
   
8. #  License: GPL
   
9. #****************************
   
10. 
    
11. import os
    
12. import sys
    
13. import Image
    
14. import ImageDraw
    
15. import ImageFont
    
16. from opencv.cv import *
    
17. from opencv.highgui import *
    
18. import win32ui, win32con
    
19. 
    
20. logo = os.path.join(os.path.dirname(sys.argv[0]), 'logo.jpg')
    
21. 
    
22. def mImageROI(infile, outfile, zoom, point, width, height):
    
23.     '''
    
24.     zoom:缩放倍数
    
25.     infile:原始文件
    
26.     outfile:处理生成的文件
    
27.     point 起始点坐标
    
28.     width, height: 截取区域的宽高
    
29.     '''
    
30.     try:
    
31.         if not os.path.exists(infile):
    
32.             return "No infile!"
    
33.         im = cvLoadImage(infile, 1)
    
34.         cw, ch = int(im.width*zoom), int(im.height*zoom)
    
35.         fm = cvCreateImage(cvSize(cw, ch), im.depth, im.nChannels)
    
36.         cvSetZero(fm)
    
37.         del cw,ch
    
38.         cvResize(im, fm)
    
39.         cv.cvRelease(im)
    
40.         del im
    
41.         om = cvCreateImage(cvSize(int(point[0] + width),
    
42.             int(point[1] + height)), fm.depth, fm.nChannels)
    
43.         cvSetZero(om)
    
44.         rcenter = cvPoint2D32f(px1 + width/2.0, py1 + height/2.0)
    
45.         cvGetRectSubPix(fm, om, rcenter)
    
46.         del rcenter
    
47.         cvSaveImage(outfile, om)
    
48.         cv.cvReleaseImage(fm)
    
49.         del fm
    
50.         cv.cvReleaseImage(om)
    
51.         del om  
    
52.         print "ok"
    
53.     except Exception, e:
    
54.         return "ROI Error: %s" % str(e)
    
55. 
    
56. def fupian(infile, outfile):
    
57.     def base_1(data1,data2):
    
58.         x1, y1 = data1
    
59.         x2, y2 = data2
    
60.         x1, x2, y1, y2 = float(x1), float(x2), float(y1), float(y2)
    
61.         a = (y2 - y1)/(x2 * x2 - x1 * x1)
    
62.         b = y1 - a * x1 * x1
    
63.         return a, b
    
64. 
    
65.     def base_2(i, points):
    
66.         points_length = len(points)
    
67.         for j in range(points_length):
    
68.             now_point = points[j]
    
69.             next_point = points[j+1]
    
70.             x1, x2 = now_point[0],next_point[0]
    
71.             if x1 <= i <= x2:
    
72.                 a, b = base_1(now_point, next_point)
    
73.                 return a * i * i + b
    
74.    
    
75.     def smoonth(i):
    
76.         points=[(0, 0), (73, 47), (179, 213), (255, 255)]
    
77.         return base_2(i, points)
    
78.             
    
79.     def smoonth_r(i):
    
80.         points=[(0, 0), (91, 22), (192, 149), (255, 232)]
    
81.         return base_2(i, points)
    
82.    
    
83.     def smoonth_g(i):
    
84.         points=[(0, 21), (70, 90), (189, 151), (255, 212)]
    
85.         return base_2(i,points)
    
86.    
    
87.     def smoonth_b(i):
    
88.         points=[(0, 2), (45, 68), (205, 147), (255, 255)]
    
89.         return base_2(i,points)
    
90.    
    
91.     try:
    
92.         im = Image.open(infile)
    
93.         R, G, B = im.split()
    
94.         R = R.point(smoonth_r)
    
95.         G = G.point(smoonth_g)
    
96.         B = B.point(smoonth_b)
    
97.         new = Image.merge('RGB', (R, G, B))
    
98.         new = new.point(smoonth)
    
99.         new.save(outfile)
    
100.     except Exception, e:
     
101.         print str(e)
     
102. 
     
103. def WaterMark(fil, dh='y'):
     
104.     '''
     
105.     第一个参数是 图片的完整路径
     
106.     第二个参数表示是否对logo图像进行透明处理
     
107.     PIL简单 就PIL OpenCV虽然速度快 写起来也多要好多行啊
     
108.     '''
     
109.     img1 = Image.open(fil).convert("RGB")
     
110.     logoimg = Image.open(logo).convert("RGBA")
     
111.    
     
112.     #这里是针对
     
113.     w2 = (img1.size[0] - 10)/5
     
114.     h2 = w2*logoimg.size[1]/logoimg.size[0]
     
115.     logoimg = logoimg.resize((w2, h2), Image.ANTIALIAS)
     
116.    
     
117.     #在属主图像中生成嵌入图像大小的box
     
118.     box = (img1.size[0] - w2 - 20, img1.size[1] - h2 - 20)
     
119.     if 'y' in dh:
     
120.         Mask = logoimg.convert("L").point(lambda x: min(x, 100))
     
121.         logoimg.putalpha(Mask)
     
122.     img1.paste(logoimg, box, logoimg)
     
123.     img1.save(fil.split(".")[0]+"_W.jpg")
     
124. 
     
125. #--------------------------------------------------------------
     
126. def WriteText(im, text = None, align = 0):
     
127.     '''
     
128.     align 表示对齐方式 1 为右下脚 0 为底部居中
     
129.     text 表示要写到图片上的文字
     
130.     如果需要在linux下用把请修改字体路径和字体,
     
131.     然后确保安装了python和pil模块
     
132.    
     
133.     这里没有用OpenCV是因为 官方库 暂时不支持中文
     
134.     如果要用中文  就要C++ 用国内爱好这开发的 第三方扩展了
     
135.     所以 直接PIL了
     
136.     '''
     
137.     if not text:
     
138.         text = 'Email:ghostwwl@gmail.com'   
     
139.    
     
140.     img = Image.open(im).convert("RGB")
     
141.     Draw = ImageDraw.ImageDraw(img, "RGBA")
     
142.    
     
143.     #自动调整字体大小
     
144.     size = 0
     
145.     while True:
     
146.         size += 1
     
147.         #如果在linux下 请去/usr/share/fonts下找个truetype吧　支持中文的字体
     
148.         font = ImageFont.truetype("C:\\Windows\\Fonts\\STXINWEI.TTF", size)
     
149.         tw, th = font.getsize(text)
     
150.         if tw > img.size[0]/5:
     
151.             break
     
152.     if align == 1:
     
153.         w1 = (img.size[0] - tw)/2
     
154.     else:
     
155.         w1 = img.size[0]*0.78
     
156.     h1 = img.size[1] - th -  10
     
157.    
     
158.     Draw.setfont(font)
     
159.     Draw.text((w1, h1), unicode(text, "utf-8"))
     
160.     img.save(im.split(".")[0]+"_Text.jpg")
     
161. 
     
162. #--------------------------------------------------------------
     
163. def GetFiles():
     
164.     openflag = "JPEG File (*.jpg)|*.jpg|ALL File (*.*)|*.*|"
     
165.     filedialog = win32ui.CreateFileDialog(1, None, None, win32con.OFN_ALLOWMULTISELECT|win32con.OFN_HIDEREADONLY, openflag)
     
166.     filedialog.SetOFNTitle("Select files to Work")
     
167.     filedialog.DoModal()
     
168.    
     
169.     return filedialog.GetPathNames()
     
170. 
     
171. if __name__ == '__main__':
     
172.     #这里需要那些函数  自己改写  还有的东西没加进来
     
173.     #会继续完善 想加入 人脸库 从海量照片中快速定位 包含自己的照片  和包含人脸的照片
     
174.     #有时间  就写一点  一点点的爬啊
     
175. #    f = GetFiles()
     
176.     import glob
     
177.     import time
     
178.     pw = raw_input("Enter The JPEG File Path:")
     
179.     fs = glob.glob(os.path.join(pw, '*.jpg'))
     
180.     for i in fs:
     
181. #        WriteText(i)
     
182.         if not i.endswith('W.jpg') or not i.endswith('W.JPG'):
     
183.             WaterMark(i)
     
184.             print "%s Water Img: %s OK!" % (time.strftime("[%Y-%m-%d %H:%M:%S]"), i)
     

复制代码

ghostwwl

说明下 那个产生负片下过的 那个函数是 google到的  然后改一点 封装成了一个fupian函数

ghostwwl

特别说明 使用OpenCV要特小心  因为图片内存区域 都是手动获取 和释放的

如果弄错 很容出现python虚拟机 直接 给你的down掉  错误是很严重

毕竟是C过来的库  小心用来着 功能是超牛X的视觉库