Supportted YOLOv5 4.0

[pango99]

Does support YOLOv5 4.0?

[linghu8812]

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]

Great,thank you very much for your sharing

[bobbilichandu]

Where can I find yolov5s6.yaml?

[linghu8812]

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

[bobbilichandu]

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]

@chandu1263
check these lines

tensorrt_inference/yolov5/export_onnx.py

Lines 39 to 40 in 5f22509

	elif isinstance(m.act, nn.SiLU):
	m.act = SiLU()

[bobbilichandu]

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