auto-download v3.1 models (#1259)
This commit is contained in:
1 parent
ca290dca24
commit
2062765e06
2 files changed
+7
-7
No files matched your search
@@ -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
|
||||
|
||||
Reference in new issue
Block a user