巧用python实现图片转换成素描和漫画格式
生活随笔
收集整理的這篇文章主要介紹了
巧用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实现图片转换成素描和漫画格式的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 你知道什么是Spring么
- 下一篇: 蔚来获阿布扎比主权基金11亿美元投资,李