整活python实现肌肉金轮词云图视

点击上方“AI算法与图像处理”,选择加"星标"或“置顶”

重磅干货,第一时间送达

一、前言

最近金轮(大司马)因为AI换脸《肌肉金轮》又火遍全网。

今天我利用周末的时间用AI技术整个活,娱乐的同时学点技术,秒啊!

大致要实现的效果是,在肌肉金轮的基础上,附上经典的语录。

二、实现的思路

1、获取素材

2、采用分割算法获取主体的mask

3、在mask上生成词云图

4、将词云图与原图融合

5、添加音频文件

三、代码实现

1、获取素材

包括:AI换脸后的视频、经典语录txt文件

视频截图:

经典语录截图(这里我手动找了一些,大家可以自行搜索更多):

2、采用抠图算法获取主体的mask

1)首先从视频中获取每一帧图片

importcv2cap=cv2.VideoCapture(r"output.mp4")num=1whileTrue:#逐帧读取视频按顺序保存到本地文件夹ret,frame=cap.read()ifret:if1=num:cv2.imwrite(f"./test_data/test_images/img_{num}.jpg",frame)#保存一帧帧的图片print(f==========已成功保存第{num}张图片==========)num+=1else:breakcap.release()#释放资源

2)使用BASNet(CVPR)的分割算法,抠图主体

环境配置:

Python3.6numpy1.15.2scikit-image0.14.0PIL5.2.0PyTorch0.4.0torchvision0.2.1glob

这里我使用高版本的pytorch依然可以用(torch1.6.0+torchvision0.7.0)

克隆项目

gitclone

转载请注明:http://www.sonphie.com/jbzd/14549.html

  • 上一篇文章:
  • 下一篇文章: 没有了
  • 网站简介| 发布优势| 服务条款| 隐私保护| 广告合作| 网站地图| 版权申明

    当前时间: