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

Question

我已经训练了一个模型，可以对python中真实渲染的结果进行降噪。

在python中，通过将图像转换为tf.constant，图形是与图像大小一起动态构建的。

现在我想导出模型并将其加载到C++中。

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

如果没有，在C++中做到这一点的最佳方法是什么？直接使用 cuDNN 并加载经过训练的 CNN 内核权重？

Answer 1

所以答案是：是的，您可以处理不同大小的图像。

默认方法是使用维度未知的tf.placeholder在 Python 中构建 TensorFlow 图[B, None, None, 3]然后简单地在 C++ 中导入图。这在前面已经讨论过： https://stackoverflow.com/a/48893889/7443104 其中包括一个例子 für C， C++， Golang。

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