源代码如下:#-*-coding:utf-8-*-importcv2importosdefcartoonise(picture_name):imgInput_FileName=picture_nameimgOutput_FileName="F:\\K\\M"+picture_namenum_down=2#缩减像素采样...
源代码如下:
# -*- coding: utf-8 -*-
import cv2
import os
def cartoonise(picture_name):
imgInput_FileName = picture_name
imgOutput_FileName = "F:\\K\\M" + 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_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)
#转换回彩色图像
img_edge = cv2.cvtColor(img_edge, cv2.COLOR_GRAY2RGB)
img_cartoon = cv2.bitwise_and(img_color, img_edge)
# 保存转换后的图片
cv2.imwrite(imgOutput_FileName, img_cartoon)
ImageList = [] #建立空的List
#循环读取"D:\pythonpractice\Image"中的文件名
for filename in os.listdir(r"F:\K"):
ImageList.append(filename) #将文件名添加到ImageList
for i in ImageList: #循环读取ImageList中的文件名,将其进行卡通化处理
print("正在卡通化" + i)
cartoonise(i)
报错如下:
Traceback (most recent call last):
File "F:/python/cartoon.py", line 46, in
cartoonise(i)
File "F:/python/cartoon.py", line 17, in cartoonise
img_color = cv2.pyrDown(img_color)
cv2.error: OpenCV(4.1.0) C:\projects\opencv-python\opencv\modules\imgproc\src\pyramids.cpp:729: error: (-215:Assertion failed) !_src.empty() in function 'cv::pyrDown_'
展开