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

原文地址：https://blog.csdn.net/weixin_73504499/article/details/142931611

免责声明：本站文章内容转载自网络资源，如本站内容侵犯了原著者的合法权益，可联系本站删除。更多内容请关注自学内容网（zxcms.com）！