Python生成数字图片代码分享

时间:2021-05-22

本文向大家分享了几段Python生成数字图片的代码,喜欢的朋友可以参考。具体如下:

最终版本

# -*- coding:utf-8 -*-from PIL import Image,ImageFont,ImageDraw,ImageFilterimport randomimport osimport timeclass Code(object): def __init__(self, imgSize=(35,35),\ fontSize=25, bgColor=(255,)*4, fontColor=(0,0,0)): self.imgSize = imgSize self.fontSize = fontSize self.bgColor = bgColor self.fontColor = fontColor def setFontSize(self, size): self.fontSize = size; def getDigit(self, digit): return str(digit) def getPannel(self): pannel = Image.new('RGBA',self.imgSize,self.bgColor) return pannel def getFont(self, fontFile='./Arial.ttf'): return ImageFont.truetype(fontFile, self.fontSize) def getTextPos(self, digit, font): text = self.getDigit(digit) textWidth,textHeight = font.getsize(text); imgWidth,imgHeight = self.imgSize textPos = ((imgWidth-textWidth)/2, (imgHeight-textHeight)/2) return textPos def rotateImg(self,image,angle=0, expand=0): rot = image.rotate(angle, expand) fff = Image.new('RGBA',rot.size,self.bgColor) image = Image.composite(rot, fff, rot) return image def createImg(self, digit, font, angle): codeImg = Image.new('RGBA',self.imgSize,self.bgColor) draw = ImageDraw.Draw(codeImg); text = self.getDigit(digit) textPos = self.getTextPos(digit, font) draw.text(xy=textPos,text=text,fill=self.fontColor,font=font) codeImg = self.rotateImg(codeImg,angle) return codeImg def saveImg(self, img, savePath, imgName): img.save(savePath+'/'+imgName)def createPath(path): if not os.path.exists(path): os.makedirs(path)def createImages(code,rootPath='./images',digitList=range(10), fontSizeList=range(18,30),\ angleList=[(45,90),(-45,45),(-45,-90)]): for index,angles in enumerate(angleList): if index==0: angleRange = '-90_-45' elif index == 1: angleRange = '-45_45' else: angleRange = '45_90' anglepath = os.path.join(rootPath, angleRange) createPath(anglepath) for digit in digitList: digitpath = os.path.join(anglepath, 'x'+str(digit)) createPath(digitpath) for size in fontSizeList: angle = round(random.uniform(angles[0], angles[1]),5) code.setFontSize(size) imgName = str(digit)+'_'+str(size)+'_'+str(angle)+'.jpg' img = code.createImg(digit, code.getFont(),angle) code.saveImg(img, digitpath, imgName)if __name__ == '__main__': imagesPath = './images' if os.path.exists(imagesPath): os.system('rm -rf '+imagesPath) os.mkdir(imagesPath) code = Code() for i in range(1000): createImages(code) # test ... # code = Code() # img = code.createImg(5,code.getFont(),0) # code.saveImg(img, savePath, 'test.jpg') # img.show() print 'hello'# -*- coding:utf-8 -*-from PIL import Image,ImageFont,ImageDraw,ImageFilterimport randomimport osclass Captcha(object): def __init__(self,size=(20,24),fontSize=20): self.font = ImageFont.truetype('./fonts/Arial.ttf',fontSize) self.size = size self.image = Image.new('RGBA',self.size,(255,)*4) # self.texts = self.randNum(1) self.text = '' def rotate(self, angle): # rot = self.image.rotate(random.randint(-10,10),expand=0) rot = self.image.rotate(angle,expand=0) fff = Image.new('RGBA',rot.size,(255,)*4) self.image = Image.composite(rot,fff,rot) def randColor(self): self.fontColor = (random.randint(0,250),random.randint(0,250),random.randint(0,250)) # def randNum(self,bits): # return ''.join(str(random.randint(0,9)) for i in range(bits)) def setNum(self, num): return num; def write(self,text,x,y): draw = ImageDraw.Draw(self.image) draw.text((x,y),text,fill=self.fontColor,font=self.font) def writeNum(self, num, angle): x = 2 y = -2 self.text = num self.fontColor = (0, 0, 0) self.write(num, x, y) self.rotate(angle) return self.text # character # xplus = 15 # for text in self.texts: # self.randColor() # self.fontColor = (0, 0, 0) # self.write(text, x, y) # self.rotate(angle) # self.rotate(random.randint(-10,10)) # x += xplus # return self.texts def save(self, save_path): # self.image.save('captcha.jpg') self.image.save(save_path)pic_root_path = './pic'if not os.path.exists(pic_root_path): os.mkdir(pic_root_path)angles = [(45,90),(-45,45),(-90,-45)]for i in range(10): pic_num_path = os.path.join(pic_root_path, 'pic'+str(i)) if not os.path.exists(pic_num_path): os.mkdir(pic_num_path) for angle_i in angles: angle_name = str(angle_i[0])+'_'+str(angle_i[1]) pic_angle_path = os.path.join(pic_num_path, angle_name) if not os.path.exists(pic_angle_path): os.mkdir(pic_angle_path) for angle in range(angle_i[0], angle_i[1]): for fontsize in range(25,28): img = Captcha(size=(20, 24), fontSize=fontsize) num = img.writeNum(str(i), angle) img_name = str(i)+'_'+str(fontsize)+'_'+str(angle)+'.bmp' save_path = os.path.join(pic_angle_path, img_name) img.save(save_path) # img = Captcha() # num = img.writeNum(str(i), random.randint(-90,-45)) # img_name = str(i)+'.jpg' # pic_path = './pic'+str(i) # if not os.path.exists(pic_path): # os.mkdir(pic_path) # save_path = os.path.join(pic_path, img_name) # save_path = os.path.join(pic_root_path, save_path) # img.save(save_path)# img.image.show()# img.save()

随机生成各种size和旋转角度的单个数字图片

# -*- coding:utf-8 -*-from PIL import Image,ImageFont,ImageDraw,ImageFilterimport randomimport osimport timeclass Captcha(object): def __init__(self,size=(20,24),fontSize=20): self.font = ImageFont.truetype('./fonts/Arial.ttf',fontSize) self.size = size self.image = Image.new('RGBA',self.size,(255,)*4) self.text = '' def rotate(self, angle): rot = self.image.rotate(angle,expand=0) fff = Image.new('RGBA',rot.size,(255,)*4) self.image = Image.composite(rot,fff,rot) def randColor(self): self.fontColor = (random.randint(0,250),random.randint(0,250),random.randint(0,250)) def setNum(self, num): return num; def write(self,text,x,y): draw = ImageDraw.Draw(self.image) draw.text((x,y),text,fill=self.fontColor,font=self.font) def writeNum(self, num, angle): x = 2 y = -2 self.text = num self.fontColor = (0, 0, 0) self.write(num, x, y) self.rotate(angle) return self.text def save(self, save_path): # self.image = self.image.filter(ImageFilter.EDGE_ENHANCE_MORE) #滤镜,边界加强 self.image.save(save_path)pic_root_path = './pic'if not os.path.exists(pic_root_path): os.mkdir(pic_root_path)angles = [(45,90),(-45,45),(-90,-45)]for i in range(10): pic_num_path = os.path.join(pic_root_path, 'x'+str(i)) if not os.path.exists(pic_num_path): os.mkdir(pic_num_path) for angle_i in angles: angle_name = str(angle_i[0])+'_'+str(angle_i[1]) pic_angle_path = os.path.join(pic_num_path, angle_name) if not os.path.exists(pic_angle_path): os.mkdir(pic_angle_path) for fontsize in range(25,29): for j in range(2500): # Keep 5 decimal places angle = round(random.uniform(angle_i[0], angle_i[1]),5) img = Captcha(size=(20, 24), fontSize=fontsize) num = img.writeNum(str(i), angle) img_name = 'x'+str(j)+'_'+str(fontsize)+'_'+str(angle)+'_'+str(num)+'.jpg' save_path = os.path.join(pic_angle_path, img_name) img.save(save_path)

文字居中

# -*- coding:utf-8 -*-from PIL import Image,ImageFont,ImageDraw,ImageFilterimport randomimport osimport timeimgWidth = 20imgHeight = 24fontSize = 28backGroundColor = (255,)*4fontColor = (0,)*3text = '0'font = ImageFont.truetype('./Arial.ttf', fontSize)codeimg = Image.new('RGBA',(imgWidth,imgHeight), backGroundColor)imagePath = './codes'if not os.path.exists(imagePath): os.mkdir(imagePath)textWidth, textHeight = font.getsize(text)textLeft = (imgWidth-textWidth)/2textTop = (imgHeight-textHeight)/2draw = ImageDraw.Draw(codeimg)draw.text(xy=(textLeft,textTop),text=text,fill=fontColor,font=font)rot = codeimg.rotate(90,expand=0)codeimg.rotatefff = Image.new('RGBA', rot.size,backGroundColor)codeimg = Image.composite(rot, fff, rot)codeimg.show()# codeimg.save('./codes/test.jpg')

以上就是本文关于Python生成数字图片代码分享的全部内容,希望对大家有所帮助。欢迎参阅:Python列表删除的三种方法代码分享Python文件的读写和异常代码示例等,有问题可以随时留言,欢迎大家交流讨论。

声明:本页内容来源网络,仅供用户参考;我单位不保证亦不表示资料全面及准确无误,也不保证亦不表示这些资料为最新信息,如因任何原因,本网内容或者用户因倚赖本网内容造成任何损失或损害,我单位将不会负任何法律责任。如涉及版权问题,请提交至online#300.cn邮箱联系删除。

相关文章