如何修改Python脚本以支持任意命名及多格式图片路径导入

要实现支持多种图片格式+兼容任意文件名的需求，最简洁高效的方式是用Python的glob模块遍历目标文件夹里所有符合条件的图片文件，替代原来固定命名的循环逻辑。这里给你两种实用的实现方式：

方法1：直观遍历所有支持格式 #

import os
import glob

# 定义需要支持的所有图片格式（包含大小写变体）
SUPPORTED_EXTENSIONS = ('*.jpg', '*.jpeg', '*.png', '*.JPG', '*.JPEG', '*.PNG')

TEST_IMAGE_PATHS = []
test_images_dir = "test_images"

# 遍历每种格式，收集所有匹配的文件路径
for ext in SUPPORTED_EXTENSIONS:
    # 用os.path.join拼接路径，避免跨平台路径分隔符问题
    matched_files = glob.glob(os.path.join(test_images_dir, ext))
    TEST_IMAGE_PATHS.extend(matched_files)

# 可选：对路径进行排序（比如按文件名顺序）
TEST_IMAGE_PATHS = sorted(TEST_IMAGE_PATHS)

print("收集到的图片路径：", TEST_IMAGE_PATHS)

方法2：列表推导式简化写法 #

如果偏好更紧凑的代码，可以用列表推导式一步完成：

import os
import glob

SUPPORTED_EXTENSIONS = ('*.jpg', '*.jpeg', '*.png', '*.JPG', '*.JPEG', '*.PNG')
test_images_dir = "test_images"

TEST_IMAGE_PATHS = [
    file_path
    for ext in SUPPORTED_EXTENSIONS
    for file_path in glob.glob(os.path.join(test_images_dir, ext))
]

# 可选排序
TEST_IMAGE_PATHS = sorted(TEST_IMAGE_PATHS)

关键改动说明 #

1. 摆脱固定命名限制：
   原来的脚本通过循环生成image0.jpeg这类固定路径，现在用glob.glob()直接匹配文件夹中所有符合后缀的文件，不管文件名是什么，彻底解决了命名约束问题。

2. 兼容多格式与大小写：
   我们把所有需要支持的后缀（包括大小写变体，比如JPG和jpg）都放进元组，确保不同命名规范的图片都能被识别。这在Linux/macOS（区分文件名大小写）环境下尤其重要，同时也保证了跨平台兼容性。

3. 跨平台路径处理：
   用os.path.join()拼接文件夹和文件名，避免手动写路径分隔符（/或\）导致的跨平台运行错误。

额外优化建议 #

• 如果需要递归遍历子文件夹中的图片，可以给glob.glob()加上recursive=True参数，路径改成os.path.join(test_images_dir, '**', ext)。
• 如果担心重复路径（比如同一个文件有多个后缀变体），可以用list(set(TEST_IMAGE_PATHS))去重，但注意这样会打乱原有顺序。

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