废话不多说,直接开干! 抖音字符视频在今年火过一段时间。
反正我是始终忘不了那段刘耕宏老师本草纲目的音乐…
这一次自己也来实现一波,做一个字符视频出来
百度好多都是显示模块,这个完整实现效果
步骤
将视频转化为一帧一帧的图片 把图片转化为字符画 按顺序播放字符画
1、准备
安装 Python-OpenCV 库 安装 Numpy 科学计算库
用到模块库
import time
import cv2
import os
from PIL import Image, ImageDraw, ImageFont
import numpy as np
import os
然后新建python代码文档,在开头添加上下面的导入语句
2. 材料
材料来个视频文件了,我这里用的是zimeng.mp4,下载下来和代码放到同一目录下 你也可以换成自己的,建议是学习时尽量选个短一点的视频,十几秒十秒就行了,方便调试用 此外,要选择对比度高的视频。否则的话,就需要彩色字符才能有足够好的表现,有时间我试试。
3、按帧读取视频
现在继续添加代码,实现第一步:按帧读取视频。 下面这个函数,接受视频路径和字符视频的尺寸信息,返回一个img列表,其中的img是尺寸都为指定大小的灰度图。
第一步截取图片
def video_img(file='zimeng.mp4'):
folder_path = "img_bear/"
if folder_path:
pass
else:
os.makedirs(folder_path)
vc = cv2.VideoCapture(file)
ret = vc.isOpened()
num = 0
while ret:
num = num + 1
ret, frame = vc.read()
if ret:
cv2.imwrite('img_bear/' + str(num) + '.jpg', frame)
print('img_bear/' + str(num) + '.jpg')
cv2.waitKey(1)
else:
break
vc.release()
time.sleep(0.5)
video_image(num)
如果运行没报错,就没问题 代码里的注释应该写得很清晰了,继续下一步
第二步对图片做灰度处理
视频转换成了图像,这一步便是把图像转换成字符画 上面这个函数,一个img对象为参数,前往对应的字符画
def video_image(num=''):
folder_path = "bear/"
if folder_path:
pass
else:
os.makedirs(folder_path)
for i in range(1, num):
filename = 'img_bear/' + str(i) + '.jpg'
im = Image.open(filename)
width = im.size[0]
heigth = im.size[1]
print('宽:%d,高:%d' % (im.size[0], im.size[1]))
ascii_char = list("$@B%8&WM#*oahkbdpqwmZO0QLCJUYXzcvunxrjft/\|()1{}[]?-_+~ <>i!lI;:,\"^`'. ")
if os.path.exists(filename):
img_array = np.array(Image.open(filename).resize((160, 160), Image.ANTIALIAS).convert('L'))
img = Image.new('L', (width, heigth), 255)
draw_object = ImageDraw.Draw(img)
font = ImageFont.truetype('consola.ttf', 10, encoding='unic')
for j in range(160):
for k in range(160):
x, y = k * 8, j * 8
index = int(img_array[j][k] / 4)
draw_object.text((x, y), ascii_char[index], font=font, fill=0)
name = 'bear/' + str(i) + '.jpg'
print(name)
img.save(name, 'JPEG')
time.sleep(0.5)
video(num)
第三步字符转视频
写了这么多代码,如今终于要出效果了。如今就是最激动人心的一步:播放字符画了。 异样的,我把它封装成了一个函数。上面这个函数承受一个字符画的列表并播放。
def video(num):
filename = 'img_bear/' + str(1) + '.jpg'
im = Image.open(filename)
width = im.size[0]
heigth = im.size[1]
fourcc = cv2.VideoWriter_fourcc(*'MJPG')
videoWriter = cv2.VideoWriter('bear_character.avi', fourcc, 20.0, (width, heigth))
for i in range(1, num):
filename = 'bear/'+str(i)+'.jpg'
if os.path.exists(filename):
img = cv2.imread(filename=filename)
cv2.waitKey(100)
videoWriter.write(img)
print(str(i) + '.jpg' + ' done!')
videoWriter.release()
time.sleep(1)
remove_img()
remove_img_bear()
下面完整代码 可能要等很久。我使用示例视频大概需要 500 秒左右。 ctrl+f10执行对应的文件 完整代码里面加了 执行生成图片,生成灰度图片,最后通过灰度生成字节视频删除多余文件 说了那太多废话就是:最后还需删除一些临时的文件及文件夹。
import time
import cv2
import os
from PIL import Image, ImageDraw, ImageFont
import numpy as np
import os
def video_img(file='zimeng.mp4'):
folder_path = "img_bear/"
if folder_path:
pass
else:
os.makedirs(folder_path)
vc = cv2.VideoCapture(file)
ret = vc.isOpened()
num = 0
while ret:
num = num + 1
ret, frame = vc.read()
if ret:
cv2.imwrite('img_bear/' + str(num) + '.jpg', frame)
print('img_bear/' + str(num) + '.jpg')
cv2.waitKey(1)
else:
break
vc.release()
time.sleep(0.5)
video_image(num)
def video_image(num=''):
folder_path = "bear/"
if folder_path:
pass
else:
os.makedirs(folder_path)
for i in range(1, num):
filename = 'img_bear/' + str(i) + '.jpg'
im = Image.open(filename)
width = im.size[0]
heigth = im.size[1]
print('宽:%d,高:%d' % (im.size[0], im.size[1]))
ascii_char = list("$@B%8&WM#*oahkbdpqwmZO0QLCJUYXzcvunxrjft/\|()1{}[]?-_+~ <>i!lI;:,\"^`'. ")
if os.path.exists(filename):
img_array = np.array(Image.open(filename).resize((160, 160), Image.ANTIALIAS).convert('L'))
img = Image.new('L', (width, heigth), 255)
draw_object = ImageDraw.Draw(img)
font = ImageFont.truetype('consola.ttf', 10, encoding='unic')
for j in range(160):
for k in range(160):
x, y = k * 8, j * 8
index = int(img_array[j][k] / 4)
draw_object.text((x, y), ascii_char[index], font=font, fill=0)
name = 'bear/' + str(i) + '.jpg'
print(name)
img.save(name, 'JPEG')
time.sleep(0.5)
video(num)
def video(num):
filename = 'img_bear/' + str(1) + '.jpg'
im = Image.open(filename)
width = im.size[0]
heigth = im.size[1]
fourcc = cv2.VideoWriter_fourcc(*'MJPG')
videoWriter = cv2.VideoWriter('bear_character.avi', fourcc, 20.0, (width, heigth))
for i in range(1, num):
filename = 'bear/'+str(i)+'.jpg'
if os.path.exists(filename):
img = cv2.imread(filename=filename)
cv2.waitKey(100)
videoWriter.write(img)
print(str(i) + '.jpg' + ' done!')
videoWriter.release()
time.sleep(1)
remove_img()
remove_img_bear()
def remove_img():
files = os.getcwd()
file_name = files + "/img_bear"
del_list = os.listdir(file_name)
for f in del_list:
file_path = os.path.join(file_name, f)
print(file_path)
if os.path.isfile(file_path):
os.remove(file_path)
print('成功删除文件:')
else:
print('未找到此文件:')
def remove_img_bear():
files = os.getcwd()
file_name = files + "/bear"
del_list = os.listdir(file_name)
for f in del_list:
file_path = os.path.join(file_name, f)
print(file_path)
if os.path.isfile(file_path):
os.remove(file_path)
print('成功删除文件:')
else:
print('未找到此文件:')
def main():
video_img('video.mp4')
if __name__ == "__main__":
main()
进一步优化 到了这里,核心功能基本都完成了。 不过仔细想想,其实还有很多可以做的:
什么是指定要转换的区间、帧率?
每次转换都要很久的时间,能不能边转换边播放?或者转换后把数据保存起来,下次播放时,就直接读缓存 总结 完整代码,要注意的是代码库的代码。
最后祝愿大家们 2022在疫情下的五一,国际劳动节也是全世界劳动者的节日,劳动创造了人类的美好和幸福,劳动者是最美丽的人。最后,我提前祝大家“五一”节日快乐、平安幸福!
看下效果图
|