视频转字符画

python烧烤吧

1. import cv2
   
2. import tkinter as tk
   
3. import os.path as op
   
4. from tkinter import messagebox as mgb
   
5. from tkinter import ttk
   
6. from tkinter import filedialog as fdl
   
7. 
   
8. def video2strs(path_, root):
   
9. 
   
10.     if not op.isfile(path_) or not path_.split(".")[1] == "mp4":
    
11.         mgb.showwarning(title="Warning", messge="目前仅支持MP4格式的转换！！")
    
12.         return
    
13. 
    
14.     dir_name, file_name = op.split(path_)
    
15.     new_text_position = dir_name + r"/"+ file_name.split(".")[0]
    
16. 
    
17.     #os.chdir(dir_name)
    
18.     _ = 0
    
19.     suffix = ".v2s"
    
20.     while op.isfile(new_text_position + suffix):
    
21.         _ += 1
    
22.         suffix = f"({_}).v2s"
    
23.     new_text_position += suffix
    
24.     print(new_text_position)
    
25. 
    
26.     def get_resize_height():
    
27.         _ = cv2.VideoCapture(path_)
    
28.         ret, temp_frame = _.read()
    
29.         temp_frame = cv2.resize(temp_frame, (0, 0), fx=0.09375,fy=0.046875)
    
30.         __ = temp_frame.shape[0]
    
31.         _.release()
    
32.         return __
    
33. 
    
34.     def start_trans():
    
35. 
    
36.         text_f = open(new_text_position, "w", encoding="utf-8")
    
37.         text_f.write(str(get_resize_height()) + "\n")
    
38.         #with open(new_text_position, "w", encoding="utf-8") as text_f:
    
39. 
    
40.         def trans():
    
41.             nonlocal count
    
42.             if video.isOpened():
    
43.                 temp_text = ""
    
44.                 ret, current_frame = video.read()
    
45.                 if ret:
    
46.                     current_frame = cv2.resize(cv2.cvtColor(current_frame, cv2.COLOR_BGR2GRAY), (0, 0), fx=0.09375,fy=0.046875)
    
47.                     for i in range(current_frame.shape[0]):
    
48.                         for j in range(current_frame.shape[1]):
    
49.                             temp_text += ascii__[int(current_frame[i, j] / 255 * 7)]
    
50.                         temp_text += '\n'
    
51.                     else:
    
52.                         text_f.write(temp_text)
    
53.                         if count > one_of_ten:
    
54.                             progress_bar["value"] += 10
    
55.                             progress_bar.pack(pady=(1, 2))
    
56.                             count = 0
    
57.                         show_progress.after(0, trans)
    
58.                         count += 1
    
59.                         #print("+1f")
    
60.                 else:
    
61.                     label1.config(text="OK!")
    
62.                     progress_bar["value"] = 100
    
63.                     progress_bar.pack(pady=(1, 2))
    
64.                     text_f.close()
    
65.                     tk.Button(show_progress, text="Quit", command=show_progress.destroy, width=5).pack(pady=15)
    
66.                     video.release()
    
67. 
    
68.         video = cv2.VideoCapture(path_)
    
69.         amount_of_frame = video.get(cv2.CAP_PROP_FRAME_COUNT)
    
70.         one_of_ten = amount_of_frame // 10
    
71.         temp_text = ""
    
72.         ascii__ = ["@", "G·", "￥", "·*·", "·+", "·=", "-'-", "·_·"]
    
73.         count = 0
    
74.         show_progress.after(0, trans)
    
75. 
    
76.     show_progress = tk.Tk()
    
77.     show_progress.geometry("300x180")
    
78. 
    
79.     label1 = tk.Label(show_progress, text="Patience！", justify="center")
    
80.     progress_bar = ttk.Progressbar(show_progress, length=200, mode="determinate")
    
81.     label1.pack(pady=(30, 1))
    
82.     progress_bar.pack(pady=(1, 2))
    
83.     show_progress.after(0, start_trans) # after方法 第二个参数之后是收集参数
    
84. 
    
85. class V2sTrans:
    
86.     def __init__(self, root):
    
87.         self.root = root
    
88.         self.root.geometry(f"500x400+{self.root.winfo_screenwidth()//2-250}+{self.root.winfo_screenheight()//2-200}")
    
89.         self.init_win()
    
90. 
    
91.     def init_win(self):
    
92. 
    
93.         def set_dir():
    
94. 
    
95.             path_ = fdl.askopenfilename(title="选择视频文件", filetypes=[("MP4", '.mp4')])
    
96. 
    
97.             dir_entry.delete(0, "end")
    
98.             dir_entry.insert(0,path_)
    
99. 
    
100.         def set_dir_play():
     
101.             path_ = fdl.askopenfilename(title="选择v2s文件", filetypes=[("V2S", '.v2s')])
     
102. 
     
103.             dir_entry_.delete(0, "end")
     
104.             dir_entry_.insert(0, path_)
     
105. 
     
106. 
     
107.         l_frame0 = tk.LabelFrame(master=self.root, text="V2S", height=30)
     
108.         l_frame0.pack(fill='x', padx=20, pady=10, ipady=50)
     
109. 
     
110.         l_frame1 = tk.LabelFrame(master=l_frame0, text="输入mp4文件路径")
     
111.         l_frame1.pack(fill="x", padx=20, pady=(50, 0))
     
112. 
     
113.         dir_entry = tk.Entry(master=l_frame1, width=40)
     
114.         dir_choice_button = tk.Button(master=l_frame1, text="...", relief="raised", command=set_dir, width=4)
     
115.         dir_entry.grid(row=0, column=0, padx=(45, 5), pady=(5,10))
     
116.         dir_choice_button.grid(row=0, column=1, padx=(5, 10), pady=(5,10))
     
117. 
     
118.         start_trans_button = tk.Button(master=l_frame0, text="转换", width=20,
     
119.                                        command=lambda: video2strs(dir_entry.get(), self.root))
     
120.         start_trans_button.pack(pady=10)
     
121. 
     
122.         l_frame2 = tk.LabelFrame(master=l_frame0, text="输入v2s文件路径")
     
123.         l_frame2.pack(fill="x", padx=20, pady=(20, 0))
     
124. 
     
125.         dir_entry_ = tk.Entry(master=l_frame2, width=40)
     
126.         dir_choice_button_ = tk.Button(master=l_frame2, text="...", relief="raised", command=set_dir_play, width=4)
     
127.         dir_entry_.grid(row=0, column=0, padx=(45, 5), pady=(5,10))
     
128.         dir_choice_button_.grid(row=0, column=1, padx=(5, 10), pady=(5,10))
     
129. 
     
130.         start_trans_button = tk.Button(master=l_frame0, text="播放", width=20,
     
131.                                        command=lambda: play_v2s(dir_entry_.get()))
     
132. 
     
133.         start_trans_button.pack(pady=10)
     
134. 
     
135. def play_v2s(path_):
     
136.     with open(path_, "r", encoding="utf-8") as f:
     
137.         v2s = f.readlines()
     
138.         height = int(v2s[0].split("\")[0])
     
139.         print(height)
     
140.         v2s.pop(0)
     
141. 
     
142.     frames = []
     
143. 
     
144.     count = 0
     
145.     temp_str = ""
     
146.     for i in v2s:
     
147.         if count == height:
     
148.             count = 0
     
149.             frames.append(temp_str)
     
150.             temp_str = ""
     
151.         temp_str += i
     
152.         count += 1
     
153. 
     
154.     frames_iter = iter(frames)
     
155. 
     
156.     def update_frame():
     
157.         label.config(text=next(frames_iter))
     
158.         label.pack()
     
159.         show_v2s.after(27, update_frame)
     
160. 
     
161. 
     
162.     show_v2s = tk.Tk()
     
163.     show_v2s.title("Playing")
     
164.     #show_v2s.geometry(f"800x600+{show_v2s.winfo_screenwidth()//2-400}+{show_v2s.winfo_screenheight()//2-300}")
     
165.     label = tk.Label(show_v2s, text="Ready", justify="center", width=1024, height=720, font=("微软雅黑", 6))
     
166.     label.pack()
     
167.     show_v2s.after(3000, update_frame)
     
168.     show_v2s.mainloop()
     
169. 
     
170. if __name__ == '__main__':
     
171.     root_ = tk.Tk()
     
172.     root__ = V2sTrans(root_)
     
173.     root__.root.mainloop()

复制代码

FishC

首先，我来帮你修正这段视频转字符画的代码。主要问题包括语法错误、逻辑错误和未定义的变量。以下是修正后的代码，并附上关键说明。

关键修正点：

1. 文件扩展名检查：使用 path_.endswith('.mp4') 替代错误的 split 比较。
2. 文件路径拼接：正确获取文件名（不带扩展名）来生成输出文件路径。
3. 字符映射：定义了 ascii_chars 字符串，用于将灰度值映射到字符。
4. 帧处理：在转换每个帧时，正确遍历像素并映射到字符。
5. 进度条更新：使用 progress_bar['value'] 来更新进度。
6. v2s 文件解析：在播放部分，正确读取和分割帧数据。

修正后的代码：

1. 
   
2. import cv2
   
3. import tkinter as tk
   
4. import os.path as op
   
5. from tkinter import messagebox as mgb
   
6. from tkinter import ttk
   
7. from tkinter import filedialog as fdl
   
8. 
   
9. def video2strs(path_, root):
   
10.     if not op.isfile(path_) or not path_.endswith('.mp4'):
    
11.         mgb.showwarning(title="Warning", message="目前仅支持MP4格式的转换！！")
    
12.         return
    
13. 
    
14.     dir_name, file_name = op.split(path_)
    
15.     file_base_name = op.splitext(file_name)[0]  # 获取不带扩展名的文件名
    
16.     new_text_position = op.join(dir_name, file_base_name)  # 使用 os.path.join 确保路径正确
    
17. 
    
18.     _ = 0
    
19.     suffix = ".v2s"
    
20.     while op.isfile(new_text_position + suffix):
    
21.         _ += 1
    
22.         suffix = f"({_}).v2s"
    
23.     new_text_position += suffix
    
24.     print("输出文件路径:", new_text_position)
    
25. 
    
26.     def get_resize_height():
    
27.         cap = cv2.VideoCapture(path_)
    
28.         ret, temp_frame = cap.read()
    
29.         if ret:
    
30.             # 调整帧大小，这里设置为宽度120，高度按比例计算
    
31.             height, width = temp_frame.shape[:2]
    
32.             new_width = 120
    
33.             new_height = int(height * (new_width / width) * 0.5)  # 高度减半以匹配字符画比例
    
34.             temp_frame = cv2.resize(temp_frame, (new_width, new_height))
    
35.             cap.release()
    
36.             return (new_height, new_width)  # 返回调整后的高度和宽度
    
37.         cap.release()
    
38.         return (0, 0)
    
39. 
    
40.     def start_trans():
    
41.         # 定义字符映射，用于将灰度值转换为字符
    
42.         ascii_chars = "@%#*+=-:. "
    
43. 
    
44.         text_f = open(new_text_position, "w", encoding="utf-8")
    
45.         resize_dim = get_resize_height()
    
46.         text_f.write(f"{resize_dim[0]},{resize_dim[1]}\n")  # 写入高度和宽度，用逗号分隔
    
47. 
    
48.         video = cv2.VideoCapture(path_)
    
49.         amount_of_frame = int(video.get(cv2.CAP_PROP_FRAME_COUNT))
    
50.         one_of_ten = max(amount_of_frame // 10, 1)  # 避免除零
    
51.         count = 0
    
52. 
    
53.         def trans():
    
54.             nonlocal count
    
55.             if video.isOpened():
    
56.                 ret, current_frame = video.read()
    
57.                 if ret:
    
58.                     # 调整帧大小并转换为灰度图
    
59.                     current_frame = cv2.cvtColor(current_frame, cv2.COLOR_BGR2GRAY)
    
60.                     current_frame = cv2.resize(current_frame, (resize_dim[1], resize_dim[0]))
    
61.                     temp_text = ""
    
62.                     for i in range(resize_dim[0]):  # 高度
    
63.                         for j in range(resize_dim[1]):  # 宽度
    
64.                             gray = current_frame[i, j]
    
65.                             # 将灰度值映射到字符索引
    
66.                             index = int(gray / 255 * (len(ascii_chars) - 1))
    
67.                             temp_text += ascii_chars[index]
    
68.                         temp_text += '\n'
    
69.                     text_f.write(temp_text)
    
70.                     count += 1
    
71.                     # 更新进度条
    
72.                     if count % one_of_ten == 0:
    
73.                         progress_bar['value'] += 10
    
74.                         progress_bar.update()
    
75.                     show_progress.after(1, trans)  # 使用 after 继续处理下一帧
    
76.                 else:
    
77.                     # 处理完成
    
78.                     label1.config(text="OK!")
    
79.                     progress_bar['value'] = 100
    
80.                     progress_bar.update()
    
81.                     text_f.close()
    
82.                     video.release()
    
83.                     tk.Button(show_progress, text="Quit", command=show_progress.destroy, width=5).pack(pady=15)
    
84.             else:
    
85.                 mgb.showerror("Error", "无法打开视频文件")
    
86. 
    
87.         show_progress.after(0, trans)
    
88. 
    
89.     # 创建进度窗口
    
90.     show_progress = tk.Toplevel(root)  # 使用 Toplevel 而非 Tk，避免多个主窗口
    
91.     show_progress.geometry("300x180")
    
92.     show_progress.title("转换进度")
    
93. 
    
94.     label1 = tk.Label(show_progress, text="Patience！", justify="center")
    
95.     progress_bar = ttk.Progressbar(show_progress, length=200, mode="determinate", maximum=100)
    
96.     label1.pack(pady=(30, 1))
    
97.     progress_bar.pack(pady=(1, 2))
    
98.     show_progress.after(0, start_trans)
    
99. 
    
100. class V2sTrans:
     
101.     def __init__(self, root):
     
102.         self.root = root
     
103.         self.root.geometry(f"500x400+{self.root.winfo_screenwidth()//2-250}+{self.root.winfo_screenheight()//2-200}")
     
104.         self.root.title("视频转字符画")
     
105.         self.init_win()
     
106. 
     
107.     def init_win(self):
     
108.         def set_dir():
     
109.             path_ = fdl.askopenfilename(title="选择视频文件", filetypes=[("MP4 files", "*.mp4")])
     
110.             if path_:
     
111.                 dir_entry.delete(0, "end")
     
112.                 dir_entry.insert(0, path_)
     
113. 
     
114.         def set_dir_play():
     
115.             path_ = fdl.askopenfilename(title="选择v2s文件", filetypes=[("V2S files", "*.v2s")])
     
116.             if path_:
     
117.                 dir_entry_.delete(0, "end")
     
118.                 dir_entry_.insert(0, path_)
     
119. 
     
120.         l_frame0 = tk.LabelFrame(master=self.root, text="V2S", height=30)
     
121.         l_frame0.pack(fill='x', padx=20, pady=10, ipady=50)
     
122. 
     
123.         l_frame1 = tk.LabelFrame(master=l_frame0, text="输入mp4文件路径")
     
124.         l_frame1.pack(fill="x", padx=20, pady=(50, 0))
     
125. 
     
126.         dir_entry = tk.Entry(master=l_frame1, width=40)
     
127.         dir_choice_button = tk.Button(master=l_frame1, text="...", relief="raised", command=set_dir, width=4)
     
128.         dir_entry.grid(row=0, column=0, padx=(45, 5), pady=(5,10))
     
129.         dir_choice_button.grid(row=0, column=1, padx=(5, 10), pady=(5,10))
     
130. 
     
131.         start_trans_button = tk.Button(master=l_frame0, text="转换", width=20,
     
132.                                        command=lambda: video2strs(dir_entry.get(), self.root))
     
133.         start_trans_button.pack(pady=10)
     
134. 
     
135.         l_frame2 = tk.LabelFrame(master=l_frame0, text="输入v2s文件路径")
     
136.         l_frame2.pack(fill="x", padx=20, pady=(20, 0))
     
137. 
     
138.         dir_entry_ = tk.Entry(master=l_frame2, width=40)
     
139.         dir_choice_button_ = tk.Button(master=l_frame2, text="...", relief="raised", command=set_dir_play, width=4)
     
140.         dir_entry_.grid(row=0, column=0, padx=(45, 5), pady=(5,10))
     
141.         dir_choice_button_.grid(row=0, column=1, padx=(5, 10), pady=(5,10))
     
142. 
     
143.         start_play_button = tk.Button(master=l_frame0, text="播放", width=20,
     
144.                                       command=lambda: play_v2s(dir_entry_.get()))
     
145.         start_play_button.pack(pady=10)
     
146. 
     
147. def play_v2s(path_):
     
148.     if not op.isfile(path_) or not path_.endswith('.v2s'):
     
149.         mgb.showwarning(title="Warning", message="请选择有效的 .v2s 文件！")
     
150.         return
     
151. 
     
152.     with open(path_, "r", encoding="utf-8") as f:
     
153.         lines = f.readlines()
     
154.         if not lines:
     
155.             mgb.showerror("Error", "文件为空或格式错误")
     
156.             return
     
157.         # 第一行是高度和宽度
     
158.         first_line = lines[0].strip()
     
159.         try:
     
160.             height, width = map(int, first_line.split(','))
     
161.         except:
     
162.             mgb.showerror("Error", "文件格式错误：第一行应为高度,宽度")
     
163.             return
     
164.         # 移除第一行
     
165.         char_lines = lines[1:]
     
166. 
     
167.     # 将字符数据分割成帧
     
168.     frames = []
     
169.     temp_str = ""
     
170.     count = 0
     
171.     for line in char_lines:
     
172.         if count < height:
     
173.             temp_str += line
     
174.             count += 1
     
175.         if count == height:
     
176.             frames.append(temp_str)
     
177.             temp_str = ""
     
178.             count = 0
     
179.     # 检查是否有剩余数据
     
180.     if temp_str:
     
181.         frames.append(temp_str)
     
182. 
     
183.     if not frames:
     
184.         mgb.showerror("Error", "没有有效的帧数据")
     
185.         return
     
186. 
     
187.     frames_iter = iter(frames)
     
188.     show_v2s = tk.Tk()
     
189.     show_v2s.title("Playing")
     
190.     # 设置窗口大小，根据字符画尺寸调整
     
191.     label = tk.Label(show_v2s, text="Ready", justify="left", font=("Courier", 10))  # 使用等宽字体
     
192.     label.pack()
     
193. 
     
194.     def update_frame():
     
195.         try:
     
196.             frame_text = next(frames_iter)
     
197.             label.config(text=frame_text)
     
198.             show_v2s.after(27, update_frame)  # 约37帧每秒，但字符画可能较慢，可调整
     
199.         except StopIteration:
     
200.             label.config(text="播放完毕")
     
201.             tk.Button(show_v2s, text="关闭", command=show_v2s.destroy).pack()
     
202. 
     
203.     show_v2s.after(100, update_frame)  # 延迟开始播放
     
204.     show_v2s.mainloop()
     
205. 
     
206. if __name__ == '__main__':
     
207.     root_ = tk.Tk()
     
208.     app = V2sTrans(root_)
     
209.     root_.mainloop()
     

复制代码

注意事项：

1. 字符映射：使用了 ascii_chars = "@%#*+=-:. " 作为灰度到字符的映射，你可以根据需要调整字符顺序或数量。
2. 帧大小调整：在 get_resize_height 中，将帧宽度固定为120，高度按比例计算并减半，以适应字符画的比例（字符通常高度大于宽度）。
3. 进度条更新：使用 progress_bar['value'] += 10 来逐步更新进度。
4. 播放部分：使用 font=("Courier", 10) 等宽字体，以确保字符对齐。
5. 错误处理：添加了基本的错误检查，如文件是否存在、格式是否正确。

运行前请确保已安装 OpenCV：可以通过 pip install opencv-python 安装。如果遇到问题，请检查视频文件路径和输出目录的写入权限。

以上回复来自 FishC AI 小助理，如未能正确解答您的问题，请继续追问。