Key resnet_v1_50 not found in checkpoint about insightface_tf HOT 11 OPEN

Solved, if you meet the same problem, please re-install your tensor layer to version 1.7

pip uninstall tensorlayer
pip install tensorlayer==1.7

from insightface_tf.

Above problem take place when I try to use model d ckpt file as model

if I use the model trained by myself, it can work normally

# from tensorflow.python import pywrap_tensorflow
    #
    # # checkpoint_path = '/home/psy/InsightFace_TF-master/output/ckpt/InsightFace_iter_best_2000.ckpt'
    # checkpoint_path = '/home/psy/ckpt_model_d/InsightFace_iter_best_710000.ckpt'
    # reader = pywrap_tensorflow.NewCheckpointReader(checkpoint_path)
    # var_to_shape_map = reader.get_variable_to_shape_map()
    # for key in var_to_shape_map:
    #     print("tensor_name: ", key)
    #

if I use this block of code to find out the content of two ckpt files

Uploading 自己训练的_self_trained.txt…

model d does not contain 'resnet_v1_50/block1/unit_1/bottleneck_v1/PReluLayer/alpha' key
model d has less than 400 tensors

Uploading model_d.txt…

the model trained by myself does contain 'resnet_v1_50/block1/unit_1/bottleneck_v1/PReluLayer/alpha' key
the model has over than 600 tensors

Hoping to know how do you produce the model d file, and how can you put these ckpt file into use
thanks for your time.

from insightface_tf.

@auroua

from insightface_tf.

@pEko97
you can see L_Resnet_E_IR_fix_issue9

from insightface_tf.

Issue 9 is good to use!

from insightface_tf.

@pEko97 Yes, I think it is a version confict.

from insightface_tf.

Key resnet_v1_50/block1/unit_1/bottleneck_v1/shortcut_bn/BatchNorm/beta not found in checkpoint

tensorflow == 1.6.0
tensorlayer== 1.7.0

the issue still exists

from insightface_tf.

has anybody solved the problem？

from insightface_tf.

yes，issue9 and model D can work。@auroua thank you very much！

from insightface_tf.

@zjz5250 Can you tell me the version of tensorflow and tensorlayer that you successfully solved this problem?

from insightface_tf.

Solved, if you meet the same problem, please re-install your tensor layer to version 1.7

pip uninstall tensorlayer
pip install tensorlayer==1.7

It works, thank you.
tensorflow(cpu) 1.12.0
tensorlayer 1.7.0

from insightface_tf.