《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()