galeone/tfgo

Create mapping from model for model.Exec

Open

#32 创建于 2020年2月17日

在 GitHub 查看
 (4 评论) (0 反应) (0 负责人)Go (157 fork)github user discovery
help wanted

仓库指标

Star
 (2,492 star)
PR 合并指标
 (PR 指标待抓取)

描述

Hi,

Is there a way to automatically create the mapping for the Placeholder (e.g. Placeholder_1) to the input name (e.g. Height) from the model?

Also is there a way to automatically read the "prob_out" from the model?

[]tf.Output and map[tf.Output]*tf.Tensor are both used for model.Exec:

height, _ := tf.NewTensor([1]int64{"Height"})
width, _ := tf.NewTensor([1]int64{"Width"})

results := model.Exec([]tf.Output{
    model.Op("prob_out", 0),
}, map[tf.Output]*tf.Tensor{
   model.Op("Placeholder_1", 0): height,
   model.Op("Placeholder_2", 0): width,
})

Thanks.

贡献者指南