如何获取OpenCV分割后图像列表的所有可能排列组合？

原代码存在的问题 #

• 未导入os模块，执行os.mkdir会触发报错
• 通过glob读取分割后的图像，无法保证读取顺序与分割时的1-4.jpg一致，导致排列的基础列表顺序混乱
• 未正确调用itertools.permutations并处理生成的排列结果

修正后的完整代码 #

# 导入所需库
import cv2 as cv
import numpy as np
import itertools
import os

# 加载图像
path = 'cat.jpg'
img = cv.imread(path)
if img is None:
    raise ValueError("无法加载图像，请检查路径是否正确")

# 获取图像尺寸并计算中心点
(h, w) = img.shape[:2] 
centerX, centerY = (w//2), (h//2) 

# 分割图像
topleft = img[0:centerY, 0:centerX]
topright = img[0:centerY, centerX:w]
bottomleft = img[centerY:h, 0:centerX]
bottomright = img[centerY:h, centerX:w]

# 直接将分割后的图像存入列表（顺序可控，无需保存再读取）
img_list = [topleft, topright, bottomleft, bottomright]

# 创建保存排列结果的目录（如果不存在）
output_dir = 'permutations'
if not os.path.exists(output_dir):
    os.makedirs(output_dir)

# 生成所有4个元素的全排列
all_permutations = itertools.permutations(img_list)

# 遍历所有排列，拼接成完整图像并保存
for idx, perm in enumerate(all_permutations):
    # 拼接排列后的图像：先拼上下两行，再拼左右两列
    top_row = np.hstack((perm[0], perm[1]))
    bottom_row = np.hstack((perm[2], perm[3]))
    combined_img = np.vstack((top_row, bottom_row))
    
    # 保存拼接后的图像
    cv.imwrite(f'{output_dir}/perm_{idx+1}.jpg', combined_img)

print(f"已生成{len(list(itertools.permutations(img_list)))}种排列，保存至{output_dir}目录")

关键说明 #

• 跳过冗余文件操作：直接将分割得到的图像数组存入列表，既保证顺序完全可控，又避免了文件读写的开销和顺序不确定性
• 全排列生成：itertools.permutations(img_list)会生成4个元素的所有全排列（共24种），每个排列是包含4个图像数组的元组
• 图像拼接逻辑：利用np.hstack拼接同一行的左右图像，np.vstack拼接上下两行，还原为与原图像尺寸一致的完整图像
• 鲁棒性优化：添加图像加载失败的判断，避免后续代码无意义执行；创建目录前先检查是否存在，避免重复创建报错

内容的提问来源于stack exchange，提问作者Umar Usman