Python抖音快手代码舞(字符舞)的实现方法


Posted in Python onFebruary 07, 2021

先上效果,视频敬上:

字符舞:

Python抖音快手代码舞(字符舞)的实现方法

代码舞

源代码:

video_2_code_video.py

import argparse
import os
import cv2
import subprocess
from cv2 import VideoWriter_fourcc
from PIL import Image, ImageFont, ImageDraw

# 命令行输入参数处理
# aparser = argparse.ArgumentParser()
# aparser.add_argument('file')
# aparser.add_argument('-o','--output')
# aparser.add_argument('-f','--fps',type = float, default = 24)#帧
# aparser.add_argument('-s','--save',type = bool, nargs='?', default = False, const = True)
# 是否保留Cache文件,默认不保存

class Video2CodeVideo:
 def __init__(self):
 self.config_dict = {
  # 原视频文件
  "input_file": "video/test.mp4",
  # 中间文件存放目录
  "cache_dir": "cache",
  # 是否保留过程文件。True--保留,False--不保留
  "save_cache_flag": False,
  # 使用使用的字符集
  "ascii_char_list": list("01B%8&WM#*oahkbdpqwmZO0QLCJUYXzcvunxrjft/\|()1{}[]?-_+~<>i!lI;:oa+>!:+. "),
 }

 # 第一步从函数,将像素转换为字符
 # 调用栈:video_2_txt_jpg -> txt_2_image -> rgb_2_char
 def rgb_2_char(self, r, g, b, alpha=256):
 if alpha == 0:
  return ''
 length = len(self.config_dict["ascii_char_list"])
 gray = int(0.2126 * r + 0.7152 * g + 0.0722 * b)
 unit = (256.0 + 1) / length
 return self.config_dict["ascii_char_list"][int(gray / unit)]

 # 第一步从函数,将txt转换为图片
 # 调用栈:video_2_txt_jpg -> txt_2_image -> rgb_2_char
 def txt_2_image(self, file_name):
 im = Image.open(file_name).convert('RGB')
 # gif拆分后的图像,需要转换,否则报错,由于gif分割后保存的是索引颜色
 raw_width = im.width
 raw_height = im.height
 width = int(raw_width / 6)
 height = int(raw_height / 15)
 im = im.resize((width, height), Image.NEAREST)

 txt = ""
 colors = []
 for i in range(height):
  for j in range(width):
  pixel = im.getpixel((j, i))
  colors.append((pixel[0], pixel[1], pixel[2]))
  if (len(pixel) == 4):
   txt += self.rgb_2_char(pixel[0], pixel[1], pixel[2], pixel[3])
  else:
   txt += self.rgb_2_char(pixel[0], pixel[1], pixel[2])
  txt += '\n'
  colors.append((255, 255, 255))

 im_txt = Image.new("RGB", (raw_width, raw_height), (255, 255, 255))
 dr = ImageDraw.Draw(im_txt)
 # font = ImageFont.truetype(os.path.join("fonts","汉仪楷体简.ttf"),18)
 font = ImageFont.load_default().font
 x = y = 0
 # 获取字体的宽高
 font_w, font_h = font.getsize(txt[1])
 font_h *= 1.37 # 调整后更佳
 # ImageDraw为每个ascii码进行上色
 for i in range(len(txt)):
  if (txt[i] == '\n'):
  x += font_h
  y = -font_w
  # self, xy, text, fill = None, font = None, anchor = None,
  # *args, ** kwargs
  dr.text((y, x), txt[i], fill=colors[i])
  # dr.text((y, x), txt[i], font=font, fill=colors[i])
  y += font_w

 name = file_name
 # print(name + ' changed')
 im_txt.save(name)


 # 第一步,将原视频转成字符图片
 # 调用栈:video_2_txt_jpg -> txt_2_image -> rgb_2_char
 def video_2_txt_jpg(self, file_name):
 vc = cv2.VideoCapture(file_name)
 c = 1
 if vc.isOpened():
  r, frame = vc.read()
  if not os.path.exists(self.config_dict["cache_dir"]):
  os.mkdir(self.config_dict["cache_dir"])
  os.chdir(self.config_dict["cache_dir"])
 else:
  r = False
 while r:
  cv2.imwrite(str(c) + '.jpg', frame)
  self.txt_2_image(str(c) + '.jpg') # 同时转换为ascii图
  r, frame = vc.read()
  c += 1
 os.chdir('..')
 return vc

 # 第二步,将字符图片合成新视频
 def txt_jpg_2_video(self, outfile_name, fps):
 fourcc = VideoWriter_fourcc(*"MJPG")

 images = os.listdir(self.config_dict["cache_dir"])
 im = Image.open(self.config_dict["cache_dir"] + '/' + images[0])
 vw = cv2.VideoWriter(outfile_name + '.avi', fourcc, fps, im.size)

 os.chdir(self.config_dict["cache_dir"])
 for image in range(len(images)):
  # Image.open(str(image)+'.jpg').convert("RGB").save(str(image)+'.jpg')
  frame = cv2.imread(str(image + 1) + '.jpg')
  vw.write(frame)
  # print(str(image + 1) + '.jpg' + ' finished')
 os.chdir('..')
 vw.release()

 # 第三步,从原视频中提取出背景音乐
 def video_extract_mp3(self, file_name):
 outfile_name = file_name.split('.')[0] + '.mp3'
 subprocess.call('ffmpeg -i ' + file_name + ' -f mp3 -y ' + outfile_name, shell=True)

 # 第四步,将背景音乐添加到新视频中
 def video_add_mp3(self, file_name, mp3_file):
 outfile_name = file_name.split('.')[0] + '-txt.mp4'
 subprocess.call('ffmpeg -i ' + file_name + ' -i ' + mp3_file + ' -strict -2 -f mp4 -y ' + outfile_name, shell=True)

 # 第五步,如果没配置保留则清除过程文件
 def clean_cache_while_need(self):
 # 为了清晰+代码比较短,直接写成内部函数
 def remove_cache_dir(path):
  if os.path.exists(path):
  if os.path.isdir(path):
   dirs = os.listdir(path)
   for d in dirs:
   if os.path.isdir(path + '/' + d):
    remove_cache_dir(path + '/' + d)
   elif os.path.isfile(path + '/' + d):
    os.remove(path + '/' + d)
   os.rmdir(path)
   return
  elif os.path.isfile(path):
   os.remove(path)
  return
 # 为了清晰+代码比较短,直接写成内部函数
 def delete_middle_media_file():
  os.remove(self.config_dict["input_file"].split('.')[0] + '.mp3')
  os.remove(self.config_dict["input_file"].split('.')[0] + '.avi')
 # 如果没配置保留则清除过程文件
 if not self.config_dict["save_cache_flag"]:
  remove_cache_dir(self.config_dict["cache_dir"])
  delete_middle_media_file()

 # 程序主要逻辑
 def main_logic(self):
 # 第一步,将原视频转成字符图片
 vc = self.video_2_txt_jpg(self.config_dict["input_file"])
 # 获取原视频帧率
 fps = vc.get(cv2.CAP_PROP_FPS)
 # print(fps)
 vc.release()
 # 第二步,将字符图片合成新视频
 self.txt_jpg_2_video(self.config_dict["input_file"].split('.')[0], fps)
 print(self.config_dict["input_file"], self.config_dict["input_file"].split('.')[0] + '.mp3')
 # 第三步,从原视频中提取出背景音乐
 self.video_extract_mp3(self.config_dict["input_file"])
 # 第四步,将背景音乐添加到新视频中
 self.video_add_mp3(self.config_dict["input_file"].split('.')[0] + '.avi', self.config_dict["input_file"].split('.')[0] + '.mp3')
 # 第五步,如果没配置保留则清除过程文件
 self.clean_cache_while_need()

if __name__ == '__main__':
 obj = Video2CodeVideo()
 obj.main_logic()

运行环境:

操作系统:win10

版本:Python 3.8.4

依赖库:pip install opencv-python pillow

管理员权限安装,我的已安装过,显示这样:

Python抖音快手代码舞(字符舞)的实现方法

依赖应用: ffpmeg(下载直接解压、将bin目录加到PATH环境变量)

Python抖音快手代码舞(字符舞)的实现方法

小白式运行(大佬请装瞎):

将上面的源代码命名video_2_code_video.py,在同一目录下新建文件夹video:

Python抖音快手代码舞(字符舞)的实现方法

在video中放入要转换的原视频,命名test.mp4:

Python抖音快手代码舞(字符舞)的实现方法

打开Python3.8

Python抖音快手代码舞(字符舞)的实现方法

运行video_2_code_video.py,如下图显示表示正在运行:

Python抖音快手代码舞(字符舞)的实现方法

会产生一些中间文件诸如:

Python抖音快手代码舞(字符舞)的实现方法

Python抖音快手代码舞(字符舞)的实现方法

经过漫长的等待,终于得偿所愿:

Python抖音快手代码舞(字符舞)的实现方法

test-txt.mp4就是所要的代码舞啦:

Python抖音快手代码舞(字符舞)的实现方法

总结 

到此这篇关于Python抖音快手代码舞实现的文章就介绍到这了,更多相关Python抖音快手代码舞内容请搜索三水点靠木以前的文章或继续浏览下面的相关文章希望大家以后多多支持三水点靠木!

Python 相关文章推荐
python访问类中docstring注释的实现方法
May 04 Python
python开发之str.format()用法实例分析
Feb 22 Python
Python selenium抓取微博内容的示例代码
May 17 Python
python使用matplotlib画饼状图
Sep 25 Python
python 实现分页显示从es中获取的数据方法
Dec 26 Python
python实现各种插值法(数值分析)
Jul 30 Python
python中open函数的基本用法示例
Sep 07 Python
Python3 合并二叉树的实现
Sep 30 Python
python调用Matplotlib绘制分布点图
Oct 18 Python
python中resample函数实现重采样和降采样代码
Feb 25 Python
解决jupyter notebook 前面书写后面内容消失的问题
Apr 13 Python
python小白学习包管理器pip安装
Jun 09 Python
Python图像处理之膨胀与腐蚀的操作
Feb 07 #Python
django inspectdb 操作已有数据库数据的使用步骤
Feb 07 #Python
python数据抓取3种方法总结
Feb 07 #Python
python 批量将中文名转换为拼音
Feb 07 #Python
如何用用Python将地址标记在地图上
Feb 07 #Python
python 三种方法提取pdf中的图片
Feb 07 #Python
Python 转移文件至云对象存储的方法
Feb 07 #Python
You might like
PHP 获取客户端真实IP地址多种方法小结
2010/05/15 PHP
Zend Framework框架实现类似Google搜索分页效果
2016/11/25 PHP
PHP实现的服务器一致性hash分布算法示例
2018/08/09 PHP
javascript引用对象的方法
2007/01/11 Javascript
js常见表单应用技巧
2008/01/09 Javascript
JavaScript脚本性能优化注意事项
2008/11/18 Javascript
如何使用Javascript获取距今n天前的日期
2013/07/08 Javascript
jquery三个关闭弹出层的小示例
2013/11/05 Javascript
jquery操作下拉列表、文本框、复选框、单选框集合(收藏)
2014/01/08 Javascript
js中传递特殊字符(+,&amp;)的方法
2014/01/16 Javascript
Jquery实现侧边栏跟随滚动条固定(兼容IE6)
2014/04/02 Javascript
js弹出对话框方式小结
2015/11/17 Javascript
jQuery自定义滚动条完整实例
2016/01/08 Javascript
JS判断输入字符串长度实例代码(汉字算两个字符,字母数字算一个)
2016/08/02 Javascript
AngularJs 最新验证手机号码的实例,成功测试通过
2017/11/26 Javascript
浅谈从React渲染流程分析Diff算法
2018/09/08 Javascript
node.js环境搭建图文详解
2018/09/19 Javascript
JavaScript RegExp 对象用法详解
2019/09/24 Javascript
使用JS监听键盘按下事件(keydown event)
2019/11/07 Javascript
vue实现将一个数组内的相同数据进行合并
2019/11/07 Javascript
原生JS实现烟花效果
2020/03/10 Javascript
Python中使用items()方法返回字典元素对的教程
2015/05/21 Python
Python编程实现的简单神经网络算法示例
2018/01/26 Python
python 对txt中每行内容进行批量替换的方法
2018/07/11 Python
python实现微信防撤回神器
2019/04/29 Python
基于python的socket实现单机五子棋到双人对战
2020/03/24 Python
python常用库之NumPy和sklearn入门
2019/07/11 Python
Pandas中两个dataframe的交集和差集的示例代码
2020/12/13 Python
Python创建简单的神经网络实例讲解
2021/01/04 Python
html svg生成环形进度条的实现方法
2019/09/23 HTML / CSS
委托书怎样写
2014/08/30 职场文书
代办出身证明书
2014/10/21 职场文书
大雁塔导游词
2015/02/04 职场文书
英文导游词
2015/02/13 职场文书
酒店采购员岗位职责
2015/04/03 职场文书
2015年班长个人工作总结
2015/04/03 职场文书