整活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