kinier88

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

2

一、转换代码如下： 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...