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

混沌水槽

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

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

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

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

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

帕特维

所以答案是:是的,您可以处理不同尺寸的图像。

默认方法是使用tf.placeholder未知维度在 Python 中构建 TensorFlow 图[B, None, None, 3],然后简单地在 C++ 中导入图。这之前已在此处讨论过:https : //stackoverflow.com/a/48893889/7443104,其中包括 C、C++、Golang 的示例。

在 TensorFlow 中,大多数操作并不关心具体的输入大小,请参阅https://stackoverflow.com/a/48558479/7443104了解更多详细信息。

本文收集自互联网,转载请注明来源。

如有侵权,请联系 [email protected] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章

Tensorflow C API占位符/输入变量设置

用于将tfrecords字节输入到tensorflow模型的Java API

Tensorflow 动态 RNN - 形状

指定输入/输出节点以在TensorFlow 1.0+中在加载了C ++ API的模型上运行推理

TensorFlow Lite C ++ API示例进行推理

如何使用C API遍历Tensorflow图?

Google Cloud TPU对Tensorflow C ++ API的支持

TensorFlow C API 日志记录设置

如何在 TensorFlow 中获得 None(动态输入形状)的真实形状?

了解Tensorflow LSTM输入形状

Tensorflow输入形状不兼容

用于输入的C#WPF动态名称

Tensorflow C ++:如何获得输出的张量形状?

Tensorflow Detection API中用于SSD检测器的灰度输入图像

如何正确使用形状未知的TensorFlow数据集API?

如何设置Tensorflow Lite C ++的输入

如何在Tensorflow的C ++ API中传输CUDA常量内存

Tensorflow对象检测API是否有C ++包装器?

如何构建和使用Google TensorFlow C ++ API

使用Tensorflow C ++ API执行在skflow中训练的模型

在Windows上可以使用TensorFlow C ++ API吗?

在TensorFlow C ++ API中,如何从张量保存图片或数组

tensorflow c ++ API是否支持自动区分以进行反向传播?

Tensorflow冻结图Protobuf无法使用C API进行预测

Tensorflow c++ api 对`tflite::DefaultErrorReporter()'的未定义引用

Tensorflow C++ API 分配变量值

Tensorflow中的A3C-我应该使用线程还是分布式Tensorflow API

TensorFlow服务:获取输入形状和标签

Tensorflow字符级CNN-输入形状