基于Python实现视频的人脸融合功能


Posted in Python onJune 12, 2020

基于Python实现视频的人脸融合功能

图片提取

为了方便技术展示,我们选取素材为演员杨紫的一段演讲视频,用例仅为技术交流演示使用,不针对任何指定人。

为达到我们AI换脸的目的,我们首先需要将这段视频逐帧提取成照片

def vedio_2_pic(self,file,save_path):
 """
 逐帧取照片
 file:视频的位置
 save_path:保存路径
 """
 # 读取视频
 video = cv2.VideoCapture(file)
 # 获取视频帧率
 fps = video.get(cv2.CAP_PROP_FPS)
 # 获取画面大小
 width = int(video.get(cv2.CAP_PROP_FRAME_WIDTH))
 height = int(video.get(cv2.CAP_PROP_FRAME_HEIGHT))
 size = (width, height)
 frame_num = str(video.get(7))
 ret, frame = video.read()
 num =1
 while True:
  ret, frame = video.read()
  if ret !=True:
  break
  cv2.imwrite(save_path + str(num) + '.jpg', frame)
  num +=1
 video.release()
 return fps, size,frame_num

脸部融合

上一环节我们已对视频进行了图片提取,我希望演员杨紫可以有杨幂一样的眼睛,长得更像杨幂一点。那么"目标选定人"就是杨幂了。

因为视频可能会有镜头切换,对焦目标可能是观众或者其他人,所以在脸部融合前,需要判断要转换的图片中是否有杨紫,如果有大于0.85的概率确定是杨紫,我们进行脸部融合,如果小于等于0.85概率我们就不进行脸部融合,下面进行脸部融合。

基于Python实现视频的人脸融合功能

核心代码

for i in range(frame_num):
 image2="""/Users/***/face_ztl/picture/%d.jpg"""%(i)
 content = video_make.img_compare(image1,image2)
 try:
 if content >0.85:
  print('-----%d-----'%(i))
  video_make.merge_face(image2,object_image,merge_rate=100)
 else:
  print(0)
  img = cv2.imread(image2)
  img_path = image2.replace('picture','picture_new')
  cv2.imwrite(img_path, img)
 except:
 pass
 print('第%d张出现了问题'%(i))

提取音频

为了合成后的视频有声音,我们将视频中的音频进行提取并保存下来,代码如下:

def getMusic(self,video_name,save_path):
 """
 获取指定视频的音频
 video_name:视频路径
 save_path:音频保存路径
 举例
 save_path='/Users/***/vedio/1.wav' 
 """
 # 读取视频文件
 video = VideoFileClip(video_name)
 # 返回音频
 audio = video.audio
 audio.write_audiofile(save_path)

合成视频

至此我们得到了音频、融合后的图片,接下来就是最后一步合成视频了,代码如下:

"""
 图片转视频
 save_path:视频保存路径
 """
 # 写入视频
 fourcc = cv2.VideoWriter_fourcc(*'mp4v')
 video = cv2.VideoWriter(save_path, fourcc, fps, size)

 # 排序目的
 for item in range(int(frame_num)):
  path = pic_path +str(item) + '.jpg'
  if os.path.exists(path):
  # 读取原图像
  img = cv2.imread(path)
  # 写入视频
  video.write(img)
 video.release()

总结

到此这篇关于基于Python实现视频的人脸融合功能的文章就介绍到这了,更多相关Python实现视频的人脸融合内容请搜索三水点靠木以前的文章或继续浏览下面的相关文章希望大家以后多多支持三水点靠木!

Python 相关文章推荐
python 网络编程详解及简单实例
Apr 25 Python
基于Django与ajax之间的json传输方法
May 29 Python
利用Python实现原创工具的Logo与Help
Dec 03 Python
Python3实现汉语转换为汉语拼音
Jul 08 Python
基于Django的乐观锁与悲观锁解决订单并发问题详解
Jul 31 Python
Python3.5 win10环境下导入kera/tensorflow报错的解决方法
Dec 19 Python
Python运行异常管理解决方案
Mar 09 Python
pytorch实现查看当前学习率
Jun 24 Python
使用keras框架cnn+ctc_loss识别不定长字符图片操作
Jun 29 Python
通过实例简单了解Python sys.argv[]使用方法
Aug 04 Python
Python爬虫简单运用爬取代理IP的实现
Dec 01 Python
Python Selenium XPath根据文本内容查找元素的方法
Dec 07 Python
使用Keras实现简单线性回归模型操作
Jun 12 #Python
Python实现Keras搭建神经网络训练分类模型教程
Jun 12 #Python
简单了解Python变量作用域正确使用方法
Jun 12 #Python
keras 读取多标签图像数据方式
Jun 12 #Python
Python数据可视化图实现过程详解
Jun 12 #Python
浅谈cv2.imread()和keras.preprocessing中的image.load_img()区别
Jun 12 #Python
升级keras解决load_weights()中的未定义skip_mismatch关键字问题
Jun 12 #Python
You might like
使用 MySQL Date/Time 类型
2008/03/26 PHP
Yii不依赖Model的表单生成器用法实例
2014/12/04 PHP
php静态文件返回304技巧分享
2015/01/06 PHP
PHP文件缓存类实现代码
2015/10/26 PHP
YII Framework框架教程之国际化实现方法
2016/03/14 PHP
PHP正则删除HTML代码中宽高样式的方法
2017/06/12 PHP
php+redis消息队列实现抢购功能
2018/02/08 PHP
javascript 支持ie和firefox杰奇翻页函数
2008/07/22 Javascript
屏蔽相应键盘按钮操作
2014/03/10 Javascript
jquery 根据name名获取元素的value值
2015/02/27 Javascript
jQuery实现自动调整字体大小的方法
2015/06/15 Javascript
AngularJS学习笔记之ng-options指令
2015/06/16 Javascript
纯JS实现可拖拽表单的简单实例
2016/09/02 Javascript
Javascript生成带参数的二维码示例
2016/10/10 Javascript
vue复合组件实现注册表单功能
2017/11/06 Javascript
详细教你微信公众号正文页SVG交互开发技巧
2019/07/25 Javascript
微信小程序 腾讯地图显示偏差问题解决
2019/07/27 Javascript
vue中watch和computed为什么能监听到数据的改变以及不同之处
2019/12/27 Javascript
nuxt 服务器渲染动态设置 title和seo关键字的操作
2020/11/05 Javascript
python中numpy包使用教程之数组和相关操作详解
2017/07/30 Python
python中正则表达式的使用方法
2018/02/25 Python
python-OpenCV 实现将数组转换成灰度图和彩图
2020/01/09 Python
Django admin管理工具TabularInline类用法详解
2020/05/14 Python
Keras 数据增强ImageDataGenerator多输入多输出实例
2020/07/03 Python
Python3获取cookie常用三种方案
2020/10/05 Python
python开发一个解析protobuf文件的简单编译器
2020/11/17 Python
Python 获取异常(Exception)信息的几种方法
2020/12/29 Python
plt.figure()参数使用详解及运行演示
2021/01/08 Python
挪威手表购物网站:Klokker
2016/09/19 全球购物
BLACKMORES澳洲官网:澳大利亚排名第一的保健品牌
2018/09/27 全球购物
EJB的角色和三个对象
2015/12/31 面试题
法定代表人授权委托书
2014/09/19 职场文书
北京故宫的导游词
2015/01/31 职场文书
罗马假日观后感
2015/06/08 职场文书
小学生优秀作文范文(六篇)
2019/07/10 职场文书
Windows Server 2019 域控制器安装图文教程
2022/04/28 Servers