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 ) ) os.listdir()用于返回包含指定文件夹所包含的所有文件或者文件夹下的文件 列表 filter(function, iterable)用于过符合function判别为true的序列元素,python3需要手动转为list() 来源: CSDN 作者: weixin_43213607 链接: https://blog.csdn.net/weixin_43213607/article/details/104502818