tensorRT icon indicating copy to clipboard operation
tensorRT copied to clipboard

YOLOv5 parse problem

Open spades10 opened this issue 4 years ago • 1 comments

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 !

spades10 avatar Jul 28 '21 03:07 spades10

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.

HFBlalalala avatar Sep 07 '21 03:09 HFBlalalala