-
将下图中的图片换一种风格展示出来
-
如下图结果：
-
完整代码如下：
import cv2image_yuantu = cv2.imread("wechat.jpg") image = cv2.resize(image_yuantu, dsize=None, fx=0.5, fy=0.5) cv2.imshow('yuan tu', image) cv2.waitKey(0)""" 图片预处理 """ (h, w) = image.shape[:2]blob = cv2.dnn.blobFromImage(image, 1, (w, h), (0, 0, 0), swapRB=True, crop=False)""" 加载模型 """net = cv2.dnn.readNet(r'model\starry_night.t7') # net = cv2.dnn.readNetFromTorch(r'.\model\la_muse.t7') # net = cv2.dnn.readNetFromTorch(r'.\model\candy.t7') # net = cv2.dnn.readNetFromTorch(r'.\model\composition_vii.t7') # net = cv2.dnn.readNetFromTorch(r'.\model\feathers.t7') # net = cv2.dnn.readNetFromTorch(r'.\model\udnie.t7') # net = cv2.dnn.readNetFromTorch(r'.\model\the_scream.t7')# 设置神经网络的输入 net.setInput(blob)out = net.forward()out_new = out.reshape(out.shape[1], out.shape[2], out.shape[3])cv2.normalize(out_new, out_new, norm_type=cv2.NORM_MINMAX)result = out_new.transpose(1, 2, 0)cv2.imshow('Stylized Image', result) cv2.waitKey(0) cv2.destroyAllWindows()
