tensorflow - 将 Tensorflow 模型从对象检测 API 转换为 uff

标签 tensorflow tensorrt

似乎没有办法从盒子 Mobilenet(和其他来自 TF OD API 的模型)转换为 uff 格式,然后转换为 TensorRT 格式,because of much unsupported layers .
有什么方法可以删除\替换该层?例如使用 graph_transform 工具?我理解并非所有这些层的目的。
这是default model ,如果有人想尝试。

最佳答案

UFF 转换工具似乎在一些无法识别的层上卡住了,也许这会在 GA 版本中得到改进。现在,您需要移除这些层(并仅保留推理所需的最小子集),然后使用 nvinfer 插件 API 实现这些层。

关于tensorflow - 将 Tensorflow 模型从对象检测 API 转换为 uff,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/47153279/

相关文章:

python - Tensorflow混合两个多元分布

tensorflow - tf.keras.utils.to_categorical 在图形模式下引发 TypeError

python - 如何只使用一个 GPU 进行 tensorflow session ?

c++ - 升级到 TensorRT 7 后加载或构建 cuda 引擎偶尔会崩溃

python - 在 Tensorflow 2.1 中转换后无法加载 Tensor RT SavedModel

python - tf.reshape 与 (tf.expand_dims + tf.squeeze...等)

tensorflow - 深度学习回归——巨大的均方误差和损失

android - Android:出现错误:添加新依赖项后,任务 ':app:processDebugGoogleServices'的执行失败

python - 可转换为 TensorRT 的 Tensorflow tf.abs() 替代方案

C++ TensorRT 批量推理给出奇怪的结果