TF2.0保存或冻结图或onnx模型中双线性调整到最近邻(NN)的方法



我已经定制了onnx模型,想将其实现到特定的嵌入式设备中。

在实现之前,需要将模型转换为嵌入式设备兼容模型。

但问题是……

我的模型有双线性调整层(ResizeBilinear),但转换工具只支持最近邻(ResizeNearestNeighbor)方法转换。

我的问题是

是否有办法将TF2保存模型中的Bilinear调整大小图层更改为最近邻调整大小?X,或TF1的冻结图。X还是onnx型号?

其实我有原始的onnx模型,并将其转换为tf2的saved_model和tf1的frozen_graph,因为转换工具只支持tf1的frozen_grpah。

谢谢。

可以,您可以直接编辑onnx模型。尝试使用onnx2py脚本。

pip install onnxconverter-common

python -m onnxconverter_common.onnx2py model.onnx model.py

然后编辑model.py,创建新模型

python model.py new_model.onnx

相关内容

  • 没有找到相关文章

最新更新