Skip to content

Commit

Permalink
Remove redundant downloads mirror (ultralytics#1461)
Browse files Browse the repository at this point in the history
  • Loading branch information
glenn-jocher authored Nov 20, 2020
1 parent 46c43b7 commit a70e554
Show file tree
Hide file tree
Showing 3 changed files with 4 additions and 2 deletions.
2 changes: 1 addition & 1 deletion test.py
Original file line number Diff line number Diff line change
Expand Up @@ -322,4 +322,4 @@ def test(data,
y.append(r + t) # results and times
np.savetxt(f, y, fmt='%10.4g') # save
os.system('zip -r study.zip study_*.txt')
# utils.general.plot_study_txt(f, x) # plot
# utils.plots.plot_study_txt(f, x) # plot
2 changes: 1 addition & 1 deletion utils/autoanchor.py
Original file line number Diff line number Diff line change
Expand Up @@ -67,7 +67,7 @@ def kmean_anchors(path='./data/coco128.yaml', n=9, img_size=640, thr=4.0, gen=10
k: kmeans evolved anchors
Usage:
from utils.general import *; _ = kmean_anchors()
from utils.autoanchor import *; _ = kmean_anchors()
"""
thr = 1. / thr

Expand Down
2 changes: 2 additions & 0 deletions utils/google_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,6 +22,7 @@ def attempt_download(weights):

msg = weights + ' missing, try downloading from https://github.com/ultralytics/yolov5/releases/'
models = ['yolov5s.pt', 'yolov5m.pt', 'yolov5l.pt', 'yolov5x.pt'] # available models
redundant = False # offer second download option

if file in models and not os.path.isfile(weights):
# Google Drive
Expand All @@ -40,6 +41,7 @@ def attempt_download(weights):
assert os.path.exists(weights) and os.path.getsize(weights) > 1E6 # check
except Exception as e: # GCP
print('Download error: %s' % e)
assert redundant, 'No secondary mirror'
url = 'https://storage.googleapis.com/ultralytics/yolov5/ckpt/' + file
print('Downloading %s to %s...' % (url, weights))
r = os.system('curl -L %s -o %s' % (url, weights)) # torch.hub.download_url_to_file(url, weights)
Expand Down

0 comments on commit a70e554

Please sign in to comment.