示例效果如下:
[短视频转动漫效果]
1 视频转动漫整体实现思路
整个实现流程如下:
读取视频帧
将每一帧图像转为动漫帧
将转换后的动漫帧转为视频
难点在于如何将图像转为动漫效果。...2 图像转动漫
为了让读者不关心深度学习模型,已经为读者准备好了转换后的onnx类型模型。接下来按顺序介绍运行onnx模型流程。...可以安装GPU版本的onnxruntime:
pip install onnxruntime-gpu
需要注意的是:
onnxruntime-gpu的版本跟CUDA有关联,具体对应关系如下:
[cuda与onnxruntime-gpu...np.pad(rgb, ((0,pad_h),(0, pad_w),(0,0)), "reflect")
return rgb, pad_w, pad_h
其中, preprocess函数确保输入图像的宽高是...2.3 单帧效果展示
[效果1]
[效果2]
[效果3]
3 视频帧读取与视频帧写入
这里使用Opencv库,提取视频中每一帧并调用回调函数将视频帧回传。