Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- X, Y = [], []
- DATA_PATH = './DATA'
- for folderName in os.listdir(DATA_PATH):
- print(f'Loading: {folderName}')
- CurrentDir = f'{DATA_PATH}/{folderName}'
- for imageName in os.listdir(CurrentDir):
- imageData = np.asarray(Image.open(f'{CurrentDir}/{imageName}'))
- imageLabel = Labels.index(folderName.replace('Upper', ''))
- X.append(imageData)
- Y.append(imageLabel)
- print('\n')
- print('Shuffling Data...')
- DATA = list(zip(X, Y))
- np.random.shuffle(DATA)
- X, Y = zip(*DATA)
- print('Splitting Data...')
- X_Train, X_Test, Y_Train, Y_Test = train_test_split(X, Y, test_size=0.1)
- print('Creating NumPy Arrays...')
- X_Train, X_Test, Y_Train, Y_Test = np.array(X_Train), np.array(X_Test), np.array(Y_Train), np.array(Y_Test)
- print('Done!')
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement