OS模块常用函数

1.BASE_DIR变量

BASE_DIR = os.path.dirname(os.path.abspath(__file__))

os.path.dirname()用于返回文件所在的路径，注意不包括文件名。

2.os.listdir()的用法

file_names = os.listdir(img_dir)  #获取数据集的图片名称组成的列表
img_names = list(filter(lambda x: x.endswith(format), file_names))

1. os.listdir()用于返回包含指定文件夹所包含的所有文件或者文件夹下的文件列表
2. filter(function, iterable)用于过符合function判别为true的序列元素，python3需要手动转为list()

来源：CSDN

作者：weixin_43213607

链接：https://blog.csdn.net/weixin_43213607/article/details/104502818