QQ登录

只需要一步,快速开始

APP扫码登录

只需要一步,快速开始

手机号码,快捷登录

查看: 168|回复: 0

[Python] 图片转素描源码

[复制链接]

等级头衔

积分成就    金币 : 786
   泡泡 : 1488
   精华 : 6
   在线时间 : 786 小时
   最后登录 : 2020-7-3

丰功伟绩

优秀达人突出贡献荣誉管理论坛元老

联系方式
发表于 2020-6-4 21:42:32 | 显示全部楼层 |阅读模式
       本程序的功能是将彩色图片转成黑白素描手绘图,使用python编写(主要使用opencv模块),在python源码里可以调节其参数,尽量调到最好的效果,使用颜色鲜艳的图片效果更佳。
1.jpg
将图片转为素描主要有4个步骤:
1、首先将彩色图转换成灰度图;
2、对灰度图进行求其反色的操作;
3、对第2步得到的结果采用一个高斯模糊的操作;
4、采用颜色亮化(color dodge)的技术将第一步的灰度图和第三步操作后的图片进行混合。
  1. # -*- coding:utf-8 -*-
  2.   
  3. import cv2
  4. import numpy as np
  5. from tkinter import filedialog, Tk
  6. from os import getcwd
  7. from re import findall
  8. def open_path():
  9.     # 图片路径
  10.     root = Tk()
  11.     root.withdraw()
  12.     file_path = (filedialog.askopenfilename(title='选择图片文件', filetypes=[('All Files', '*')]))
  13.     return file_path
  14. def dodgeNaive(image, mask):
  15.     # determine the shape of the input image
  16.     width, height = image.shape[:2]
  17.     # prepare output argument with same size as image
  18.     blend = np.zeros((width, height), np.uint8)
  19.     for col in range(width):
  20.         for row in range(height):
  21.             # do for every pixel
  22.             if mask[col, row] == 255:
  23.                 # avoid division by zero
  24.                 blend[col, row] = 255
  25.             else:
  26.                 # shift image pixel value by 8 bits
  27.                 # divide by the inverse of the mask
  28.                 tmp = (image[col, row] << 8) / (255 - mask)
  29.                 # print('tmp={}'.format(tmp.shape))
  30.                 # make sure resulting value stays within bounds
  31.                 if tmp.any() > 255:
  32.                     tmp = 255
  33.                     blend[col, row] = tmp
  34.     return blend
  35. def dodgeV2(image, mask):
  36.     return cv2.divide(image, 255 - mask, scale=256)
  37. def burnV2(image, mask):
  38.     return 255 - cv2.divide(255 - image, 255 - mask, scale=256)
  39. def rgb_to_sketch(src_image_name):
  40.     print('转换中......')
  41.     img_rgb = cv2.imread(src_image_name)
  42.     img_gray = cv2.cvtColor(img_rgb, cv2.COLOR_BGR2GRAY)
  43.     # 读取图片时直接转换操作
  44.     # img_gray = cv2.imread('example.jpg', cv2.IMREAD_GRAYSCALE)
  45.     img_gray_inv = 255 - img_gray
  46.     img_blur = cv2.GaussianBlur(img_gray_inv, ksize=(21, 21),
  47.                                 sigmaX=0, sigmaY=0)
  48.     img_blend = dodgeV2(img_gray, img_blur)
  49.     # cv2.imshow('original', img_rgb)
  50.     # cv2.imshow('gray', img_gray)
  51.     # cv2.imshow('gray_inv', img_gray_inv)
  52.     # cv2.imshow('gray_blur', img_blur)
  53.     cv2.imwrite(dst_image_name, img_blend)
  54.     save_path = getcwd() + "\" + dst_image_name  # 保存路径
  55.     print('转换完成!!!\n')
  56.     print('保存路径:' + save_path)
  57.     cv2.imshow(save_path, img_blend)
  58.     cv2.waitKey(0)
  59.     cv2.destroyAllWindows()
  60. if __name__ == '__main__':
  61.     print('请选择图片(路径不要含中文):')
  62.     src_image_name = open_path()  # 文件路径
  63.     print(src_image_name+'\n')
  64.     image_name = ''.join(findall(r'[^\\/:*?"<>|\r\n]+$ ', src_image_name))  # 获取文件名
  65.     dst_image_name = 'Sketch_' + image_name
  66.     rgb_to_sketch(src_image_name)
复制代码
您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

QQ|小黑屋|paopaomj.COM ( 渝ICP备18007172号 )

GMT+8, 2020-7-4 04:52

Powered by paopaomj X3.4 © 2016-2021 sitemap

快速回复 返回顶部 返回列表