Skip to content

Instantly share code, notes, and snippets.

@FavioVazquez

FavioVazquez/dl_sql_2.py

Created May 2, 2018 16:28
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save FavioVazquez/a02094a5848ab1f7e42ce52820a09fbe to your computer and use it in GitHub Desktop.
Save FavioVazquez/a02094a5848ab1f7e42ce52820a09fbe to your computer and use it in GitHub Desktop.
Download ZIP
Raw
dl_sql_2.py
from keras.applications import InceptionV3
from sparkdl.udf.keras_image_model import registerKerasImageUDF
def keras_load_img(fpath):
from keras.preprocessing.image import load_img, img_to_array
import numpy as np
img = load_img(fpath, target_size=(299, 299))
return img_to_array(img).astype(np.uint8)
registerKerasImageUDF("inceptionV3_udf_with_preprocessing", InceptionV3(weights="imagenet"), keras_load_img)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment