python - 一次读取成对的图像文件

Question

因此，我遍历目录并尝试读取一些图像文件，如下所示:

for root, dirs, files in os.walk(path):
    for file in files:
            im1 = cv2.imread(file)

如何一次读取一对一对 图像？这样，在上面的代码中，我有类似以下内容:

im2 = cv2.imread(file)

其中的file代表紧随第一张图片之后的即时图片。

谢谢。

Answer 1

要处理文件对:

import os
import itertools

path = "SRC PATH"
for root, dirs, files in os.walk(path):
    for file1, file2 in itertools.izip_longest(files[::2], files[1::2]):
        im1 = cv2.imread(file1)
        if file2:
            im2 = cv2.imread(file2)

我使用 izip_longest 方法只是为了防止您的目录中只有奇数个文件。