使用TF再培训模型

时间:2017-12-31 09:53:55

标签: python go tensorflow

我正在尝试了解如何正确使用使用TensorFlow Retrain生成的重新训练模型 在Golang classifier

当我使用不同的模型(使用Golang分类器时极其不同且错误的结果)时,我得到的结果不同了

在迭代图形操作后,我对Go应用程序进行了一些修改,使其工作:

session.Run(
map[tf.Output]*tf.Tensor{
    graph.Operation("Mul").Output(0): tensor, // was "input"
},
[]tf.Output{
    graph.Operation("final_result").Output(0), // was "output"
},
nil)

const (
    H, W= 299, 299 // was 224,224
    Mean= float32(128) // was 117
    Scale= float32(1)
)

我正确加载图表吗? Mul操作首先是错误的吗? 也许是调整大小

1 个答案:

答案 0 :(得分:1)

我想我发现了这个问题。我修改了尺寸以对应新训练的模型。但是缩放比例保持为1而不是128。 现在结果相当不错。