2

我是一名业余程序员，正在尝试使用 python 在 Ubuntu 中制作屏幕录像机。使用此代码能够截取屏幕截图。

import wx
app = wx.App(False)
s = wx.ScreenDC()
w, h = s.Size.Get()
b = wx.EmptyBitmap(w, h)
m = wx.MemoryDCFromDC(s)
m.SelectObject(b)
m.Blit(0, 0, w, h, s, 0, 0)
m.SelectObject(wx.NullBitmap)
b.SaveFile("screenshot.png", wx.BITMAP_TYPE_PNG)

并使用循环拍摄更多照片并使用这些屏幕截图创建视频。我的代码如下所示，

import wx,os
app=wx.App(False)
s = wx.ScreenDC()
w, h = s.Size.Get()
b = wx.EmptyBitmap(w, h)
m = wx.MemoryDCFromDC(s)
i=0
while i<50:
   m.SelectObject(b)
   m.Blit(0, 0, w, h, s, 0, 0)
   m.SelectObject(wx.NullBitmap)
   b.SaveFile('{0:05d}.png'.format(i), wx.BITMAP_TYPE_PNG)
   i+=1
os.system('ffmpeg -f image2 -r 8 -i %05d.png -vcodec mpeg4 -y movie1.mp4')
i=0  
while i<50:
   os.remove('{0:05d}.png'.format(i))
   i += 1   `

在上面的代码中，我拍摄了 50 张图片并存储为 00000.png 到 00049.png 并使用 ffmpeg 制作视频。创建视频后，我删除所有图片。

当前问题：

  • 屏幕拍摄之间的延迟非常小。如果尝试使用此代码录制视频，则输出并不完美。
  • 录制时间长，效率不高。存储屏幕截图需要大量硬盘内存。并使用更多的CPU。

我做什么来让代码更有效率？使用纯python如何从图片创建视频？有没有其他方法来记录屏幕？我喜欢改进我的代码。

4

1 回答 1

2

按照您编写代码的方式，您必须等待每个文件保存后才能获取下一个屏幕截图。这就是您的“非常小的延迟”的来源。

您可以抓取内存中的所有快照，然后在最后写入它们：

snapshots = []
for i in range(20):
   b = wx.EmptyBitmap(w, h)
   m.SelectObject(b)
   m.Blit(0, 0, w, h, s, 0, 0)
   m.SelectObject(wx.NullBitmap)
   snapshots.append(b)
for snapshot in snapshots:
   snapshot.SaveFile('{0:05d}.png'.format(i), wx.BITMAP_TYPE_PNG)

但这会占用大量内存——可能足以让malloc调用（或者更糟糕的是，交换 thrash）减慢您的速度。

另一种选择是将注销推送到后台线程。（由于工作可能是 I/O 绑定的，所以普通的 Python 线程在这里应该没问题。）例如：

with concurrent.futures.ThreadPoolExecutor() as executor:
    for i in range(20):
       b = wx.EmptyBitmap(w, h)
       m.SelectObject(b)
       m.Blit(0, 0, w, h, s, 0, 0)
       m.SelectObject(wx.NullBitmap)
       executor.submit(b.SaveFile, '{0:05d}.png'.format(i), wx.BITMAP_TYPE_PNG)

但无论如何，如果你解决了这个问题……你不希望快照尽可能快地出现。例如，如果您的屏幕刷新率是 60 次/秒，并且您的 20 个循环都在前 1/60 秒内完成，那么您将只获得同一帧的 20 个副本。那不好。因此，您确实需要为快照编写某种简单的调度程序。（通常，屏幕截图实际上比屏幕刷新率慢得多——1/20 秒，甚至 1/6 或 1/2。）

幸运的是，wx有一些很好的方法可以做到这一点。例如，用于wx.Timer每 1000/60 毫秒运行一次代码。

这还有一个额外的优势，即您在拍摄快照时不会阻塞整个事件循环，只会每 1/60（或 1/20 或其他）秒短暂阻塞一次。

当然，您可以并且可能必须将这两种解决方案结合起来：使用计时器或其他调度程序来设置最大帧速率，并卸载一些工作以尽可能接近该最大值。

至于处理更长的屏幕抓取：如果您尝试将数万帧保存为 PNG 文件，那将占用大量空间。（为什么？嗯，PNG 使用无损压缩——它可能会让你的 4MB 屏幕抓取到 400K。但是 H264 视频使用有损压缩，更重要的是，只需要跟踪从一帧到下一帧的变化，所以它可能只每个关键帧取 100K，每个差异帧取 4K。）

您可以做的一件事是每隔一段时间（例如，每 120 帧）启动一项工作以将它们压缩成一个视频并删除它们，它们最后将所有视频连接起来。为简单起见，您可能希望每个批次的第一帧成为新的关键帧。MP4/H264 不是最有趣的连接格式，但它仍然可能比切换到流压缩器更容易。

于 2013-08-02T22:21:12.170 回答