获取所有文件的路径及文件对应标签(文件夹)

def load_data(path):
'''
每个文件夹下是一种图片
:param path:种类文件夹路径
:return: 图片路径列表和标签列表
'''
categories = os.listdir(path)
img_pathes = []
labels = []
for path, dirs, files in os.walk(path):
img_pathes.extend([os.path.join(path, file) for file in files])
print path
if len(files)>0:
labels.extend([path.split('/')[-1]] * len(files))
#print len(img_pathes),img_pathes
#print len(labels),labels
return img_pathes,labels