使用转换工具转换自己的模型出错:'CaseGenerator' object has no attribute 'nbg_graph_file_path'

我使用模型转换工具转化自己的模型的时候出现错误 如图


我的模型里用到了5维度的数据,是用TensorFlow1.12构建的模型,请问这个改怎么解决。 @Frank
和demo生成的文件做对比,多了一个的文件:graph.json。demo里是没有这个文件。

@lwx080227531x 完整的log贴出来一下,你这个截图这些信息,我看不到什么内容

我没有搜索到你们的日志文件,我吧截取了控制台更多的信息,

@lwx080227531x log里面提示了inputs层的节点0,size[2]里面包含了一个Invaild broadcast

看了你的回复仍然不知道哪里有问题,这个pb转换tensorRT是没有问题的,Invaild broadcast问题是我前两步的转换出了问题吗?

@lwx080227531x 我认为是你的层里包括了不支持的节点

换了个模型还是有上述问题,如果我用到了5维数据,是不是一定不行?另外,VIM3支持GPU吗? @Frank

@lwx080227531x VIM3有 开源的GPU的驱动,但是不能作为训练来用,因为没有cuda

转换代码时 生成的文件graph.json如下


是不是有问题? @Frank

@lwx080227531x 我也不也是很清楚这个,我也是认为超过4维度的这个工具也许转换不了,从特性上来说,这个工具处理的常见情况都是4维度的.毕竟这个工具的输出来说,基本都是图像处理类的