Как конвертировать файлы изображений в CSV с меткой

Question

Тестовая папка имеет папки с именами от 0 до 9. Папки 0-9 содержат соответствующие рукописные изображения цифр.Я хочу преобразовать изображения в один файл test.csv, чтобы в первом столбце была метка цифры (т. Е. 0–9), а в остальных столбцах - значение пикселя, если изображение.

Я создалCSV, но первый столбец для метки отображается пустым.

from scipy.misc import imread
import numpy as np
import pandas as pd
import os
import imageio
import glob
root = './test' 

# go through each directory in the root folder given above
for directory, subdirectories, files in os.walk(root):
    # go through each file in that directory
    for file in files:  
        # read the image file and extract its pixels
        im = imread(os.path.join(directory,file))
        value = im.flatten()
        value = np.hstack((directory[8:],value))
        df = pd.DataFrame(value).T
        df = df.sample(frac=1) # shuffle the dataset
        with open('test.csv', 'a') as dataset: 
            df.to_csv(dataset, header=False, index=False)