加载所有图片「Python」

• 学一点语法
• 训练或者推理的时候，可能要读取文件夹内的所有图片，下面给出一种解决方法

import os
import os.path as osp

# 假设所有图片放在 images 文件夹内
# 这里假设在图片所在文件夹的上一级目录运行此代码

# 加载images文件夹内的所有图片
imlist = [osp.join(osp.realpath('.'), images, img) for img in os.listdir(images) if
          os.path.splitext(img)[1] == '.png' or
          os.path.splitext(img)[1] == '.jpeg' or
          os.path.splitext(img)[1] == '.jpg']

# print("imlist: ", imlist)

• 分析

1. osp.realpath 是获得当前的绝对路径
2. osp.join 把路径连接起来
3. 用迭代器遍历了所有文件