search

HiKapok / SSD.TensorFlow

State-of-the-art Single Shot MultiBox Detector in Pure TensorFlow, QQ Group: 758790869
Apache License 2.0
323 stars 153 forks source link

I can't get the weights of the model. #122

Closed WangFengtu1996 closed 5 years ago

WangFengtu1996 commented 5 years ago

Describtion

when i use the .NewCheckpointReader to get the weights and bias of the model,i just can get the bias of the model ,i can't get the weiths ,how can i resolve the porblem.

code

#从该模型中读取内容
reader = pywrap_tensorflow.NewCheckpointReader(checkpoint_path) 
var_to_shape_map = reader.get_variable_to_shape_map()

#存放文件
f = open('/home/ubuntu/GAP8/gap_sdk_old/tf2gap8/examples/portmodel/data/params.txt','w')
#打印tensor名及其值
for key in var_to_shape_map:
    print("tensor_name: ", key)
    w = reader.get_tensor(key) 
#    print (w)
#    print(type(key))  
#     print(key.shape) 

    print('\n' + key + '\n',file=f)
    f.write(str(w))
f.close()

rusule

tensor_name: ssd300/multibox_head/loc_4/kernel/Momentum tensor_name: ssd300/multibox_head/loc_4/kernel tensor_name: ssd300/multibox_head/loc_4/bias/Momentum tensor_name: ssd300/multibox_head/loc_4/bias tensor_name: ssd300/multibox_head/loc_3/kernel/Momentum tensor_name: ssd300/multibox_head/loc_3/kernel tensor_name: ssd300/multibox_head/loc_5/kernel tensor_name: ssd300/multibox_head/loc_3/bias/Momentum tensor_name: ssd300/multibox_head/loc_5/kernel/Momentum tensor_name: ssd300/multibox_head/loc_3/bias tensor_name: ssd300/multibox_head/loc_2/kernel/Momentum tensor_name: ssd300/multibox_head/loc_2/bias tensor_name: ssd300/multibox_head/loc_1/kernel tensor_name: ssd300/multibox_head/loc_1/bias tensor_name: ssd300/multibox_head/loc_0/kernel/Momentum tensor_name: ssd300/multibox_head/loc_1/bias/Momentum tensor_name: ssd300/multibox_head/loc_0/bias/Momentum tensor_name: ssd300/multibox_head/loc_0/bias tensor_name: ssd300/multibox_head/cls_5/kernel tensor_name: ssd300/multibox_head/cls_4/bias/Momentum tensor_name: ssd300/multibox_head/cls_4/bias tensor_name: ssd300/multibox_head/loc_1/kernel/Momentum tensor_name: ssd300/multibox_head/cls_3/kernel/Momentum tensor_name: ssd300/multibox_head/cls_3/kernel tensor_name: ssd300/multibox_head/cls_3/bias tensor_name: ssd300/multibox_head/cls_2/bias tensor_name: ssd300/multibox_head/cls_0/kernel/Momentum tensor_name: ssd300/multibox_head/cls_2/kernel tensor_name: ssd300/multibox_head/cls_0/kernel tensor_name: ssd300/multibox_head/cls_0/bias/Momentum tensor_name: ssd300/multibox_head/loc_2/kernel tensor_name: ssd300/multibox_head/cls_1/kernel tensor_name: ssd300/conv1/conv1_2/bias tensor_name: ssd300/conv3/conv3_3/kernel tensor_name: ssd300/conv1/conv1_2/kernel tensor_name: ssd300/conv1/conv1_1/kernel tensor_name: ssd300/conv1/conv1_2/bias/Momentum tensor_name: ssd300/conv1/conv1_1/bias tensor_name: ssd300/additional_layers/conv10/conv10_2/kernel tensor_name: ssd300/multibox_head/cls_3/bias/Momentum tensor_name: ssd300/additional_layers/conv10/conv10_2/kernel/Momentum tensor_name: ssd300/additional_layers/conv11/conv11_1/kernel tensor_name: ssd300/additional_layers/conv9/conv9_2/bias/Momentum tensor_name: ssd300/additional_layers/conv9/conv9_1/bias/Momentum tensor_name: ssd300/additional_layers/conv9/conv9_2/kernel tensor_name: ssd300/additional_layers/conv8/conv8_1/bias/Momentum tensor_name: ssd300/additional_layers/conv8/conv8_2/kernel tensor_name: ssd300/conv2/conv2_1/bias tensor_name: ssd300/additional_layers/conv8/conv8_2/kernel/Momentum tensor_name: ssd300/conv2/conv2_1/bias/Momentum tensor_name: ssd300/conv2/conv2_1/kernel tensor_name: ssd300/conv2/conv2_2/bias tensor_name: ssd300/additional_layers/conv9/conv9_1/kernel tensor_name: ssd300/additional_layers/conv11/conv11_2/kernel/Momentum tensor_name: ssd300/additional_layers/conv8/conv8_2/bias/Momentum tensor_name: global_step tensor_name: ssd300/multibox_head/cls_2/bias/Momentum tensor_name: ssd300/conv2/conv2_2/bias/Momentum tensor_name: ssd300/additional_layers/conv10/conv10_2/bias/Momentum tensor_name: ssd300/conv4/conv4_2/kernel/Momentum tensor_name: ssd300/conv3/conv3_3/bias/Momentum tensor_name: ssd300/additional_layers/conv11/conv11_1/kernel/Momentum tensor_name: ssd300/additional_layers/conv10/conv10_1/kernel/Momentum tensor_name: ssd300/additional_layers/conv10/conv10_1/bias tensor_name: ssd300/conv5/conv5_2/kernel tensor_name: ssd300/additional_layers/conv9/conv9_2/bias tensor_name: ssd300/conv1/conv1_2/kernel/Momentum tensor_name: ssd300/additional_layers/conv11/conv11_1/bias tensor_name: ssd300/conv2/conv2_2/kernel tensor_name: ssd300/additional_layers/conv11/conv11_2/bias/Momentum tensor_name: ssd300/multibox_head/loc_0/kernel tensor_name: ssd300/conv2/conv2_1/kernel/Momentum tensor_name: ssd300/multibox_head/cls_4/kernel/Momentum tensor_name: ssd300/multibox_head/cls_4/kernel tensor_name: ssd300/fc7/bias/Momentum tensor_name: ssd300/additional_layers/conv10/conv10_1/bias/Momentum tensor_name: ssd300/additional_layers/conv11/conv11_2/bias tensor_name: ssd300/additional_layers/conv8/conv8_1/kernel tensor_name: ssd300/conv5/conv5_3/bias tensor_name: ssd300/multibox_head/loc_2/bias/Momentum tensor_name: ssd300/additional_layers/conv8/conv8_2/bias tensor_name: ssd300/additional_layers/conv9/conv9_1/kernel/Momentum tensor_name: ssd300/conv3/conv3_2/bias/Momentum tensor_name: ssd300/additional_layers/conv10/conv10_1/kernel tensor_name: ssd300/multibox_head/loc_5/bias/Momentum tensor_name: ssd300/multibox_head/cls_5/kernel/Momentum tensor_name: ssd300/additional_layers/conv11/conv11_1/bias/Momentum tensor_name: ssd300/additional_layers/conv9/conv9_2/kernel/Momentum tensor_name: ssd300/conv5/conv5_2/bias/Momentum tensor_name: ssd300/conv4_3_scale/weights tensor_name: ssd300/conv5/conv5_3/kernel/Momentum tensor_name: ssd300/multibox_head/loc_5/bias tensor_name: ssd300/additional_layers/conv8/conv8_1/bias tensor_name: ssd300/multibox_head/cls_5/bias/Momentum tensor_name: ssd300/multibox_head/cls_1/bias/Momentum tensor_name: ssd300/conv5/conv5_1/kernel tensor_name: ssd300/conv2/conv2_2/kernel/Momentum tensor_name: ssd300/conv3/conv3_1/kernel/Momentum tensor_name: ssd300/additional_layers/conv8/conv8_1/kernel/Momentum tensor_name: ssd300/conv3/conv3_1/bias tensor_name: ssd300/conv3/conv3_1/bias/Momentum tensor_name: ssd300/conv4/conv4_1/bias tensor_name: ssd300/multibox_head/cls_1/bias tensor_name: ssd300/conv3/conv3_1/kernel tensor_name: ssd300/conv3/conv3_2/kernel tensor_name: ssd300/fc7/bias tensor_name: ssd300/conv3/conv3_2/bias tensor_name: ssd300/conv3/conv3_2/kernel/Momentum tensor_name: ssd300/conv4/conv4_1/bias/Momentum tensor_name: ssd300/conv3/conv3_3/bias tensor_name: ssd300/additional_layers/conv10/conv10_2/bias tensor_name: ssd300/conv5/conv5_2/kernel/Momentum tensor_name: ssd300/multibox_head/cls_1/kernel/Momentum tensor_name: ssd300/conv3/conv3_3/kernel/Momentum tensor_name: ssd300/conv5/conv5_3/bias/Momentum tensor_name: ssd300/multibox_head/cls_2/kernel/Momentum tensor_name: ssd300/conv1/conv1_1/bias/Momentum tensor_name: ssd300/conv4/conv4_1/kernel tensor_name: ssd300/additional_layers/conv11/conv11_2/kernel tensor_name: ssd300/conv4/conv4_1/kernel/Momentum tensor_name: ssd300/conv4/conv4_2/bias tensor_name: ssd300/conv4/conv4_2/bias/Momentum tensor_name: ssd300/conv4/conv4_2/kernel tensor_name: ssd300/conv4/conv4_3/bias tensor_name: ssd300/fc7/kernel/Momentum tensor_name: ssd300/conv4/conv4_3/bias/Momentum tensor_name: ssd300/conv4/conv4_3/kernel tensor_name: ssd300/conv5/conv5_1/bias tensor_name: ssd300/fc6/kernel tensor_name: ssd300/conv4/conv4_3/kernel/Momentum tensor_name: ssd300/conv5/conv5_1/kernel/Momentum tensor_name: ssd300/additional_layers/conv9/conv9_1/bias tensor_name: ssd300/conv5/conv5_2/bias tensor_name: ssd300/conv1/conv1_1/kernel/Momentum tensor_name: ssd300/conv5/conv5_1/bias/Momentum tensor_name: ssd300/conv5/conv5_3/kernel tensor_name: ssd300/fc6/bias tensor_name: ssd300/fc6/bias/Momentum tensor_name: ssd300/multibox_head/cls_0/bias tensor_name: ssd300/multibox_head/cls_5/bias tensor_name: ssd300/fc6/kernel/Momentum tensor_name: ssd300/conv4_3_scale/weights/Momentum tensor_name: ssd300/fc7/kernel

WangFengtu1996 commented 5 years ago

The variables ending in kernel are the weights.