首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >将Tensorflow模型从对象检测API转换为uff

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

Stack Overflow用户
提问于 2017-11-07 08:35:50
回答 2查看 1.7K关注 0票数 5

似乎没有办法从box Mobilenet (和其他模型从to )转换为uff格式,然后再转换为TensorRT格式,因为很多不支持的层。有什么方法可以移除\替换那些层吗?例如,使用graph_transform工具?我明白并非所有这些层次的目的。这是缺省模型,如果有人想尝试的话。

EN

回答 2

Stack Overflow用户

回答已采纳

发布于 2019-07-12 10:39:54

从2017年开始,仍未取得重大进展。许多新的网络已经开发出来,但UFF和TensorRT转换器仍然不能与许多模型工作,甚至从2017年起,我可以说2019年。

也有一些间接信息在互联网上,在意义上,与UFF和TRT转换在NVIDIA非常小的部门,只有少数几个人。看来这条路是死胡同。

更好的方法和他们所依赖的方式是框架内部的转换工具。就像Tensorflow处理它们的TRT绑定

使用这种方法,新的层是由框架开发人员添加的,而不是Nvidia,因为那里的人跟不上市场营销。所以最终有更多的机会得到工作工具。

票数 0
EN

Stack Overflow用户

发布于 2017-11-30 00:27:44

UFF转换工具似乎被一些未被识别的层阻塞了,也许这将在GA版本中得到改进。现在,您需要删除这些层(并且只保留推理所需的最小子集),然后实现使用nvinfer插件API的层。

票数 2
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/47153279

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档