Skip to content

Commit

Permalink
add model download info. (meituan#826)
Browse files Browse the repository at this point in the history
  • Loading branch information
mtjhl authored May 12, 2023
1 parent 6e58d1b commit eee6b74
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions yolov6/utils/general.py
Original file line number Diff line number Diff line change
Expand Up @@ -93,6 +93,7 @@ def download_ckpt(path):
LOGGER.info(f"checkpoint {basename} not exist, try to downloaded it from github.")
# need to update the link with every release
url = f"https://github.com/meituan/YOLOv6/releases/download/0.4.0/{basename}"
LOGGER.warning(f"downloading url is: {url}, pealse make sure the version of the downloading model is correspoing to the code version!")
r = requests.get(url, allow_redirects=True)
assert r.status_code == 200, "Unable to download checkpoints, manually download it"
open(path, 'wb').write(r.content)
Expand Down

0 comments on commit eee6b74

Please sign in to comment.