Python+OpenCV图像风格迁移的实现方法讲解

import cv2

# 加载模型

net = cv2.dnn.readNetFromTorch('the_scream.t7')

net.setPreferableBackend(cv2.dnn.DNN_BACKEND_OPENCV);

# 读取图片

image = cv2.imread('test.jpg')

(h, w) = image.shape[:2]

blob = cv2.dnn.blobFromImage(image, 1.0, (w, h), (103.939, 116.779, 123.680), swapRB=False, crop=False)

# 进行计算

net.setInput(blob)

out = net.forward()

out = out.reshape(3, out.shape[2], out.shape[3])

out[0] += 103.939

out[1] += 116.779

out[2] += 123.68

out /= 255

out = out.transpose(1, 2, 0)

# 输出图片

cv2.imshow('Styled image', out)

cv2.waitKey(0)