oduerr / ki Goto Github PK

from scipy.io import loadmat

def load_emnist_data():
    # Source: Stack Overflow
    # Pfad zur entpackten Datei
    data_dir = "matlab/"
    mat_file_path = os.path.join(data_dir, 'emnist-balanced.mat')

    # Laden der .mat-Datei
    mat = loadmat(mat_file_path)

    # Extrahieren der Trainings- und Testdaten
    input_train = mat['dataset']['train'][0, 0]['images'][0, 0]
    target_train = mat['dataset']['train'][0, 0]['labels'][0, 0].flatten()
    input_test = mat['dataset']['test'][0, 0]['images'][0, 0]
    target_test = mat['dataset']['test'][0, 0]['labels'][0, 0].flatten()

    # Transformation der Daten
    input_train = input_train.reshape((input_train.shape[0], 28, 28), order="F")
    input_test = input_test.reshape((input_test.shape[0], 28, 28), order="F")

    return (input_train, target_train), (input_test, target_test)

(input_train, target_train), (input_test, target_test) = load_emnist_data()