训练模型时经常需要对数据文件进行相关处理,这里实时进行总结
一、获取路径
函数 | 作用 |
---|
os.path.abspath(__ file__) | 返回当前文件的绝对路径,也可是指定路径的绝对路径 |
os.path.dirname(path) | 返回路径的上一级路径 |
os.path.join() | 可以进行路径拼接 |
os.listdir(path) | 得到该路径下的所有文件名(注意仅仅是当前目录下) |
二、筛选文件名
例子(从下面数列中选取 .jpg图片):
img_names=['alexnet-owt-4df8aa71.pth', 'Golden Retriever from baidu.jpg',
'imagenet1000.json', 'imagenet_classnames.txt', 'test',
'tiger cat.jpg', 'train']
img_names = list(filter(lambda x:x.endswith('.jpg'),img_names))
print(iimg_names)
结果:
>>['Golden Retriever from baidu.jpg', 'tiger cat.jpg']
三、判断文件/目录
函数 | 作用 |
---|
os.path.exists(path) | 判断路径是否存在 |
os.path.isdir(path) | 判断是否是文件夹 |
os.path.isfile(path) | 判断是否是文件 |
四、获取文件名
函数 | 作用 |
---|
os.path.basename(path) | 获取目录的最后文件名 |