daigo0927/pwcnet

Please clarify Input and Output node names for trained model - model_250epochs_ft_Final/model_250.ckpt.meta

rkartheek2022 opened this issue · 0 comments

Hi All,
What would be the input node name and output node name for the below model?
model_250epochs_ft_Final/model_250.ckpt.meta

From the code https://github.com/daigo0927/pwcnet/blob/master/train.py
Looks like the input node name is 'images' and output node name is 'flows'. How ever - SNPE tools errors on input node.

Coudl some one please confirm the input node name for model_250epochs_ft_Final/model_250.ckpt.meta model?
with tf.name_scope('Data'):
self.images = tf.placeholder(tf.float32, shape = (self.args.batch_size, 2, *self.image_size, 3),
name = 'images')
images_0, images_1 = tf.unstack(self.images, axis = 1)
self.flows_gt = tf.placeholder(tf.float32, shape = (self.args.batch_size, *self.image_size, 2),
name = 'flows')

Thanks,
Rathi