欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

python实现图片转换成素描和漫画格式

程序员文章站 2022-03-07 12:47:00
本文实例为大家分享了python实现图片转换成素描和漫画格式的具体代码,供大家参考,具体内容如下原图图片转换后的成果源码# -*- coding: utf-8 -*-import cv2from pi...

本文实例为大家分享了python实现图片转换成素描和漫画格式的具体代码,供大家参考,具体内容如下

原图

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

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持。

相关标签: python 图片转换