在TF2.0保存或冻结图或onnx模型中将双线性resize更改为最近邻（NN）的方法

Question

我有定制的 onnx 模型，想将其实现到特定的嵌入式设备中。

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

但问题是..

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

所以我的问题是

TF2.X 的 Saved 模型、TF1.X 的冻结图、onnx 模型中的 Bilinear resize layer 有没有办法改成 Nearest Neighbor resize？

其实我有原来的onnx模型，转换成tf2的saved_model和tf1的frozen_graph，因为转换工具只支持tf1的frozen_grpah。

谢谢。

Answer 1

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

pip install onnxconverter-common

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

然后编辑 model.py 并制作新模型

python model.py new_model.onnx