【相关学习推荐:python视频教程】
本文实例为大家分享了python实现图片转换成素描和漫画格式的具体代码,供大家参考,具体内容如下
原图
图片转换后的成果
源码
# -*- coding: utf-8 -*- import cv2 from PIL import Image, ImageOps, ImageFilter # 转换成漫画风格 def toCarttonStyle(picturePath): # 设置输入输出路径和文件名称 imgInput_FileName = picturePath imgOutput_FileName = picturePath.split(".")[0] + '_cartoon.' + picturePath.split(".")[1] # 属性设置 num_down = 2 # 缩减像素采样的数目 num_bilateral = 7 # 定义双边滤波的数目 # 读取图片 img_rgb = cv2.imread(imgInput_FileName) # 用高斯金字塔降低取样 img_color = img_rgb for _ in range(num_down): img_color = cv2.pyrDown(img_color) # 重复使用小的双边滤波代替一个大的滤波 for _ in range(num_bilateral): img_color = cv2.bilateralFilter(img_color, d=9, sigmaColor=9, sigmaSpace=7) # 升采样图片到原始大小 for _ in range(num_down): img_color = cv2.pyrUp(img_color) # 转换为灰度并且使其产生中等的模糊 img_gray = cv2.cvtColor(img_rgb, cv2.COLOR_RGB2GRAY) img_blur = cv2.medianBlur(img_gray, 7) # 检测到边缘并且增强其效果 img_edge = cv2.adaptiveThreshold(img_blur, 255, cv2.ADAPTIVE_THRESH_MEAN_C, cv2.THRESH_BINARY, blockSize=9, C=2) # 算法处理后,照片的尺寸可能会不统一 # 把照片的尺寸统一化 height=img_rgb.shape[0] width = img_rgb.shape[1] img_color=cv2.resize(img_color,(width,height)) # 转换回彩色图像 img_edge = cv2.cvtColor(img_edge, cv2.COLOR_GRAY2RGB) img_cartoon = cv2.bitwise_and(img_color, img_edge) # 保存转换后的图片 cv2.imwrite(imgOutput_FileName, img_cartoon) print('文件转换成漫画成功,保存在' + imgOutput_FileName) # 透明度转换 素描转换的一部分 def dodge(a, b, alpha): # alpha为图片透明度 return min(int(a * 255 / (256 - b * alpha)), 255) # 图片转换为素描 def toSketchStyle(picturePath, blur=25, alpha=1.0): # 设置输入输出路径和文件名称 imgInput_FileName = picturePath imgOutput_FileName = picturePath.split(".")[0] + '_Sketch.' + picturePath.split(".")[1] # 转化成ima对象 img = Image.open(picturePath) # 将文件转成灰色 img1 = img.convert('L') img2 = img1.copy() img2 = ImageOps.invert(img2) # 模糊度 for i in range(blur): img2 = img2.filter(ImageFilter.BLUR) width, height = img1.size for x in range(width): for y in range(height): a = img1.getpixel((x, y)) b = img2.getpixel((x, y)) img1.putpixel((x, y), dodge(a, b, alpha)) # 保存转换后文件 img1.save(imgOutput_FileName) print('文件转换成漫画成功,保存在' + imgOutput_FileName) if __name__ == '__main__': imgInput_FileName = input('输入文件路径:') while True: print('1、漫画风格') print('2、素描风格') userChoose = input('请选择风格(输入序号即可):') if userChoose.__eq__('1'): toCarttonStyle(imgInput_FileName) break elif userChoose.__eq__('2'): toSketchStyle(imgInput_FileName) break else: print('违法输入(请输入序号)') break
相关学习推荐:编程视频
最新评论