分享

用Python快速实现视频的AI换脸

 wenxuefeng360 2022-08-16 发布于四川

对于娱乐圈来说,良好的艺人形象是吃这碗饭的基础,而对于剧组来说,最重要的就是要擦亮眼睛看人,若是不小心选到污点艺人,之前所有的努力都将化为泡影!我们先列举两部因为污点艺人无法播出、一度叫停的电视剧。

《深渊行者》《无名侦探》为啥要把两部剧放在一说呢?因为这两部是被同一个“阵容”毁掉的,他们便是吴秀波,翟天临这对难兄难弟了!吴秀波的出轨门让这位大器晚成的中年大叔一夜之间人设尽毁,不仅参与录制的多档节目被重新剪辑,参与主演的电视剧也被一度叫停。

因明星丑闻无法播放或被剪辑或换人的电视剧和综艺有很多,对于已经拍摄完成的作品来说,只能依靠剪辑了。通过“一剪没”的手法,简单粗暴,但会影响整个剧情的完整度。如果是一帧一帧修复,这个工作量会让人累吐的。是否可以畅想一下,能否可以把不喜欢、有污点艺人给他“换脸”,声音给他“换声”呢?把喜欢的艺人给加上呢?

转入正题,本篇文章以技术讲解为主,通过人脸融合技术把“指定的人脸”进行更换为“目标选定人”,该方法可以快速进行人脸融合,不用Ps或者剪辑那么苦逼的。需要完整版代码,在公众号后台回复:换脸。本文讲解流程如下图所示:

图片

图片提取

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

为达到我们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概率我们就不进行脸部融合,下面进行脸部融合。

图片

核心代码

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

成果展示

至此我们已经完成了所有环节,下面是成果展示。兴趣是最好的老师,如果大家喜欢某位明星或者把某位明星cut了,都可以尝试一下哦,你有什么想法,请在留言区说出来。

用PC端看效果更佳,本案例参数融合比例为30%,如果需要面部全部替换调高融合比例即可


    本站是提供个人知识管理的网络存储空间,所有内容均由用户发布,不代表本站观点。请注意甄别内容中的联系方式、诱导购买等信息,谨防诈骗。如发现有害或侵权内容,请点击一键举报。
    转藏 分享 献花(0

    0条评论

    发表

    请遵守用户 评论公约

    类似文章 更多