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

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