问如何在tensorflow模型中将uint8的所有相关实例更改为int8
EN

Stack Overflow用户

提问于 2020-01-18 23:19:33

回答 1查看 209关注 0票数 0

我正在将冻结的pb模型转换为onnx，但onnx不支持uint8。如何在模型文件中用int8替换uint8？如果有必要，我也可以重新训练，但我不确定要修改哪个文件以及如何修改。我会猜到我需要修改包含神经网络架构的文件：https://github.com/tensorflow/models/blob/master/research/slim/nets/mobilenet/mobilenet_v2.py

但事实似乎并非如此。

发布于 2020-02-05 02:53:14

在ONNX model zoo中有一个已经被转换的MobileNet v2 ONNX Model (尽管源模型最初来自mxnet)。这可能会很有用。

ONNX运算符支持uint8数据类型，如ONNX IR documentation中所述。

票数 0

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

原文链接：

https://stackoverflow.com/questions/59801913

复制

相似问题

问如何在tensorflow模型中将uint8的所有相关实例更改为int8EN