似乎没有办法从盒子 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/