Open bobbilichandu opened 3 years ago
yes, modify the batch size in cfg file
@linghu8812 Does your code supports weights in pytorch format [.pt]?
@chandu1263 refer ScaleYOLOv4 or yolov5
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.
Any help would be appreciated!
is the anchors right?
Yes. Anchor boxes are correct.
How to convert yolov4-csp pt file to onnx with explicit batchsize?
Is changing the batch variable in cfg file correct method?