用Python OpenCV 一键把照片变成手绘素描,效果绝了!

# encoding=utf8'''Python-opencv批量将任意图片转化为素描图'''# 导入包import cv2import numpy as npimport os# 定义方法def dodgeNaive(image, mask):# 确定输入图像的形状	width, height = image.shape[:2]# 准备与图像大小相同的输出参数	blend = np.zeros((width, height), np.uint8)for col in range(width):for row in range(height):# 对每个像素执行此操作if mask[col, row] == 255:# 避免被零除				blend[col, row] = 255else:# 将图像像素值移位8位# 除以掩模的倒数				tmp = (image[col, row] << 8) / (255 - mask)# print('tmp={}'.format(tmp.shape)) # 调试代码# 确保结果值保持在范围内if tmp.any() > 255:					tmp = 255					blend[col, row] = tmpreturn blenddef dodgeV2(image, mask):return cv2.divide(image, 255 - mask, scale=256)def burnV2(image, mask):return 255 - cv2.divide(255 - image, 255 - mask, scale=256)def rgb_to_sketch(src_image_name, dst_image_name):	img_rgb = cv2.imread(src_image_name)	img_gray = cv2.cvtColor(img_rgb, cv2.COLOR_BGR2GRAY)# 读取图片时直接转换操作# img_gray = cv2.imread('example.jpg', cv2.IMREAD_GRAYSCALE)	img_gray_inv = 255 - img_gray	img_blur = cv2.GaussianBlur(img_gray_inv, ksize=(21, 21),	                            sigmaX=0, sigmaY=0)	img_blend = dodgeV2(img_gray, img_blur)# cv2.imshow('original', img_rgb) # 调试转换1# cv2.imshow('gray', img_gray) # 调试转换2# cv2.imshow('gray_inv', img_gray_inv) # 调试转换3# cv2.imshow('gray_blur', img_blur) # 调试转换4# cv2.imshow("pencil sketch", img_blend) # 调试转换5	cv2.waitKey(0)	cv2.destroyAllWindows()	cv2.imwrite(dst_image_name, img_blend)def imageSketch(imagepath, squarepath):'''读取图片并进行批量素描图转换'''	files = os.listdir(imagepath) # 读取指定目录中的所有文件for f in files:		file_p =  os.path.join(imagepath, f) # 图片原始位置		squar_p = os.path.join(squarepath, f) # 素描图保存位置if not os.path.exists(squarepath): 			os.mkdir(squarepath) # 如果目不存在就创建		rgb_to_sketch(file_p, squar_p)print(file_p, '==>' ,squar_p)if __name__ == '__main__':# 存放待处理图片的目录	imagepath = r'D:\Test\image'# 图片处理结果保存目录	squarepath = r'D:\Test\square'# 执行方法	imageSketch(imagepath, squarepath)# src_image_name = r'D:\Test\image\24e4 (1).jpeg'# dst_image_name = r'D:\Test\square\24e4 (1).jpeg'# rgb_to_sketch(src_image_name, dst_image_name)