os.path模块中常用方法os.path.dirname与os.path.basename的结合

最近在做卷积神经网络,对于数据集,我们经常获取到的是下图中的格式。文件夹名是标签(类名),文件夹内是对应标签的图片
os.path模块中常用方法os.path.dirname与os.path.basename的结合
os.path模块中常用方法os.path.dirname与os.path.basename的结合我们最好对数据集进行整理,把每张图与标签存成txt文件或者tfrecord,方便后面网络训练。
如果我们知道一张图片的绝对路径path,那么我们就可以用
label = os.path.basename(os.path.dirname(path))
获得这个图片的label。

os.path.dirname(path)的意思是返回path的目录,即获得path的上一层路径。
os.path.basename(path)是返回路径最后的文件名(或目录)。

举个例子:假设path = E:\python_project\opencv\gesture_class\1\1.jpg

运行os.path.dirname(path)后会返回路径path_1 = E:\python_project\opencv\gesture_class\1

再运行os.path.basename(path_1),则返回的是label = 1

这样一来,我们就轻松获取了图片的类别。

os.path模块下还有很多常用方法,下面有很详细的介绍
python os.path模块常用方法详解