issue using converted vnect.pkl about vnect-tensorflow HOT 13 OPEN

@lavage
with tf.Session() as sess: saver = tf.train.Saver() model.load_weights(sess, model_file) save_path = saver.save(sess, "./vnect_tf")

from vnect-tensorflow.

check vnect_model.py in the models folder. There's a function called "load_weights" and you could use saver.save to get vnect_tf.

from vnect-tensorflow.

@yukaliu07 could you maybe provide some more detail on how to use saver.save? I have implemented it as follows, but the model files that this returns, give me wrong results:

with tf.Session() as sess:
    saver = tf.train.Saver()
    tf_writer = tf.summary.FileWriter(logdir='./', graph=sess.graph)

    sess.run(tf.global_variables_initializer())
    saver.save(sess, './vnect_tf')
    print(model.res5b_branch2c_new)
    print(model.heatmap, model.x_heatmap, model.y_heatmap, model.z_heatmap)

from vnect-tensorflow.

@yukaliu07 @lavage
Thank you so much!! It perfectly solved my problem. :D

from vnect-tensorflow.

@yukaliu07
I use the code but the pkl can't convert correctly.
Error message:
Traceback (most recent call last):
File "models/vnect_model.py", line 175, in
model = VNect(368)
File "models/vnect_model.py", line 13, in init
self._create_network()
File "models/vnect_model.py", line 94, in _create_network
self.res4d_branch2a = tc.lvnectayers.conv2d(self.res4c, kernel_size=1, num_outputs=256, scope='res4d_branch2a')
AttributeError: 'module' object has no attribute 'lvnectayers'

from vnect-tensorflow.

@yukaliu07 @powermew @lavage

Hello,
I had tried this code with these follows.

1.Download weights from the author of paper
2.change weights with using './caffe_weight_to_pickle.py'

3. run '/models/vnect_model.py' to get

vnect_tf.data-00000-of-00001
vnect_tf.index
vnect_tf.meta

run '/demo_tf.py'

But I am getting an error like this
2018-09-27 16:11:48.784541: W tensorflow/core/util/tensor_slice_reader.cc:95] Could not open models/weights/vnect_tf: Failed precondition: models/weights/vnect_tf; Is a directory: perhaps your file is in a different file format and you need to use a different restore operator?

I have tried editing vnect_model.py file as per

@lavage
with tf.Session() as sess: saver = tf.train.Saver() model.load_weights(sess, model_file) save_path = saver.save(sess, "./vnect_tf")

But I couldn't solve the problem, Can someone help me with this

from vnect-tensorflow.

@Robinchacko
At step 3 on your method, i guess you should fix the location of model file in demo_tf.py

from vnect-tensorflow.

I changed code line 173-181 in vnect_tf.py

if __name__ == '__main__':
    model_file = 'vnect.pkl'
    model = VNect(368)

    with tf.Session() as sess:
        saver = tf.train.Saver()
        model.load_weights(sess, model_file)
        save_path = saver.save(sess, "./vnect_tf")

And also changed line 21 in demo_tf.py

parser.add_argument('--model_file', default='models/weights/vnect_tf')

from vnect-tensorflow.

can any one provide model files please? [email protected] , author is not replying.

from vnect-tensorflow.

@musab1234,
Hi, did you get the model files? Could you share it?

from vnect-tensorflow.

Can you send your weight and model files?I have sent a email to the author since last week, but he haven't replied yet. my email is [email protected]

from vnect-tensorflow.

can you send the weight and model file? much appreciation!! [email protected]

from vnect-tensorflow.

Can you send the weight and the model file? Much appreciation!! [email protected]

from vnect-tensorflow.