Tianxiaomo / pytorch-YOLOv4

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

Bugfix: do_detect gets 4 or 5 arg, but receives 6 #547

Open DShaience opened 1 year ago

DShaience commented 1 year ago

do_detect() has an extra argument. Comparing to previous version used to be boxes = do_detect(model, sized, 0.0, 0.4, use_cuda) but now it is boxes = do_detect(model, sized, 0.0, 80, 0.4, use_cuda) Is this a typo? do_detect() wasn't changed to accommodate an extra argument.

DShaience commented 1 year ago

Suspected bug