yolov8 / README.md
loreleva's picture
Update README.md
2e3640b verified

Download the models

Use the package huggingface_hub to download the model using Python.

  • Install it with
pip install huggingface_hub
  • Download the weights (e.g., yolov8n-fp32-imgsz640.engine) with:
from huggingface_hub import hf_hub_download

hf_hub_download(repo_id="loreleva/yolov8",
filename="yolov8n-seg/tensorrt/yolov8n-fp32-imgsz640.engine",
local_dir="/path/to/dir")