search

linghu8812 / tensorrt_inference

699 stars 205 forks source link

[YOLOV4 CSP] input and output names for onnx file #91

Open bobbilichandu opened 3 years ago

bobbilichandu commented 3 years ago

I am trying to server the model using triton, I need to fill in input and output names for the onnx file. I couldn't find those in export_onnx.py I can easily find them for ScaledYOLOv4. as 'images' and 'output' There are so many nodes in Yolov4. Can you please tell