用于动态输入形状的Tensorflow C ++ API

时间:2018-05-14 05:46:59

标签: python c++ tensorflow

我已经训练了一个模型,用于在python中对真实渲染的结果进行去噪。

在python中,通过将图像转换为tf.constant,可以结合图像大小动态构建图形。

现在我想导出模型并用C ++加载它。

我的问题是:是否可以在Tensorflow C ++ API中处理不同大小的图像?

如果没有,在C ++中执行此操作的最佳方法是什么?直接使用cuDNN并加载训练好的CNN内核权重?

1 个答案:

答案 0 :(得分:1)

所以答案是:是的,你可以处理不同大小的图像。

默认方法是使用未知维度tf.placeholder的{​​{1}}在Python中构建TensorFlow图,然后只需在C ++中导入图形。这已在此处讨论过: https://stackoverflow.com/a/48893889/7443104其中包括一个例子,C,C ++,Golang。

在TensorFlow中,大多数操作都不关心特定的输入大小,有关详细信息,请参阅https://stackoverflow.com/a/48558479/7443104