-
Notifications
You must be signed in to change notification settings - Fork 3
modify example with analytics-zoo #6
base: master
Are you sure you want to change the base?
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
_ = tf.import_graph_def(graph_def, name='') | ||
_ = tf.import_graph_def(graph_def, | ||
input_map={'DecodeJpeg:0': image_tensor[0]}, | ||
name='zoo') |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
to make the change minimal, just use name='' in the original code instead of name='zoo'
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks, I've modified the code.
"""Creates a graph from saved GraphDef file and returns a saver.""" | ||
# Creates graph from saved graph_def.pb. | ||
w, h, c = image_array.shape |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
is the order correct? shouldn't it be HWC?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks, it should be HWC. I've fixed it.
modified analytics-zoo example of tf image recognition