ScatterND generated after being exported to onnx while using yolov5

linghu8812 / tensorrt_inference

702 stars 206 forks source link

Open fan-chao opened 3 years ago

fan-chao commented 3 years ago

Hi,

When I used the script export_onnx.py to export .pt to onnx, some ScatterND layers are generated. However scatter_nd is not suppported for TensorRT.

I use the latest version yolov5-4.0 yolov5s.pt to exporting onnx. How do you handle this issue?

Thanks.

2021-01-11_171752