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
上述函数get_img_file 返回一个包含文件路径的list列表