为啥我的python脚本输出文件为空

Posted

技术标签:

【中文标题】为啥我的python脚本输出文件为空【英文标题】:why my output file of python script is empty为什么我的python脚本输出文件为空 【发布时间】:2020-07-15 11:14:19 【问题描述】:

我正在创建一个 python 脚本来记录桌面屏幕。

在此我只取用户选择的区域,只记录选择的区域并将其转换为视频文件。

但在执行此操作时,脚本的输出是一个空的视频文件(0 字节)

谁能告诉我我哪里做错了。因为没有bbox(x1,x2,y1,y2) 也能正常工作。

代码如下: 编辑:

import tkinter as tk
from tkinter import *
from tkinter import ttk ,FLAT
from PIL import Image, ImageTk, ImageGrab, ImageEnhance
import cv2
import numpy as np
import threading
filename="test.avi"
fourcc = cv2.VideoWriter_fourcc(*'mp4v')
frame_rate = 10
root = tk.Tk()


def show_image(image):
    win = tk.Toplevel()
    win.image = ImageTk.PhotoImage(image)
    tk.Label(win, image=win.image).pack()
    win.grab_set()
    win.wait_window(win)

def area_sel():
    x1 = y1 = x2 = y2 = 0
    roi_image = None

    def on_mouse_down(event):
        nonlocal x1, y1
        x1, y1 = event.x, event.y
        canvas.create_rectangle(x1, y1, x1, y1, outline='red', tag='roi')

    def on_mouse_move(event):
        nonlocal roi_image, x2, y2
        x2, y2 = event.x, event.y
        canvas.delete('roi-image') 
        roi_image = image.crop((x1, y1, x2, y2)) 
        canvas.image = ImageTk.PhotoImage(roi_image)
        canvas.create_image(x1, y1, image=canvas.image, tag=('roi-image'), anchor='nw')
        canvas.coords('roi', x1, y1, x2, y2)
        canvas.lift('roi') 

    root.withdraw()  
    image = ImageGrab.grab()  
    bgimage = ImageEnhance.Brightness(image).enhance(0.3)  
    win = tk.Toplevel()
    win.attributes('-fullscreen', 1)
    win.attributes('-topmost', 1)
    canvas = tk.Canvas(win, highlightthickness=0)
    canvas.pack(fill='both', expand=1)
    tkimage = ImageTk.PhotoImage(bgimage)
    canvas.create_image(0, 0, image=tkimage, anchor='nw', tag='images')
    win.bind('<ButtonPress-1>', on_mouse_down)
    win.bind('<B1-Motion>', on_mouse_move)
    win.bind('<ButtonRelease-1>', lambda e: win.destroy())
    win.focus_force()
    win.grab_set()
    win.wait_window(win)
    root.deiconify()  

    if roi_image:
        region = x1, y1, x2, y2

        start_recording(region) #calling main function to record screen
        return region


def recording_screen(x1, y1, x2, y2):
    global recording
    recording = True

    while recording:

        img = ImageGrab.grab(bbox=(x1, y1, x2, y2))
        frame=np.array(img) # for recording
        out.write(cv2.cvtColor(frame,cv2.COLOR_RGB2BGR))

out = cv2.VideoWriter() 
def start_recording(region):
    x1,y1,x2,y2 = region
    if not out.isOpened():

            out.open(filename,fourcc, frame_rate,(x2-x1,y2-y1))
    threading.Thread(target=recording_screen, args=region, daemon=True).start()

def stop_recording():
    global recording
    recording = False
    out.release()

sel_area = ttk.Button(root, text='select area to Record', width=30, command=area_sel)
sel_area.grid(row=0, column=0)

stp_rec = ttk.Button(root, text='Stop Recording', width=30, command=stop_recording)
stp_rec.grid(row=0, column=1)

root.mainloop()

【问题讨论】:

尽量不要使用链接到您的代码。只需将其添加到您的帖子中。 确保video_size的值与所选区域的大小相同。 @IbrahimYousuf 我不能,因为该区域是由用户而不是我选择的。 区域被用户选中后,赋值给video_size,然后初始化cv2.VideoWriter 您确认x1x2y1y2 是您假设它们在recording_screen 中的值吗? 【参考方案1】:

您正在使用ImageGrab,它仅适用于 macOS 和 Windows,因此我无法在我的设置中测试您的完整代码。

您的代码中的问题是您将区域选择和视频录制结合在一起。选择区域后,您将拥有 x1y1x2y2。返回这些值。第一次初始化cv2.VideoWriter

您现在可以尝试使用传递给start_recording((x1, y1, x2, y2)) 的参数。

def start_recording(region):
    x1,y1,x2,y2 = region
    if not out.isOpened():

        out.open(filename,fourcc, frame_rate,(y2-y1,x2-x1)) #given x2 > x1, y2 > y1
    threading.Thread(target=recording_screen, args=region, daemon=True).start()

编辑:

问题是在以下调用中传递了错误的形状。

out.open(filename,fourcc, frame_rate,(y2-y1,x2-x1))

传递的Size 值为(y2-y1,x2-x1),即(height, width)。但是,构造函数需要(width, height)。请参阅文档here。

另外,img = ImageGrab.grab(bbox=(x1, y1, x2, y2)) 返回一个RGB image。但是out.write(frame) 需要BGR image。可以固定为:

img = ImageGrab.grab(bbox=(x1, y1, x2, y2))
frame = np.array(img)
out.write(cv2.cvtColor(frame, cv2.COLOR_RGB2BGR))

【讨论】:

return regionbefore start_recording(region)。此外,region = print(x1, y1, x2, y2) 调用返回 None。应该是region = x1, y1, x2, y2。我认为这些是故意的,所以我在这里提到了它们。 它可以工作,但质量很差。有什么可以解决的吗? 您如何定义质量? 视频有点模糊,因为当我全屏录制时它更清晰。 如果您以全屏方式查看裁剪后的视频,视频将被拉伸,因此像素化程度更高。

以上是关于为啥我的python脚本输出文件为空的主要内容,如果未能解决你的问题,请参考以下文章

为啥我的 Python3 脚本不愿将其输出通过管道传输到 head 或 tail(sys 模块)?

为啥我不能从我的 python 脚本创建可执行文件?

为啥从我的python脚本写入文件被覆盖[重复]

为啥我的脚本目录不在 Python sys.path 中?

为啥我的 python 脚本没有显示为进程,即使它正在运行?

为啥 Sonic Visualizer 和我的 Python 脚本之间的频谱分析存在 dB 差异?