auto-download v3.1 models (#1259)

This commit is contained in:
Glenn Jocher authored and GitHub committed 2020-11-01 20:11:41 +01:00
1 parent ca290dca24
commit 2062765e06
2 files changed
+7 -7

No files matched your search

+1 -1
View File
@@ -36,7 +36,7 @@ def attempt_download(weights):
# return
try: # GitHub
url = 'https://github.com/ultralytics/yolov5/releases/download/v3.0/' + file
url = 'https://github.com/ultralytics/yolov5/releases/download/v3.1/' + file
print('Downloading %s to %s...' % (url, weights))
torch.hub.download_url_to_file(url, weights)
assert os.path.exists(weights) and os.path.getsize(weights) > 1E6 # check