- 学一点语法
- 训练或者推理的时候,可能要读取文件夹内的所有图片,下面给出一种解决方法
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)
- 分析
- osp.realpath 是获得当前的绝对路径
- osp.join 把路径连接起来
- 用迭代器遍历了所有文件