TF2.0 saved or frozen graph or onnx model中bilinear resize to nearest neighbor(NN)的修改方法
The way to change the bilinear resize to nearest neighbor(NN) in TF2.0 saved or frozen graph or onnx model
我有定制的 onnx 模型,想将其实现到特定的嵌入式设备中。
在实现之前,需要将模型转换为嵌入式设备兼容模型。
但问题是..
我的模型有双线性调整层(ResizeBilinear)但是转换工具只支持最近邻(ResizeNearestNeighbor)方法转换。
所以我的问题是
在 TF2.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
我有定制的 onnx 模型,想将其实现到特定的嵌入式设备中。
在实现之前,需要将模型转换为嵌入式设备兼容模型。
但问题是..
我的模型有双线性调整层(ResizeBilinear)但是转换工具只支持最近邻(ResizeNearestNeighbor)方法转换。
所以我的问题是
在 TF2.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