Python 獲取資料夾下的所有圖片
阿新 • • 發佈:2018-12-09
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