之前毒教材被曝光引发争议,那些编写度教材的人着实可恶。咱程序员也没有手绘插画能力,但我们可以借助强大的深度学习模型将视频转动漫。所以的目标是让任何具有python语言基本能力的程序员,实现短视频转动漫效果。
一、思路流程
1. 读取视频帧
2. 将每一帧图像转为动漫帧
3. 将转换后的动漫帧转为视频
难点在于如何将图像转为动漫效果。这里我们使用基于深度学习的动漫效果转换模型,考虑到许多读者对这块不了解,因此我这边准备好了源码和模型,直接调用即可。
动漫卡片印刷系统开发,动漫卡片印刷开发,动漫卡片印刷源码开发,动漫卡片印刷平台开发,动漫卡片印刷app开发,动漫卡片印刷系统多少钱,动漫卡片印刷价格,动漫卡片印刷app功能,动漫卡片印刷平台介绍,动漫卡片印刷系统搭建。
二、图像转动漫
为了让大家不关心深度学习模型,已经为大家准备好了转换后的onnx类型模型。接下来按顺序介绍运行onnx模型流程。
安装onnxruntime库
pip installonnxruntime
1
如果想要用GPU加速,可以安装GPU版本的onnxruntime:
pip installonnxruntime-gpu
1
需要注意的是:
onnxruntime-gpu的版本跟CUDA有关联,具体对应关系如下
当然,如果用CPU运行,那就不需要考虑那么多了。考虑到通用性,本文全部以CPU版本onnxruntime。
运行模型
先导入onnxruntime库,创建InferenceSession对象,调用run函数。
其中,preprocess函数确保输入图像的宽高是8的整数倍。这里主要是因为考虑到深度学习模型有下采样,确保每次下采样能被2整除。
单帧效果展示
三、视频帧读取与视频帧写入
这里使用Opencv库,提取视频中每一帧并调用回调函数将视频帧回传。在将图片转视频过程中,通过定义VideoWriter类型变量WRITE确保唯一性。具体实现代码如下:
ok分享到这里就结束了,赶紧去试试吧
对文章有问题的,或者有其他关于python的问题,可以在评论区留言或者私信我哦