tensorrt_inference ScatterND generated after being exported to onnx while using yolov5

ScatterND generated after being exported to onnx while using yolov5

Open fan-chao opened this issue 3 years ago • 0 comments

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

Jan 11 '21 09:01 fan-chao