Tianxiaomo / pytorch-YOLOv4

PyTorch ,ONNX and TensorRT implementation of YOLOv4
Apache License 2.0
4.46k stars 1.49k forks source link

darknet2onnx(inference part) won't work well with yolov4-csp-s-mish.weights #525

Open JiaPai12138 opened 2 years ago

JiaPai12138 commented 2 years ago

weights: yolov4-csp-s-mish cfg: yolov4-csp-s-mish

Boxes are larger and confidences are dropping for example, trained with a same dateset to generate both yolov4-tiny and yolov4-csp-s-mish weights convert to onnx using demo_darknet2onnx darknet version of both weights worked fine onnx version of yolov4-tiny worked fine while results of yolov4-csp-s-mish looks wired