将学习到的参数转换为PB文件

时间:2018-12-20 07:24:35

标签: tensorflow deep-learning caffe

我有一个看起来像这样的numpy文件

array({'bbox_reg_1': {'weights': array([[[[ 1.41573034e-03, -1.68433134e-03, -1.46378449e-03],
         [ 4.94853244e-04,  2.70358869e-04,  1.17152464e-03],
         [ 1.46097073e-03,  1.64969335e-03,  2.44257652e-04],
         [-1.96780078e-03, -1.16519595e-03,  1.38781685e-03],   
         [ 9.05778143e-05,  1.55463757e-03, -1.17503619e-03],
         [ 6.90302986e-04,  1.20219309e-03, -6.66214619e-04],
         ...............
         ...............
         ...............  
         'biases': array([0.15425067, 0.22107883, 0.15874247,0.18794474, 0.3142768 ,
         0.22744565, 0.4702694 , 0.37981156, 0.2936976 , 0.26642567,
         0.27769807, 0.19335541, 0.56588143, 0.19396164, 0.22467329], dtype=float32)}}, dtype=object)

上下文是我正在尝试将caffe模型转换为tensorflow。哪一个要引入caffe模型并重新生成此文件?我感到困惑的是如何将其转换为pb文件?

谢谢。

0 个答案:

没有答案