ToB企服应用市场:ToB评测及商务社交产业平台
标题:
基于Opencv中的DNN模块实现图像/视频的风格迁徙
[打印本页]
作者:
没腿的鸟
时间:
2024-10-15 14:18
标题:
基于Opencv中的DNN模块实现图像/视频的风格迁徙
一、DNN模块的介绍
1、简介
OpenCV中的DNN（Deep Neural Network）模块是一个功能强大的组件，它支持深度学习网络模子的加载和推理。虽然DNN模块不提供模子的训练功能，但它可以与主流的深度学习框架（如TensorFlow、Caffe、Torch和Darknet）兼容，加载这些框架训练好的模子举行推理。
2、特点
（1）
、轻量型
：DNN模块仅提供推理功能，因此代码量较小，运行开销也较小。
（2）
、使用方便
：如果项目中已经使用OpenCV，那么可以很轻易地集成DNN模块，为项目添加深度学习本领。
（3）
、通用性
：支持多种网络模子格式，无需举行额外的模子转换。
二、DNN模块的用法
（1）、图像预处理功能函数cv2.dnn.blobFromImage
实现图像预处理，从原始图像构建一个符合人工神经网络输入格式的四维块。
复制代码
blob = cv2.dnn.blobFromImage(image, scalefactor=None, size=None, mean=None, swapRB=None, crop=None)
参数：
image
:表示输入图像。
scalefactor
:表示对图像内的数据举行缩放的比例因子。具体运算是每个像素值*scalefactor，该值默认为 1。
size
:用于控制blob的宽度、高度。
mean
:表示从每个通道减去的均值。 (0, 0, 0)：表示不举行均值减法。即，不对图像的B、G、R通道举行任何减法操纵。
若输入图像本身是B、G、R通道顺序的，而且下一个参数swapRB值为True，
swapRB
:表示在须要时交换通道的R通道和B通道。一般情况下使用的是RGB通道。而OpenCV通常接纳的是BGR通道。因此可以根据需要交换第1个和第3个通道。该值默认为 False。
crop
:布尔值，如果为 True，则在调整大小后举行居中裁剪。
返回值：blob: 表示在颠末缩放、裁剪、减均值后得到的符合人工神经网络输入的数据。该数据是一个四维数据，
布局通常使用N(表示batch size)、C(图像通道数，如RGB图像具有三个通道)、H(图像高度)、W(图像宽度)
（2）、模子加载方法
加载模子net=cv2.dnn.readNet( model[, config[， framework]] ）
各参数的寄义如下:
model
:模子权重参数文件路径。文件内存储的是训练好的模子的权重值，是二进制文件。
config
:模子配置文件路径。模子配置文件内存储的模子形貌文件,形貌的是网络布局，是文本文件，文件较小。
framework
NN框架，可省略，DNN模块会主动推断框架种类。
net
:返回值，返回网络模子对象。
三、DNN模块的实践，实现图像的风格迁徙
代码
import cv2
image = cv2.imread('image/cat.jpg')
cv2.imshow('cat', 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/candy.t7')
# net = cv2.dnn.readNet(r'model/composition_vii.t7')
# net = cv2.dnn.readNet(r'model/feathers.t7')
net = cv2.dnn.readNet(r'model/la_muse.t7')
# net = cv2.dnn.readNet(r'model/mosaic.t7')
# net = cv2.dnn.readNet(r'model/the_wave.t7')
# net = cv2.dnn.readNet(r'model/starry_night.t7')
# net = cv2.dnn.readNet(r'model/udnie.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('result', result)
cv2.waitKey(0)
cv2.destroyAllWindows()
复制代码
结果展示
免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。
欢迎光临 ToB企服应用市场:ToB评测及商务社交产业平台 (https://dis.qidao123.com/)
Powered by Discuz! X3.4
NN框架，可省略，DNN模块会主动推断框架种类。