Python 获取文件夹下的所有图片

最新推荐文章于 2024-07-12 11:37:11 发布

原创最新推荐文章于 2024-07-12 11:37:11 发布 · 2.1w 阅读

54 ·

CC 4.0 BY-SA版权

文章标签：

#python #image

python 专栏收录该内容

15 篇文章

订阅专栏

本文介绍了一种使用Python从指定文件夹中批量获取图片文件路径的方法，通过定义函数判断文件类型并利用os模块遍历目录，适用于图像处理、深度学习等场景的数据预处理阶段。

部署运行你感兴趣的模型镜像

Python 获取文件夹下的所有图片

判断文件是否是图片:

def is_image_file(filename):
    return any(filename.endswith(extension) for extension in ['.png', '.jpg', '.jpeg', '.PNG', '.JPG', '.JPEG'])

从文件夹中获取图片并保存图片路径:

image_filenames = [join(dataset_dir, x) for x in listdir(dataset_dir) if is_image_file(x)]

eg:

dataset_dir='data/VOC2012/train1'
image_filenames = [join(dataset_dir, x) for x in listdir(dataset_dir) if is_image_file(x)]
for image_filename in image_filenames:
    print(image_filename)

data/VOC2012/train1/VOC2012_00009.jpg
data/VOC2012/train1/VOC2012_00002.jpg
data/VOC2012/train1/VOC2012_00007.jpg
data/VOC2012/train1/VOC2012_00006.jpg
data/VOC2012/train1/VOC2012_00008.jpg
data/VOC2012/train1/VOC2012_00005.jpg
data/VOC2012/train1/VOC2012_00004.jpg
data/VOC2012/train1/VOC2012_00001.jpg
data/VOC2012/train1/VOC2012_00003.jpg

更新:

import os
trainpath='/home/*/*/SR/raisr/test'
def get_img_file(file_name):
    imagelist = []
    for parent, dirnames, filenames in os.walk(file_name):
        for filename in filenames:
            if filename.lower().endswith(('.bmp', '.dib', '.png', '.jpg', '.jpeg', '.pbm', '.pgm', '.ppm', '.tif', '.tiff')):
                imagelist.append(os.path.join(parent, filename))
        return imagelist