Advertisement
EXTREMEXPLOIT

Imagenes a matrices para entrenar la Red Neuronal, sin usar TensorFlow.

Sep 8th, 2020
1,217
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.77 KB | None | 0 0
  1. X, Y = [], []
  2. DATA_PATH = './DATA'
  3. for folderName in os.listdir(DATA_PATH):
  4.     print(f'Loading: {folderName}')
  5.     CurrentDir = f'{DATA_PATH}/{folderName}'
  6.     for imageName in os.listdir(CurrentDir):
  7.         imageData = np.asarray(Image.open(f'{CurrentDir}/{imageName}'))
  8.         imageLabel = Labels.index(folderName.replace('Upper', ''))
  9.         X.append(imageData)
  10.         Y.append(imageLabel)
  11.        
  12. print('\n')
  13. print('Shuffling Data...')
  14. DATA = list(zip(X, Y))
  15. np.random.shuffle(DATA)
  16. X, Y = zip(*DATA)
  17. print('Splitting Data...')
  18. X_Train, X_Test, Y_Train, Y_Test = train_test_split(X, Y, test_size=0.1)
  19. print('Creating NumPy Arrays...')
  20. X_Train, X_Test, Y_Train, Y_Test = np.array(X_Train), np.array(X_Test), np.array(Y_Train), np.array(Y_Test)
  21. print('Done!')
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement