python卡通化

卡通化

# -*- coding: utf-8 -*-
import cv2
import os

def cartoonise(picture_name):
    imgInput_FileName = picture_name
    imgOutput_FileName = "cartoonise_" + picture_name
    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_color, 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)
    #转换回彩色图像
    img_edge = cv2.cvtColor(img_edge, cv2.COLOR_GRAY2RGB)
    img_cartoon = cv2.bitwise_and(img_color, img_edge)
    # 保存转换后的图片
    cv2.imwrite(imgOutput_FileName, img_cartoon)

cartoonise('me.jpg')