Python PIL创建文字图片

#-*- coding:gb2312 -*-
from PIL import Image,ImageDraw,ImageFont,ImageOps
import numpy as np
import random


class LetterImage():
    
    def __init__(self,fontFile=‘‘,imgSize=(0,0),imgMode=‘RGB‘,bg_color=(0,0,0),fg_color=(255,255,255),fontsize=20):
        self.imgSize = imgSize
        self.imgMode = imgMode
        self.fontsize = fontsize
        self.bg_color = bg_color
        self.fg_color = fg_color
#         self.font = ImageFont.load(‘车牌字体.ttf‘)
        if ‘‘==fontFile:
            self.font = ImageFont.truetype(‘DIN1451.ttf‘, fontsize)
        else:
            self.font = ImageFont.truetype(fontFile,fontsize)


    def GenLetterImage(self,letters):
        ‘‘‘Generate the Image of letters‘‘‘ 
        self.letters = letters
        (self.letterWidth,self.letterHeight) = self.font.getsize(letters)
        if self.imgSize==(0,0):
            self.imgSize=(self.letterWidth+2,self.letterHeight+2)
        self.imgWidth,self.imgHeight=self.imgSize
        self.img = Image.new(self.imgMode, self.imgSize, self.bg_color)
        self.drawBrush = ImageDraw.Draw(self.img)
        textY0 = (self.imgHeight-self.letterHeight+1)/2
        textY0 = int(textY0)
        textX0 = int((self.imgWidth-self.letterWidth+1)/2)
        print ‘text location:‘,(textX0,textY0)
        print ‘text size (width,height):‘,self.letterWidth,self.letterHeight
        print ‘img size(width,height):‘,self.imgSize
#         if textX0<0 or textY0<0:
#             raise Exception(‘size error text location x0:%d,y0:%d‘%(textX0,textY0))
        self.drawBrush.text((textX0,textY0), self.letters, fill=self.fg_color,font=self.font)
    
    def SaveImg(self,saveName=‘‘):
        if ‘‘==saveName.strip():
            saveName = str(self.letters.encode(‘gb2312‘))+‘.png‘
        fileName,file_format = saveName.split(‘.‘)
        fileName+=‘_‘+str(self.fontsize)+‘.‘+file_format
        print fileName,file_format
        self.img.save(fileName, file_format)
    
    def Show(self):
        self.img.show()

def clearpictures():
    import os
    png = os.listdir(os.curdir)
    for i in png:
        if os.path.splitext(i)[1]==".png":
            os.remove(i)
            

if __name__==‘__main__‘:
    letterList = []
    letterList.append(LetterImage(bg_color=(0,0,255),fontsize=10))
    letterList.append(LetterImage(fontFile=‘‘,bg_color=(0,0,255),fontsize=400))
    letter=[u‘u‘,u‘v‘]
    num_letter = 2

    svd_u=[]
    svd_s=[]
    svd_v=[]
    import cv2
    mergeImg = np.zeros((470,444))
    npareiImg =[]
    for i in range(num_letter):
        letterList[i].GenLetterImage(letter[i])
#         letterList[i].Show()
#         letterList[i].SaveImg()
        grayImg = ImageOps.grayscale(letterList[i].img)
        grayImg = grayImg.resize((222,470),resample=Image.BICUBIC)
        npareiImg.append( np.asarray(grayImg))
        cv2.namedWindow(‘%s‘%i)
        cv2.imshow(‘%s‘%i, npareiImg[i])
        mergeImg[0:470,i*222:(i+1)*222]=npareiImg[i]
        u,s,v=np.linalg.svd(npareiImg[i])
        print ‘u and img \‘s shape‘,u.shape,npareiImg[i].shape
        svd_u.append(u)
        svd_v.append(v)
        svd_s.append(s)
        
#     mergeImgNp=Image.fromarray(mergeImg)#, mode)
#     mergeImgNp.show()
    uDifNorm=np.linalg.norm(svd_u[0]-svd_u[1])
    print uDifNorm
    vDifNorm = np.linalg.norm(svd_v[0]-svd_v[1])
    print vDifNorm
    sDifNorm = np.linalg.norm(svd_s[0]-svd_s[1])
    print sDifNorm 
    ou_norm = np.linalg.norm(np.asarray(npareiImg[0])-np.asarray(npareiImg[1]))
    print ou_norm
    f=open(‘record.txt‘,‘a‘)
    lines=[]
    lines.append(‘letters: %s,%s‘%(letter[0],letter[1]))
    lines.append(‘SVD u diff norm:\t%f‘%uDifNorm)
    lines.append(‘SVD v diff norm:\t%f‘%vDifNorm)
    lines.append(‘SVD s diff norm:\t%f‘%sDifNorm)
    lines.append(‘Ou norm:        \t%f‘%ou_norm)
    str_to_write=‘\n‘.join(lines)+‘\n‘
    print str_to_write
    f.write(str_to_write)
    f.close()
    cv2.waitKey()