我已经定制了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