在 Python 中的多个文件夹中组合多个图像

Question

我有一个灰度图像文件夹（100x100 像素），文件名如下：

• file_123_A.ome.tif
• file_123_B.ome.tif
• file_123_C.ome.tif
• file_125_A.ome.tif
• file_125_B.ome.tif
• file_125_C.ome.tif

每个文件名的编号（在我上面的示例中为 123 和 125）链接图像，因此这些图像集是相关的。

我正在尝试将文件读入一个 numpy 数组，然后将三个相关图像组合成一个 100x100x3 或 100x300 的新图像。

我可以很高兴地将 A 集中的图像读取到一个数组中，然后将 B 和 C 集中的图像读取到它们自己的单独数组中。但是，一旦我这样做了，我就不能再检查文件名并且图像的顺序相同，所以如果 A 数组中的图像 0 是 B 和 C 数组中正确的相关图像。我可以做到这一点，并让它成功运行，但我不确定这是最好的解决方案。

由于每个图像的编号都是唯一的，我想编写一个解决方案，读取文件名并检查三个图像是否具有相同的唯一编号并且是一个集合，例如然后 A、B、C 图像相应地组合它们。如果有人能指出我正确的方向，那我正在努力解决这个问题。

不确定我对此的解释有多好，所以如果您需要澄清，请询问。

谢谢

更新：

这段代码（见下文）为我提供了目录和子目录中的一组唯一文件名。它可能不是最有效的，但它确实有效。我现在正在努力的是现在使用这个循环对实际图像文件进行比较并合并或加入具有每个唯一 ID 的三个图像。

#Set directory to look in
file = glob.glob(r'my_directory\**\*.tif',recursive = True)

#Loop though the files and put all the uniques files name into a list
fname_lst = []
for img in file:
    i = img.split('\\')[5][:-12]
    fname_lst.append(i)
fname_lst = set(fname_lst)

显然 .split 对我的文件名来说是唯一的，但它确实有效。

Answer 1

这应该对当前目录中的所有文件组执行您想要的操作。我会让你添加错误处理和其他小变化：

#!/usr/bin/env python3

import cv2
from glob import glob

# Get list of files matching "*A.ome.tif"
files = glob("*A.ome.tif")

# Get list of unique stems
stems = [f.replace('A.ome.tif', '') for f in files]

# Iterate over unique stems
for stem in stems:
    A = cv2.imread(stem + 'A.ome.tif')
    B = cv2.imread(stem + 'B.ome.tif')
    C = cv2.imread(stem + 'C.ome.tif')
    # Stack images horizontally (side-by-side), use np.dstack() for merging channels
    result = np.hstack((A,B,C))
    cv2.imwrite(stem + 'result.tif', result)