代码之家  ›  专栏  ›  技术社区  ›  empty Debilski

“inputs”和“outputs”参数是什么tf.U模式.simple\u save()?

  •  1
  • empty Debilski  · 技术社区  · 7 年前

    所以我在看 tf.saved_model.simple_save

    simple_save(session, export_dir, inputs={"x": x, "y": y}, outputs={"z": z})
    

    如果我运行推理产品代码,如:

    session = Session()
    inputs = tf.placeholder(dtype=tf.float32, shape=(None, height, width, in_channel_size), name='input_img')
    model = Some_Model(inputs, num_classes=no_of_defects, is_training=False)
    logits, _ = model.build_model()
    predictor = tf.nn.softmax(self.logits, name='logits_to_softmax')
    feed_dict = {inputs: inputs}
    prediction_probabilities = session.run(self.predictor, feed_dict=feed_dict)
    

    inputs outputs 用于:

    simple_save(session, export_dir, inputs=???, outputs=???)
    

    即“输入”和“输出”的参数是什么 tf.saved_mode.simple_save()

    如果相关的话,我正在PyCharm中运行调试会话。

    2 回复  |  直到 7 年前
        1
  •  -1
  •   empty Debilski    7 年前
    simple_save(session, export_dir, inputs={"inputs" : inputs}, outputs={"predictor" : predictor})
    
        2
  •  -1
  •   rocksteady    7 年前

    在我看来,应该是:

     simple_save(session, export_dir, inputs=feed_dict, outputs={"predictor" : predictor}))
    
    推荐文章