tensorRT
tensorRT copied to clipboard
YOLOv5 parse problem
The problem is that I can not parse the yolov5 model
the error is
While parsing node number 176 [Resize]: ERROR: ModelImporter.cpp:124 In function parseGraph: [5] Assertion failed: ctx->tensors().count(inputName) [07/28/2021-11:07:41] [E] Parsing File Failed [07/28/2021-11:07:41] [E] Init Session Failed! Segmentation fault
My environment
Tensorrt 7.0 cuda 10.2 opencv 3.4
I hope I can get some help, thanks !
hello, this question is caused by node number 176 [Resize]. you could check it in your onnx, and find the Corresponding op in your model. the op may be F.interpolate, but i am not sure.