Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import numpy as np
- def load_image(img_file, target_size=(224,224)):
- X = np.zeros((1, *target_size, 3))
- X[0, ] = np.asarray(tf.keras.preprocessing.image.load_img(
- img_file,
- target_size=target_size)
- )
- X = tf.keras.applications.mobilenet.preprocess_input(X)
- return X
- import os
- def ensure_folder_exists(folder):
- if not os.path.exists(folder):
- os.makedirs(folder)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement