enput-better.py

from PIL import Image
import numpy as np
import glob, os

g3 = 'g3'
g4c = 'g4c'
g4nc = 'g4nc'
np = 'np'

filename = 'Prostate_Cancer_Data'

with open(filename, "wb") as f:
    for img in glob.glob("*bmp"):
        if g3 in img:
            label = [0]
        elif g4c in img:
            label = [1]
        elif gn4c in img:
            label = [2]
        elif np in img:
            label = [3]
        else:
            # idk do something better here
            throw Exception('something went tits up')
        
        im = Image.open(img)
        im = np.array(im)

        red = im[:, :, 0].flatten()
        green = im[:, :, 1].flatten()
        blue = im[:, :, 2].flatten()

        im = np.array(list(label) + list(red) + list(green) + list(blue), dtype = np.uint8)

        f.write(im)
        print(im)