从github或是其他地方看到的一些算法感觉很有意思,在这里介绍一下
1??图像转手绘
import cv2
from PIL import Image
import numpy as np
frame = cv2.imread('old.jpg')
pilimg = Image.fromarray(np.uint8(frame))
a = np.asarray(pilimg.convert('L')).astype('float')
depth = 10. # (0-100)
grad = np.gradient(a) # 取图像灰度的梯度值
grad_x, grad_y = grad # 分别取横纵图像梯度值
grad_x = grad_x * depth / 100.
grad_y = grad_y * depth / 100.
A = np.sqrt(grad_x ** 2 + grad_y ** 2 + 1.) # 构造x和y轴梯度的三维归一化单位坐标系
uni_x = grad_x / A
uni_y = grad_y / A
uni_z = 1. / A
vec_el = np.pi / 2.2 # 光源的俯视角度,弧度值
vec_az = np.pi / 4. # 光源的方位角度,弧度值
dx = np.cos(vec_el) * np.cos(vec_az) # 光源对x 轴的影响
dy = np.cos(vec_el) * np.sin(vec_az) # 光源对y 轴的影响
dz = np.sin(vec_el) # 光源对z 轴的影响
b = 255 * (dx * uni_x + dy * uni_y + dz * uni_z) # 光源归一化,(梯度和光源相互作用,将梯度转化为灰度)
b = b.clip(0, 255)
im2 = Image.fromarray(b.astype('uint8')) #重构图像
charimg = np.asarray(im2)
cv2.imwrite('new.png',charimg)
这个使用了opencv,numpy与PIL
- opencv用 pip install opencv-python
- numpy 用 pip install numpy
- PIL用 pip install pillow
我们需要一张图像,命名为old.jpg
我们用这张图像
运行一下代码,我们就得到了这张图像
2??图像转水墨画?
import cv2
from PIL import Image
import numpy as np
frame = cv2.imread('old.jpg')
grayImage = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
grayImage = cv2.GaussianBlur(grayImage, (1, 1), 0)
edgeImage = cv2.Laplacian(grayImage, -1, ksize=5)
edgeImage = 255 - edgeImage
ret, edgeImage = cv2.threshold(edgeImage, 150, 255, cv2.THRESH_BINARY)
edgePreservingImage = cv2.edgePreservingFilter(frame, flags=2, sigma_s=50, sigma_r=0.4)
output = np.zeros(grayImage.shape)
output = cv2.bitwise_and(edgePreservingImage, edgePreservingImage, mask=edgeImage)
cartoon_image = cv2.stylization(frame, sigma_s=150, sigma_r=0.25)
cv2.imwrite('new.jpg',cartoon_image)
需要一张图像,我们用这个
运行后转出来的结果是这样的
3??批量修改图像尺寸
opencv是不能读取中文路径的,所以我们下面用的文件不能使用中文名称,如果用了就会报错
import os
import cv2
files = os.listdir('old/')
for file_name in files:
img = cv2.imread('old/'+ file_name)
img = cv2.resize(img,(256,256))
cv2.imwrite('new/' + file_name,img)
print(file_name,'已完成')
这个是用os和opencv做的,opencv直接pip install opencv-python就可以了,os是python自带的
使用前我们需要创建old和new两个文件夹,我们把要改变的图像放old中,改变完的图像会放在new中
(256,256)是我们要生成的图像大小,.jpg是文件格式,new与old都可以改,我们old中的文件是这样的
运行之后的new中的文件是这样的
|