转换RIFE模型到onnx再转换到tensorfow失败

[kinier88]

一、转换代码如下： onnx_path = 'flownet.onnx'
dummy_input = torch.randn(1, 6, 256, 448).to(device) torch.onnx.export(self.flownet, dummy_input, onnx_path, export_params=True, opset_version=16, do_constant_folding=True, input_names = ['input'], output_names = ['output'], ) print(f'Model has been exported to {onnx_path}')

   onnx_model = onnx.load(onnx_path)
   tf_rep = prepare(onnx_model)
   tf_rep.export_graph("flownet.pb")
   ## 加载 SavedModel
   model = tf.saved_model.load("flownet.pb")

二、报错如下： BackendIsNotSupposedToImplementIt: LeakyRelu version 16 is not implemented.

如果opset_version设置11则提示grid_sample要opset_version 16

或者请问下能不能提供Tf或者tflite的预训练模型？

[hzwer]

I have no relevant skills ... 完全不懂