gpt4 book ai didi

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

转载 作者:行者123 更新时间:2023-12-04 21:05:34 26 4
gpt4 key购买 nike

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

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com