欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

Python3视频转字符动画的实例代码

程序员文章站 2023-08-13 22:28:57
python3视频转字符动画,具体代码如下所示: # -*- coding:utf-8 -*- import json import os import...

python3视频转字符动画,具体代码如下所示:

# -*- coding:utf-8 -*-
import json
import os
import subprocess
from pathlib import path
from cv2 import cv2
import numpy as np
from time import time
import webbrowser
play_chars_js = '''
let i = 0;
window.setinterval(function(){
  let img = frames[i++];
  let html = ""
  for(let line of img){
    for(let char of line){
      let [[r,g,b], ch] = char;
      html += '<span style="color:rgb(' + r + ', ' + g + ', '+ b + ');">'+ ch + '</span>'
      // html += '<span style="background-color:rgb(' + r + ', ' + g + ', '+ b + ');">'+ ch + '</span>'
    }
    html += "<br>"
  }
  document.getelementsbyclassname("video-panel")[0].innerhtml = html
}, 1000/fps);
document.getelementsbytagname("audio")[0].play();
'''
class videotohtml:
  # 像素形状,因为颜色已经用rgb控制了,这里的pixels其实可以随意排
  pixels = "$#@&%zyxwvutsrqponmlkjihgfedcba098765432?][}{/)(><zyxwvutsrqponmlkjihgfedcba*+1-."
  def __init__(self, video_path, fps_for_html=8, time_interval=none):
    """
    :param video_path: 字符串, 视频文件的路径
    :param fps_for_html: 生成的html的帧率
    :param time_interval: 用于截取视频(开始时间,结束时间)单位秒
    """
    self.video_path = path(video_path)
    # 从指定文件创建一个videocapture对象
    self.cap = cv2.videocapture(video_path)
    self.width = self.cap.get(cv2.cap_prop_frame_width)
    self.height = self.cap.get(cv2.cap_prop_frame_height)
    self.frames_count_all = self.cap.get(cv2.cap_prop_frame_count)
    self.fps = self.cap.get(cv2.cap_prop_fps)
    self.resize_width = none
    self.resize_height = none
    self.frames_count = 0
    self.fps_for_html = fps_for_html
    self.time_interval = time_interval
  def video2mp3(self):
    """#调用ffmpeg获取mp3音频文件"""
    mp3_path = self.video_path.with_suffix('.mp3')
    subprocess.call('ffmpeg -i ' + str(self.video_path) + ' -f mp3 ' + str(mp3_path), shell=true)
    return mp3_path
  def set_width(self, width):
    """只能缩小,而且始终保持长宽比"""
    if width >= self.width:
      return false
    else:
      self.resize_width = width
      self.resize_height = int(self.height * (width / self.width))
      return true
  def set_height(self, height):
    """只能缩小,而且始终保持长宽比"""
    if height >= self.height:
      return false
    else:
      self.resize_height = height
      self.resize_width = int(self.width * (height / self.height))
      return true
  def resize(self, img):
    """
    将img转换成需要的大小
    原则:只缩小,不放大。
    """
    # 没指定就不需resize了
    if not self.resize_width or not self.resize_height:
      return img
    else:
      size = (self.resize_width, self.resize_height)
      return cv2.resize(img, size, interpolation=cv2.inter_cubic)
  def get_img_by_pos(self, pos):
    """获取到指定位置的帧"""
    # 把指针移动到指定帧的位置
    self.cap.set(cv2.cap_prop_pos_frames, pos)
    # cap.read() 返回值介绍:
    #  ret 布尔值,表示是否读取到图像
    #  frame 为图像矩阵,类型为 numpy.ndarray.
    ret, frame = self.cap.read()
    return ret, frame
  def get_frame_pos(self):
    """生成需要获取的帧的位置,使用了惰性求值"""
    step = self.fps / self.fps_for_html
    # 如果未指定
    if not self.time_interval:
      self.frames_count = int(self.frames_count_all / step) # 更新count
      return (int(step * i) for i in range(self.frames_count))
    # 如果指定了
    start, end = self.time_interval
    pos_start = int(self.fps * start)
    pos_end = int(self.fps * end)
    self.frames_count = int((pos_end - pos_start) / step) # 更新count
    return (pos_start + int(step * i) for i in range(self.frames_count))
  def get_imgs(self):
    assert self.cap.isopened()
    for i in self.get_frame_pos():
      ret, frame = self.get_img_by_pos(i)
      if not ret:
        print("读取失败,跳出循环")
        break
      yield self.resize(frame) # 惰性求值
    # 结束时要释放空间
    self.cap.release()
  def get_char(self, gray):
    percent = gray / 255 # 转换到 0-1 之间
    index = int(percent * (len(self.pixels) - 1)) # 拿到index
    return self.pixels[index]
  def get_json_pic(self, img):
    """测试阶段,不实用"""
    json_pic = []
    # 宽高刚好和size相反,要注意。(这是numpy的特性。。)
    height, width, channel = img.shape
    # 转换成灰度图,用来选择合适的字符
    img_gray = cv2.cvtcolor(img, cv2.color_bgr2gray)
    for y in range(height):
      line = []
      for x in range(width):
        r, g, b = img[y][x]
        gray = img_gray[y][x]
        char = self.get_char(gray)
        line.append([[int(r), int(g), int(b)], char])
      json_pic.append(line)
    return json.dumps(json_pic)
  def write_html_with_json(self, file_name):
    """测试阶段,不实用"""
    mp3_path = self.video2mp3()
    time_start = time()
    with open(file_name, 'w') as html:
      # 要记得设置monospace等宽字体,不然没法玩
      html.write('<!doctype html>'
            '<html>'
            '<body style="font-family: monospace; font-size: small; font-weight: bold; text-align: center; line-height: 0.8;">'
            '<div class="video-panel"></div>'
            f'<audio src="{mp3_path.name}" autoplay controls></audio>'
            '</body>'
            '<script>'
            'var frames=[\n')
      try:
        i = 0
        for img in self.get_imgs():
          json_pic = self.get_json_pic(img)
          html.write(f"{json_pic},")
          if i % 20:
            print(f"进度:{i/self.frames_count * 100:.2f}%, 已用时:{time() - time_start:.2f}")
          i += 1
      finally:
        html.write('\n];\n'
              f'let fps={self.fps_for_html};\n'
              f'{play_chars_js}'
              '</script>\n'
              '</html>')
def main():
  # 视频路径,换成你自己的
  video_path = "ceshi.mp4"
  video2html = videotohtml(video_path, fps_for_html=8)
  video2html.set_width(120)
  html_name = path(video_path).with_suffix(".html").name
  video2html.write_html_with_json(html_name)
if __name__ == "__main__":
  main()

总结

以上所述是小编给大家介绍的python3视频转字符动画的实例代码,希望对大家有所帮助