附录1 opencv的一些简易算法

从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中的文件是这样的