Tensorflow:NotImplementedError:当前只能导入以tf.saved_model.save导出的SavedModels

时间:2019-04-27 06:39:00

标签: tensorflow tensorflow-serving tensorflow.js tensorflowjs-converter

我有一个张量流图,使用以下代码将其保存为“ saved_model.pb”:

builder = tf.saved_model.builder.SavedModelBuilder('saved_location') 
with tf.Session(graph=tf.Graph()) as sess, open('logfile', 'wb') as log_file:
...
...
builder.add_meta_graph_and_variables(sess, [tf.saved_model.tag_constants.SERVING])

我定期调用builder.save()。并查看被更新的saved_model.pb文件。

现在,当我尝试将其导入JavaScript时,reference

>  tensorflowjs_converter --input_format=tf_saved_model 
 --output_format=tfjs_graph_model saved_location/  target

我收到以下错误:

  

文件   “ /anaconda3/lib/python3.5/site-packages/tensorflowjs/converters/tf_saved_model_conversion_v2.py”,   第269行,在convert_tf_saved_model中       model = load(saved_model_dir)文件“ /anaconda3/lib/python3.5/site-packages/tensorflow/python/saved_model/load.py”,>第108行,处于加载状态

     

“当前仅使用tf.saved_model.save导出的SavedModels可能是” NotImplementedError:当前仅使用tf.saved_model.save导出的SavedModels    $daten = $row[0]; $no1 = $row[1]; $no2 = $row[2]; $no3 = $row[3]; $no4 = $row[4]; $no5 = $row[5]; //Here, You can insert data into database. $qry = "INSERT INTO `rqn_list`(`rqn_list_date`, `rqn_no`, `rqn_dsc`, `rqn_list_cc`, `rqn_list_wf`, `rqn_list_remarks`) VALUES ('$daten', '$no1', '$no2', '$no3', '$no4', '$no5')"; $res = mysqli_query($con,$qry); 可以被导入。其他SavedModels可能   最终将通过load()得到支持。

  1. 我要去哪里错了?

  2. 我如何将tensorflow图模型转换为可在浏览器中加载以使用javascript进行推理的格式?

注意:我能够将keras图层模型导入js并通过loadLayersModel api使用它,但是我正在寻找一种无需将整个代码库转换为keras的解决方案。

0 个答案:

没有答案