只需要在最后一行括号填入视频路径即可
import os
from PIL import Image, ImageDraw, ImageFont
import cv2
import time
from tqdm import tqdm
import threading
if __name__ == '__main__':
# path = r'img\sportball\0.jpg'
def cv2_pit_imgchar(path,save_path=''):
def create_img(text,show='no',save_path=save_path):#创建图片,将文本写入到图片
s = time.time()
img = Image.new('RGB', (1260, 915), color = (255, 255, 255))#创建一张图片
draw = ImageDraw.Draw(img)
font = ImageFont.truetype('SIMLI.TTF', 10)#字体,文字大小
draw.text((1,0),text, font=font, fill=(0, 0, 0))#起始坐标,字符串,字体,字体颜色
e = time.time()
# print(f'创建图片写入字符耗时{round(e-s,2)}s')
if show=='yes':#如果show为yes,
img.show()
if not save_path=='':#如果路径不为空,保存图片
img.save(save_path)
return img
ASCII_CHARS = ['.',',',':',';','+','*','?','T','P','%','@']
# ascii_chars = "MNHQ$OC67+>!:-. "
# ASCII_CHARS=list(ascii_chars)
ASCII_CHARS = ASCII_CHARS[::-1]#反转
img1 = cv2.imread(path,cv2.IMREAD_GRAYSCALE)#读取图片并转为灰度图
def cv2_img_resize(img1,w_size):#缩放图片
h,w = img1.shape[0:2]#获取图片高宽
# print(f"图片原始高度{h}宽度{w}")
re_rate=h/w#缩放比例
# print(f'比例{re_rate}')
img1=cv2.resize(img1,(w_size,int(w_size*re_rate/2)))#缩放图片,因为字符宽高不一致需要压缩一半图片高度
return img1
img1 = cv2_img_resize(img1,250)
def img_to_char(img1):#将图片按灰度值转为字符
img_text = ''
s1 = time.time()
for y in img1:
for x in y:
new_pixels = [ASCII_CHARS[x//25]]
new_pixels= ''.join(new_pixels)
img_text+=str(new_pixels)
# print(x,end=',')
img_text+='\n'
e1 = time.time()
# print(f'读取全图灰度值耗时{round(e1-s1,2)}s')
return img_text
img_text = img_to_char(img1)
create_img(img_text)#创建图片
# cv2_pit_imgchar(path)
def get_folder_imgs(folder):#读取获得文件夹下所有jpg文件路径,返回图片路径列表
x=0
path_list = []
for img_name in os.listdir(folder):
if '.jpg' in img_name:
# print(img_name)
path_list.append(folder+img_name)
x+=1
print(f'总共{x}张图片')
return path_list
def video_to_imgs(video_path = r"D:\1111\sportballcn.mp4",save_path='img/sportball/',th=4):#将视频拆成图片
video = cv2.VideoCapture(video_path)
fps = int(video.get(cv2.CAP_PROP_FPS)) # 获取视频每秒的帧数
if not os.path.exists(save_path):
print('创建保存路径')
os.makedirs(save_path)
def run(t,th):
video = cv2.VideoCapture(video_path)
frames = video.get(cv2.CAP_PROP_FRAME_COUNT)#获取视频总帧数
print(f'第{t+1}个线程启动,处理{int(frames*t/th)}到{int(frames*(t+1)/th)}帧')
t1 = time.time()
for x in tqdm(range(int(frames))):
retval,image = video.read()
if not retval:
break
if frames*t/th<=x<=frames*(t+1)/th:
cv2.imwrite(f'{save_path}{x}.jpg',image)
t2 = time.time()
print(f"线程{t+1},耗时{round(t2-t1,2)}s")
tt = dict()
for t in range(th):
tt[t]=threading.Thread(target=run,args=(t,th))
for t in range(th):
tt[t].start()
for t in range(th):
tt[t].join()
#print('\n\n\n\n\n\n\naaaaaaaaaaaa')
return fps
def batch_img_to_charimg(folder='img/sportball/',save_path='img/spaortchar/',th=1):#批量将图片转为字符图
path_list = get_folder_imgs(folder)
if not os.path.exists(save_path):
print(f'创建字符图保存路径{save_path}')
os.makedirs(save_path)
print('正在将图片转换为字符图')
img_counting = len(path_list)
def run(t,th):
t1=time.time()
print(f'第{t+1}个线程启动,处理{int(img_counting*t/th)}到{int(img_counting*(t+1)/th)}张图')
x=0
for path in tqdm(path_list):
name = path.split('/')[-1]
if img_counting*t/th<=x<=img_counting*(t+1)/th:
cv2_pit_imgchar(path=path,save_path=f'{save_path}{name}')
x+=1
t2 = time.time()
print(f"线程{t+1},耗时{round(t2-t1,2)}s")
tt=dict()
for t in range(th):
tt[t]=threading.Thread(target=run,args=(t,th))
for t in range(th):
tt[t].start()
for t in range(th):
tt[t].join()
def char_imgs_to_video(char_img_folder = "img/spaortchar",save_name='streetchar.mp4',fps=30): #将所有字符图合成视频 #包含所有字符图的路径
fps=fps # 每秒帧数，最好跟原视频一致
size = (1260, 915) # (宽, 高) 必须跟图片一致
fourcc = cv2.VideoWriter_fourcc(*'mp4v') # 编码为 mp4v 格式
char_video = cv2.VideoWriter(save_name, fourcc, fps, size, isColor=True)
img_names = os.listdir(char_img_folder)
img_names.sort(key=lambda x:int(x[:-4]))#进行排序
print('将字符图合并成视频中...')
for img_name in tqdm(img_names):
if '.jpg' in img_name:
# print(char_img_folder+'/'+img_name)
img = cv2.imread(char_img_folder+'/'+img_name)
char_video.write(img)#将图片添加视频
char_video.release()
print('完成')
def batch_img_to_charimg1(folder='img/aq/',save_path='img/aqchar/'):#批量将图片转为字符图
path_list = get_folder_imgs(folder)
if not os.path.exists(save_path):
print(f'创建字符图保存路径{save_path}')
os.makedirs(save_path)
print('正在将图片转换为字符图')
for path in tqdm(path_list):
name = path.split('/')[-1]
cv2_pit_imgchar(path=path,save_path=f'{save_path}{name}')
def v2cv(video_path = r"D:\1111\aq.mp4"):#video_path需要转换的视频路径
s_p1 = 'img/img/'#图片保存路径
s_p2 = 'img/charimg/'#字符图保存路径
fps=30
# 第一步,将视频拆成图片
fps=video_to_imgs(video_path=video_path,save_path=s_p1,th=4)
# 第二部,将图片批量转换成字符图
batch_img_to_charimg(folder=s_p1,save_path=s_p2,th=1)#多线程很慢是啥情况
# 第三步,将字符图合成视频
char_imgs_to_video(char_img_folder = s_p2,save_name='char.mp4',fps=fps)
v2cv('d:/1111/sd.mp4')
