python来画Bad apple! 50行实现视频转字符画

基本思路

1. 读取一段视频,读取每一帧
2. 对每一帧的图片的尺寸进行处理，并转成灰度图
3. 对于灰度图的每一像素将其映射到自己选择的字符集中，形成字符串
4. 循环打印字符串

版本一

样例

问题

使用os.system(‘cls’)来清空控制台输出效率低下，会出现下图展示的这种情况，导致看起来一闪一闪的

此问题在版本二中解决。

代码

#-*- coding:utf-8 -*-  
import time
import cv2
import numpy as np
from PIL import Image
import os
import curses


codeLib = '''@B%8&WM#*oahkbdpqwmZO0QLCJUYXzcvunxrjft/\|()1{}[]?-_+~<>i!lI;:,"^`'. '''#生成字符画所需的字符集
count = len(codeLib)

def main(video_path):
	'''
	对视频文件切割成帧
	'''
	'''
	@param video_path:视频路径
	'''
	vc=cv2.VideoCapture(video_path)
	#一帧一帧的分割 需要几帧写几
	c=0
	if vc.isOpened():
		rval,frame=vc.read()
	else:
		rval=False
	while rval:
		rval,frame=vc.read()
		if c % 5 == 0:
		    console_print(frame)
		c=c+1

def transform(image_file):
    codePic = ''
    for row in range(0,image_file.shape[0]):  #size属性表示图片的分辨率，'0'为横向大小，'1'为纵向
        for col in range(0,image_file.shape[1]):
            r,g,b = image_file[row][col]
            gray = int(0.2126*r+0.7152*g+0.0722*b) #返回指定位置的像素，如果所打开的图像是多层次的图片，那这个方法就返回一个元组
            codePic = codePic + codeLib[int(((count-1)*gray)/256)]#建立灰度与字符集的映射
        codePic = codePic+'\n'
    return codePic

def console_print(image_file):
    rows,cols,_ = image_file.shape 
    image_file = cv2.resize(image_file,(int(cols*0.25),int(rows*0.12)))
    os.system('cls')
    print(transform(image_file))


main("bad.mp4")

版本二

样例

解决方案

使用python中的curses库来管理控制台，这个库是c写的所以很快，但是只适用于linux系统，windows需要自己手动安装非官方版。
curses安装教程

代码

没什么大的改动，只是改变了输出方式。

#-*- coding:utf-8 -*-  
import time
import cv2
import numpy as np
from PIL import Image
import os
import curses

stdscr = curses.initscr()
stdscr.border(0)
codeLib = '''@B%8&WM#*oahkbdpqwmZO0QLCJUYXzcvunxrjft/\|()1{}[]?-_+~<>i!lI;:,"^`'. '''#生成字符画所需的字符集
count = len(codeLib)

def main(video_path):
	'''
	对视频文件切割成帧
	@param video_path:视频路径
	'''
	vc=cv2.VideoCapture(video_path)
	c=0
	if vc.isOpened():
		rval,frame=vc.read()
	else:
		rval=False
	while rval:
		rval,frame=vc.read()
		#if c % 2 == 0:
		console_print(frame)
		#c=c+1

def transform(image_file):
    codePic = ''
    for row in range(0,image_file.shape[0]):  #size属性表示图片的分辨率，'0'为横向大小，'1'为纵向
        for col in range(0,image_file.shape[1]):
            r,g,b = image_file[row][col]
            gray = int(0.2126*r+0.7152*g+0.0722*b) #返回指定位置的像素，如果所打开的图像是多层次的图片，那这个方法就返回一个元组
            codePic = codePic + codeLib[int(((count-1)*gray)/256)]#建立灰度与字符集的映射
        codePic = codePic+'\n'
    return codePic

def console_print(image_file):
    rows,cols,_ = image_file.shape 
    image_file = cv2.resize(image_file,(int(cols*0.25),int(rows*0.12)))
    stdscr.addstr(0,0,transform(image_file))
    stdscr.refresh()


main("bad.mp4")