search

linghu8812 / tensorrt_inference

706 stars 208 forks source link

Supportted YOLOv5 4.0 #42

Open pango99 opened 3 years ago

pango99 commented 3 years ago

Does support YOLOv5 4.0?

linghu8812 commented 3 years ago

yes, it supports YOLOv5 4.0 now and support yolov5s6.pt and so on in YOLOv5 4.0.

just run with

./yolov5_trt ../config6.yaml ../samples
pango99 commented 3 years ago

Great,thank you very much for your sharing

bobbilichandu commented 3 years ago

Where can I find yolov5s6.yaml?

linghu8812 commented 3 years ago

@chandu1263 https://github.com/ultralytics/yolov5/blob/master/models/hub/yolov5s6.yaml

bobbilichandu commented 3 years ago

I am not able to convert yolov5s6 to onnx. Facing this issue ONNX export failure: Exporting the operator silu to ONNX opset version 10 is not supported. Please open a bug to request ONNX export support for the missing operator. I tried opset versions 10-12, but couldn't solve it. Any help would be great!

linghu8812 commented 3 years ago

@chandu1263 check these lines https://github.com/linghu8812/tensorrt_inference/blob/5f22509ead8c5b3432b682dde956d1cbcae0137c/yolov5/export_onnx.py#L39-L40

bobbilichandu commented 3 years ago

Thanks, it works. I have been using some other export_onnx.py file.