经验首页 前端设计 程序设计 Java相关 移动开发 数据库/运维 软件/图像 大数据/云计算 其他经验
当前位置:技术经验 » 程序设计 » Python » 查看文章
Python编程实现小姐姐跳舞并生成词云视频示例
来源:jb51  时间:2021/10/8 13:06:31  对本文有异议

Python做了一个词云视频,以另一种角度来看小姐姐跳舞

制作过程分为以下几个部分

1、视频下载

首先需要下载一个小姐姐跳舞的视频,这里我用的是 you-get 工具,可借助 Python 的 pip 命令进行安装

  1. pip install you-get

you-get 支持下载平台包括:Youtube、Blili、TED、腾讯、优酷、爱奇艺(涵盖所有视频平台下载链接),

以 youtube 视频为例,you-get 下载命令

  1. you-get -o ~/Videos(存放视频路径) -O zoo.webm(视频命名) 'https://www.youtube.com/watch?v=jNQXAC9IVRw'

图片

这里通过 os 模块来实现 you-get 下载命令,使用时传入三个参数即可:

1,视频链接,

2,要存放视频的文件路径;

3,视频命名;

  1. def download(video_url,save_path,video_name):
  2. '''
  3. youget 下载视频
  4. :param video_url:视频链接
  5. :param save_path: 保存路径
  6. :param video_name: 视频命名
  7. :return:
  8. '''
  9. cmd = 'you-get -o {} -O {} {}'.format(save_path,video_name,video_url)
  10. res = os.popen(cmd,)
  11. res.encoding = 'utf-8'
  12. print(res.read())# 打印输出

关于 you-get 更多用法, 可参考官网,里面关于用法介绍的非常详细:

https://you-get.org/#getting-started

图片

2、B 站弹幕下载

做词云图需要有文本数据支持,这里选取 B 站弹幕为素材;关于 B 站视频弹幕下载方式,这里一个快捷方法,用 requests 访问指定视频的 API 接口,就能得到该视频下的全部弹幕

  1. http://comment.bilibili.com/{cid}.xml # cid 为B站视频的cid 编号

但 API 接口的构造,需要知道视频的 cid 编号

B站视频 cid 编号获取方式:

F12打开开发者模式->NetWork->XHR->v2?cid=… 链接 ,该网页链接中有一个”cid=一串数字“ 的字符串,其中等号后面的连续数字就是该视频的 cid 编号

图片

以上面视频为例,291424805 就是这个视频的 cid 编号,

有了 cid 之后,通过 requests 请求 API 接口,就能获取到里面的弹幕数据

  1. http://comment.bilibili.com/291424805.xml

图片

图片

  1. def download_danmu():
  2. '''弹幕下载并存储'''
  3. cid = '141367679'# video_id
  4. url = 'http://comment.bilibili.com/{}.xml'.format(cid)
  5. f = open('danmu.txt','w+',encoding='utf-8') #打开 txt 文件
  6. res = requests.get(url)
  7. res.encoding = 'utf-8'
  8. soup = BeautifulSoup(res.text,'lxml')
  9. items = soup.find_all('d')# 找到 d 标签
  10. for item in items:
  11. text = item.text
  12. print('---------------------------------'*10)
  13. print(text)
  14. seg_list = jieba.cut(text,cut_all =True)# 对字符串进行分词处理,方便后面制作词云图
  15. for j in seg_list:
  16. print(j)
  17. f.write(j)
  18. f.write('\n')
  19. f.close()

3、视频切帧,人像分割

下载到视频之后,先把视频拆分成一帧一帧图像;

  1. vc = cv2.VideoCapture(video_path)
  2. c =0
  3. if vc.isOpened():
  4. rval,frame = vc.read()# 读取视频帧
  5. else:
  6. rval=False
  7. while rval:
  8. rval,frame = vc.read()# 读取每一视频帧,并保存至图片中
  9.  
  10. cv2.imwrite(os.path.join(Pic_path,'{}.jpg'.format(c)),frame)
  11. c += 1
  12. print('第 {} 张图片存放成功!'.format(c))

图片

对每一帧中的小姐姐进行识别提取,也就是人像分割,这里借助了百度 API 接口,

  1. APP_ID = "23633750"
  2. API_KEY = 'uqnHjMZfChbDHvPqWgjeZHCR'
  3. SECRET_KEY = '************************************'
  4. client = AipBodyAnalysis(APP_ID, API_KEY, SECRET_KEY)
  5. # 文件夹
  6. jpg_file = os.listdir(jpg_path)
  7. # 要保存的文件夹
  8. for i in jpg_file:
  9. open_file = os.path.join(jpg_path,i)
  10. save_file = os.path.join(save_path,i)
  11. if not os.path.exists(save_file):#文件不存在时,进行下步操作
  12. img = cv2.imread(open_file) # 获取图像尺寸
  13. height, width, _ = img.shape
  14. if crop_path:# Crop_path 不为 None,则不进行裁剪
  15. crop_file = os.path.join(crop_path,i)
  16. img = img[100:-1,300:-400] #图片太大,对图像进行裁剪里面参数根据自己情况设定
  17. cv2.imwrite(crop_file,img)
  18. image= get_file_content(crop_file)
  19. else:
  20.  
  21. image = get_file_content(open_file)
  22. res = client.bodySeg(image)#调用百度API 对人像进行分割
  23. labelmap = base64.b64decode(res['labelmap'])
  24. labelimg = np.frombuffer(labelmap,np.uint8)# 转化为np数组 0-255
  25. labelimg = cv2.imdecode(labelimg,1)
  26. labelimg = cv2.resize(labelimg,(width,height),interpolation=cv2.INTER_NEAREST)
  27. img_new = np.where(labelimg==1,255,labelimg)# 1 转化为 255
  28. cv2.imwrite(save_file,img_new)
  29. print(save_file,'save successfully')

将含有人像的图像转化为二值化图像,前景为人物,其余部分为背景

图片

API 使用之前需要用自己账号在百度智能云平台创建一个 人体分析 应用,里面需要三个参数:ID、AK、SK

图片

关于百度 API 使用方法,可参考官方文档资料

图片

4、对分割后的图像制作词云图

根据步骤 3 中得到了小姐姐 人像 Mask,

图片

借助 wordcloud 词云库和采集到的弹幕信息,对每一张二值化图像绘制词云图(在制作之前,请确保每一张都是二值化图像,全部为黑色像素图像需要剔除)

图片

  1. word_list = []
  2. with open('danmu.txt',encoding='utf-8') as f:
  3. con = f.read().split('\n')# 读取txt文本词云文本
  4. for i in con:
  5. if re.findall('[\u4e00-\u9fa5]+', str(i), re.S): #去除无中文的词频
  6. word_list.append(i)
  7. for i in os.listdir(mask_path):
  8. open_file = os.path.join(mask_path,i)
  9. save_file = os.path.join(cloud_path,i)
  10. if not os.path.exists(save_file):
  11. # 随机索引前 start 频率词
  12. start = random.randint(0, 15)
  13. word_counts = collections.Counter(word_list)
  14. word_counts = dict(word_counts.most_common()[start:])
  15. background = 255- np.array(Image.open(open_file))
  16. wc =WordCloud(
  17. background_color='black',
  18. max_words=500,
  19. mask=background,
  20. mode = 'RGB',
  21. font_path ="D:/Data/fonts/HGXK_CNKI.ttf",# 设置字体路径,用于设置中文,
  22.  
  23. ).generate_from_frequencies(word_counts)
  24. wc.to_file(save_file)
  25. print(save_file,'Save Sucessfully!')

5、图片拼接,合成视频

词云图全部生成完毕之后,如果一张一张图像看肯定没意思,如果把处理后的词云图合成视频会更酷一点!

为了视频前后对比效果这里我多加了一个步骤,在合并之前先对原图和词云图进行拼接,合成效果如下:

图片

  1. num_list = [int(str(i).split('.')[0]) for i in os.listdir(origin_path)]
  2. fps = 24# 视频帧率,越大越流畅
  3. height,width,_=cv2.imread(os.path.join(origin_path,'{}.jpg'.format(num_list[0]))).shape # 视频高度和宽度
  4. width = width*2
  5. # 创建一个写入操作;
  6. video_writer = cv2.VideoWriter(video_path,cv2.VideoWriter_fourcc(*'mp4v'),fps,(width,height))
  7. for i in sorted(num_list):
  8. i = '{}.jpg'.format(i)
  9. ori_jpg = os.path.join(origin_path,str(i))
  10. word_jpg = os.path.join(wordart_path,str(i))
  11. # com_jpg = os.path.join(Composite_path,str(i))
  12. ori_arr = cv2.imread(ori_jpg)
  13. word_arr = cv2.imread(word_jpg)
  14. # 利用 Numpy 进行拼接
  15. com_arr = np.hstack((ori_arr,word_arr))
  16. # cv2.imwrite(com_jpg,com_arr)# 合成图保存
  17. video_writer.write(com_arr) # 将每一帧画面写入视频流中
  18. print("{} Save Sucessfully---------".format(ori_jpg))

再加上背景音乐,视频又能提升一个档次~

最后

关于视频中的素材,特此声明

弹幕取自B站 Up 主 半佛仙人《【半佛】你知道奶茶加盟到底有多坑人吗?》

小姐姐跳舞视频取自Youtube Channel Lilifilm Official 《LILI's FILM #3 - LISA Dance Performance Video》

最后,感谢大家的阅读,我们下期见~

更多关于Python实现跳舞词云视频的资料请关注w3xue其它相关文章!

 友情链接:直通硅谷  点职佳  北美留学生论坛

本站QQ群:前端 618073944 | Java 606181507 | Python 626812652 | C/C++ 612253063 | 微信 634508462 | 苹果 692586424 | C#/.net 182808419 | PHP 305140648 | 运维 608723728

W3xue 的所有内容仅供测试,对任何法律问题及风险不承担任何责任。通过使用本站内容随之而来的风险与本站无关。
关于我们  |  意见建议  |  捐助我们  |  报错有奖  |  广告合作、友情链接(目前9元/月)请联系QQ:27243702 沸活量
皖ICP备17017327号-2 皖公网安备34020702000426号