sahi icon indicating copy to clipboard operation
sahi copied to clipboard

Improve yolov8 config

Open GuillaumeBruand opened this issue 1 year ago • 1 comments

  • Allow user to pass custom yolo config through the standard yolo yaml config file
  • Pass the confidence threshold to inner model to avoid prefiltering with default yolo threshold=0.25

For yolov8, the predict configuration is given at inference time. We can pass full parameters using the "cfg" argument and a yaml configuration file.

Passing the "conf" argument using confidence_threshold also avoids to prefilter boxes in the underlying yolo model.

GuillaumeBruand avatar Jan 17 '24 17:01 GuillaumeBruand

Amazing contribution @GuillaumeBruand ! 🚀

Please fix the errors raised in tests by:

  • Install required development packages:
pip install -e ."[dev]"
  • Reformat with black and isort:
python -m scripts.run_code_style format

Then commit+push

fcakyon avatar Apr 08 '24 23:04 fcakyon

Thanks for the amazing contribution @GuillaumeBruand !

fcakyon avatar May 19 '24 23:05 fcakyon