search

linghu8812 / tensorrt_inference

699 stars 205 forks source link

[YOLOV4-CSP] ONNX CONVERSION #71

Open bobbilichandu opened 3 years ago

bobbilichandu commented 3 years ago

How to convert yolov4-csp pt file to onnx with explicit batchsize?

Is changing the batch variable in cfg file correct method?

linghu8812 commented 3 years ago

yes, modify the batch size in cfg file

bobbilichandu commented 3 years ago

@linghu8812 Does your code supports weights in pytorch format [.pt]?

linghu8812 commented 3 years ago

@chandu1263 refer ScaleYOLOv4 or yolov5

bobbilichandu commented 3 years ago

Hey @linghu8812 I tried doing the same, but it didn't work. Following ScaledYOLOv4 to train csp model, I got a .pt weights file. I tried converting it, but it was not working. There were multiple bboxes for the same object. person_ zidane_ bus_

Any help would be appreciated!

linghu8812 commented 3 years ago

is the anchors right?

bobbilichandu commented 3 years ago

Yes. Anchor boxes are correct.