Advertisement
Guest User

Untitled

a guest
Oct 22nd, 2019
102
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.40 KB | None | 0 0
  1. import numpy as np
  2. def load_image(img_file, target_size=(224,224)):
  3. X = np.zeros((1, *target_size, 3))
  4. X[0, ] = np.asarray(tf.keras.preprocessing.image.load_img(
  5. img_file,
  6. target_size=target_size)
  7. )
  8. X = tf.keras.applications.mobilenet.preprocess_input(X)
  9. return X
  10.  
  11. import os
  12. def ensure_folder_exists(folder):
  13. if not os.path.exists(folder):
  14. os.makedirs(folder)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement