Yolo-Fastest icon indicating copy to clipboard operation
Yolo-Fastest copied to clipboard

onnx convert error

Open AIgiraffe opened this issue 4 years ago • 4 comments

the code of https://github.com/CaoWGG/TensorRT-YOLOv4 can`t converts the yolo-fastest to onnx model. typeerror: buffer is too small for requested array

AIgiraffe avatar Sep 21 '20 10:09 AIgiraffe

@jasonlove521 Have you resolved this issue? Are you trying to run Nvidia Jetson devices? Xavier NX etc?

MuhammadAsadJaved avatar Oct 10 '20 07:10 MuhammadAsadJaved

@MuhammadAsadJaved I train the model by using yolov5-torch, then convert to tensort engine. There is some problem with trt-engine. The mAP and inference speed of trt-engine`s are verlt low.

AIgiraffe avatar Oct 12 '20 07:10 AIgiraffe

Have you tried to convert the Yolo-Fastest model which the provided in the repo?

MuhammadAsadJaved avatar Oct 12 '20 13:10 MuhammadAsadJaved

@MuhammadAsadJaved yes. i have done, but that repo can`t convert the Yolo-Fastest to trt-engine,because Depthwise Convolution is used in the Yolo-Fastest, which occur error. i try to modify the code of the provided repo so that it can work, but the performance of trt_engine is not expected with low AP and slow speed.

AIgiraffe avatar Nov 22 '20 09:11 AIgiraffe