Question

我正在尝试将一个保存的自定义对象检测模型加载到unity中。我正在tensorflow 1.12中训练模型，并在1.4中冻结图，因为只有1.4与单位兼容。

这是我的代码。

`

pixels = new byte[INPUT_SIZE  INPUT_SIZE  3];
 _catalog = CatalogUtil.ReadCatalogItems(labelMap.text);
       Debug.Log("LabelMap Text : " + labelMap.text);
       Debug.Log("Loading graph...");
       graph = new TFGraph();
       graph.Import(model.bytes);
       session = new TFSession(graph);
       Debug.Log("Graph Loaded!!!");`

我的统一版本为2017，张量流插件为0.4

这是我的错误跟踪。

`TFException: NodeDef mentions attr 'Truncate' not in Op<name=Cast; signature=x:SrcT -> y:DstT; attr=SrcT:type; attr=DstT:type>; NodeDef: Postprocessor/ToFloat_2 = Cast[DstT=DT_FLOAT, SrcT=DT_INT32, Truncate=false](Postprocessor/ToFloat_2/x). (Check whether your GraphDef-interpreting binary is up to date with your GraphDef-generating binary.).

TensorFlow.TFStatus.CheckMaybeRaise (TensorFlow.TFStatus incomingStatus, System.Boolean last) (at <1fe2de69842a4a4ba15256b83cca05f3>:0)`

这是因为我们正在一个版本中进行培训，而在另一个版本中进行冻结？。

如何解决此问题？

如何将保存为.bytes的模型统一存储在unites environment中，该模型在tnesorflow1.4中已冻结？

0 个答案: