1. 程式人生 > >Python 獲取資料夾下的所有圖片

Python 獲取資料夾下的所有圖片

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