根据文件名以一定顺序循环浏览文件夹中的所有图像

Question

因此，我想创建一个将图像拼接在一起的实用程序。 到目前为止，我有以下代码：

def merger(file1, file2):

    # File I/O: Open source images
    image1 = Image.open(file1)
    image2 = Image.open(file2)

    # Read the source image dimensions using pillow Image class
    (width1, height1) = image1.size
    (width2, height2) = image2.size

    # Calculate the output image dimensions
    merged_width = width1 + width2          # The width is a sum of the 2 widths
    merged_height = max(height1, height2)   # The height is the larger of the two

    # Merge the images using pillow Image class
    output = Image.new('RGB', (merged_width, merged_height))   # Create new image object (for output)
    output.paste(im=image1, box=(0, 0))                        # Paste the 1st source image into the output object
    output.paste(im=image2, box=(width1, 0))                   # Paste the 2nd source image into the output object
    return output

如何循环浏览文件夹中的所有图像文件？ 我想我会使用循环，但是如何读取文件夹中存在的每对图像文件，从文件名中识别它们的顺序，将每对图像缝合在一起，然后转到下一对文件？

文件应根据文件名中的数字进行拼接。 例子：

1.jpg和2.jpg应该第一缝制，然后3.jpg和4.jpg ， 5.jpg和6.jpg等。

要么

应该先缝合01.jpg和02.jpg ，然后再缝合03.jpg和04.jpg ， 05.jpg和06.jpg等等。

要么

scan01.tif和scan02.tif ，然后scan03.tif和scan04.tif ， scan05.tif和scan06.tif ...

要么

newpic0001.png和newpic0002.png ，然后是newpic0003.png和newpic0004.png ，然后是newpic0005.png和newpic0006.png ...

您有个主意：按照文件末尾的数字进行操作，而忽略前导零。

如果这很重要，我将枕头用于图像处理，将tkinter用于GUI。

谢谢。

Answer 1

尝试这个：

import os

file_list = [x for x in sorted([x for x in os.listdir('/path/to/directory/')])]
for i in range (0, len(file_list), 2):
    if i+1 < len(file_list):
        f1, f2 = file_list[i], file_list[i+1]
    else:
        f1, f2 = file_list[i], None
    # do your merge magic here with f1 and f2

参考： 列出路径中的所有目录

Answer 2

这感觉有些残酷，但是有效。 想法是获取文件名列表，然后仅使用文件名中的数字对它们进行排序。

import os
import re


def sort_filenames(all_files):
    filenames_sorted = []
    original_filenames = {}
    for full_filename in all_files:
        filename, file_extension = os.path.splitext(full_filename)

        # Save all the files names to be sorted
        filenames_sorted.append(filename)
        # Save original full filename in a dictionary for later retrieval
        original_filenames[filename] = full_filename

    # Sort the list using our own key
    filenames_sorted.sort(key=get_file_key)
    filenames = []
    for key in filenames_sorted:
        filenames.append(original_filenames[key])

    return filenames


def get_file_key(filename):
    # Remove all non-digits from the filename
    key = re.sub("[^0-9]", "", filename)
    return int(key)


# Start with the list of all files in the current directory
all_files = os.listdir("./")

# FOR TESTING ONLY....fill all_files with some testing examples
all_files = ['scan01.tif', 'scan04.tif', 'scan03.tif', 'scan05.tif', 'scan06.tif', 'scan02.tif']
all_files = ['newpic0002.png', 'newpic0001.png', 'newpic0003.png', 'newpic0006.png', 'newpic0005.png', 'newpic0004.png']

sorted_files = sort_filenames(all_files)

for i in range(0, len(sorted_files) - 1, 2):
    # Just printing the statement to confirm we have it right
    print('merger({}, {})'.format(sorted_files[i], sorted_files[i + 1]))

注意 ： get_file_key()从文件名中提取数字，并使用它们对数组进行排序。 如果文件名包含的数字不属于排序范围，则此处可能会出现问题。