OpenCV 计算机视觉 - 风格迁移

import cv2 image_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()