我在 python 中使用 imageio 以打开目录中的所有视频文件并将它们转换为 numpy 数组。
这是我正在使用的脚本：
1 from __future__ import print_function
2 from avi_to_numpy import *
3 from os import listdir
4 import numpy as np
5 import imageio
6
7 class_path = '../Diving/'
8 max_frames = 16
9 stride = 8
10 videos = [vid for vid in listdir(class_path)]
11 train = []
12
13 for vid in videos:
14 print(str.format('Loading {}...', vid), end="")
15 filename = class_path + vid
16 reader = imageio.get_reader(filename, 'ffmpeg')
17 frames = []
18
19 for i, im in enumerate(reader):
20 if len(frames) == max_frames:
21 break
22
23 if i % stride == 0:
24 frames.append(im)
25
26 reader.close()
27 train.append(np.array(frames))
28 print('done')
29
30
31 print(len(train))
最终此脚本崩溃并出现以下错误输出：
Traceback (most recent call last):
File "load_class_test.py", line 16, in <module>
reader = imageio.get_reader(filename, 'ffmpeg')
File "/usr/local/lib/python2.7/site-packages/imageio/core/functions.py", line 111, in get_reader
return format.get_reader(request)
File "/usr/local/lib/python2.7/site-packages/imageio/core/format.py", line 158, in get_reader
return self.Reader(self, request)
File "/usr/local/lib/python2.7/site-packages/imageio/core/format.py", line 207, in __init__
self._open(**self.request.kwargs.copy())
File "/usr/local/lib/python2.7/site-packages/imageio/plugins/ffmpeg.py", line 260, in _open
self._initialize()
File "/usr/local/lib/python2.7/site-packages/imageio/plugins/ffmpeg.py", line 326, in _initialize
stdout=sp.PIPE, stderr=sp.PIPE)
File "/usr/local/Cellar/python/2.7.11/Frameworks/Python.framework/Versions/2.7/lib/python2.7/subprocess.py", line 710, in __init__
errread, errwrite)
File "/usr/local/Cellar/python/2.7.11/Frameworks/Python.framework/Versions/2.7/lib/python2.7/subprocess.py", line 1223, in _execute_child
errpipe_read, errpipe_write = self.pipe_cloexec()
File "/usr/local/Cellar/python/2.7.11/Frameworks/Python.framework/Versions/2.7/lib/python2.7/subprocess.py", line 1175, in pipe_cloexec
r, w = os.pipe()
OSError: [Errno 24] Too many open files
我正在从 imageio 关闭 Reader 对象。似乎 ffmpeg 打开的文件没有正确关闭。
我在这里缺少一个明显的步骤吗？我是否正确关闭了文件？
编辑：找到临时解决方案。在 github 上打开了一个新问题。
我可以通过取消注释“imageio/plugins/ffmpeg.py”中的以下代码行来解决该问题：
381 def _close_streams(self):
382 for std in (self._proc.stdin,
383 self._proc.stdout,
384 self._proc.stderr):
385 try:
386 std.close()
387 except Exception: # pragma: no cover
388 pass
然后我在以下位置添加了对上述函数的调用_close(self)：
271 def _close(self):
272 self._terminate(0.05) # Short timeout
273 self._close_streams()
274 self._proc = None
我不确定这样做的副作用是什么，但它为我提供了一个解决方案。