TrojanXu / onnxparser-trt-plugin-sample

A sample for onnxparser working with trt user defined plugins for TRT7.0
Apache License 2.0
166 stars 36 forks source link

Converting from ONNX to Tensorflow #15

Open Jove125 opened 3 years ago

Jove125 commented 3 years ago

Thanks for the plugin. I converted the model from Pytorch to ONNX format. How to convert it now to tensorflow format?

When I try to convert, I get an error: ` File "/home/user/anaconda3/envs/pytorch/lib/python3.7/site-packages/onnx_tf/backend.py", line 67, in prepare super(TensorflowBackend, cls).prepare(model, device, **kwargs) File "/home/user/anaconda3/envs/pytorch/lib/python3.7/site-packages/onnx/backend/base.py", line 76, in prepare onnx.checker.check_model(model) File "/home/user/anaconda3/envs/pytorch/lib/python3.7/site-packages/onnx/checker.py", line 106, in check_model C.check_model(protobuf_string) onnx.onnx_cpp2py_export.checker.ValidationError: No Op registered for TRT_PluginV2 with domain_version of 11

==> Context: Bad node spec for node. Name: GridSampler_1441 OpType: TRT_PluginV2`