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

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