データセットの前処理

Preprocessing.py

from tensorflow.keras.utils import to_categorical

def preprocessing(data, ch=3, label=False):
    if label:
        return to_categorical(data)
    data = data / 255.0
    w, h = data[0:3]
    data = data.reshape((-1, w, h, ch))
    return data