spark-deep-learning
spark-deep-learning copied to clipboard
stripAndFreezeGraph not found
Trying to do hands-on exercises given in the main github page. While doing Tensorflow transformer example (with TFImageTransformer),
from sparkdl import readImages, TFImageTransformer from sparkdl.transformers import utils import tensorflow as tf
g = tf.Graph() with g.as_default(): image_arr = utils.imageInputPlaceholder() resized_images = tf.image.resize_images(image_arr, (299, 299)) # the following step is not necessary for this graph, but can be for graphs with variables, etc frozen_graph = utils.stripAndFreezeGraph(g.as_graph_def(add_shapes=True), tf.Session(graph=g), [resized_images])
Got the following error:
Traceback (most recent call last):
File "
Checked transformers/utils.py code and found that there is no function stripAndFreezeGraph.
pyspark invoked with the locally assembled jar file : spark-deep-learning-assembly-0.1.0-spark2.1.jar