isohrab / pupil-locator Goto Github PK

Hi Alireza Keshavarzi, last week i found your amazing work, and it work like a charm, i'm trying to porting it to ios application but so far i'm stucked, can you guide me some tip,

i'm trying to convert model to PB format then to tflite, i'm stuck at define output_node_name for conver to pb step (['InceptionV4/Logits/Predictions']), then i'm using all possible node name

min_graph = tf.compat.v1.graph_util.convert_variables_to_constants(session, session.graph_def, [node.name for node in graph.as_graph_def().node])
tf.train.write_graph(min_graph, "logdir/", "file.pb", as_text=False)
with open('output_graph.pb', 'wb') as f:
f.write(min_graph.SerializeToString())

i got success output pb file, but when load it i got these error

GRAPH_PB_PATH = 'logdir/file.pb'
with tf.Session() as sess:
print("load graph")
with gfile.FastGFile(GRAPH_PB_PATH,'rb') as f:
graph_def = tf.GraphDef()
graph_def.ParseFromString(f.read())
sess.graph.as_default()
tf.import_graph_def(graph_def, name='3A4Bh-Ref25')

=> ERROR: ValueError:ValueError: Input 0 of node 3A4Bh-Ref25/global_step/Assign was passed int32 from 3A4Bh-Ref25/global_step:0 incompatible with expected int32_ref.

so i'm stucked here

Please help me,
Thanks and Besst Regard

are the noisy_videos used for reflection effects in the augmentor.py publicly available?