Guest User

Untitled

a guest
Nov 13th, 2018
100
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.48 KB | None | 0 0
  1. def path_to_tensor(path):
  2.  
  3. img = image.load_img(path, target_size = (224,224))
  4. x= image.img_to_array(img)
  5. return np.expand_dims(x, axis=0)
  6.  
  7. def paths_to_tensor(paths):
  8. list_of_tensors = [path_to_tensor(path) for path in tqdm(paths)]
  9. return np.vstack(list_of_tensors)
  10.  
  11. train_tensors = paths_to_tensor(train_files).astype('float32')/255
  12. valid_tensors = paths_to_tensor(valid_files).astype('float32')/255
  13. test_tensors = paths_to_tensor(test_files).astype('float32')/255
Add Comment
Please, Sign In to add comment