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

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