Glenn Jocher
4 years ago
2 changed files with 2 additions and 8 deletions
+ 1
- 4
utils/general.py
View File
| print('Downloading %s ...' % s) | print('Downloading %s ...' % s) | ||||
| if s.startswith('http') and s.endswith('.zip'): # URL | if s.startswith('http') and s.endswith('.zip'): # URL | ||||
| f = Path(s).name # filename | f = Path(s).name # filename | ||||
| if platform.system() == 'Darwin': # avoid MacOS python requests certificate error | |||||
| os.system('curl -L %s -o %s' % (s, f)) | |||||
| else: | |||||
| torch.hub.download_url_to_file(s, f) | |||||
| torch.hub.download_url_to_file(s, f) | |||||
| r = os.system('unzip -q %s -d ../ && rm %s' % (f, f)) # unzip | r = os.system('unzip -q %s -d ../ && rm %s' % (f, f)) # unzip | ||||
| else: # bash script | else: # bash script | ||||
| r = os.system(s) | r = os.system(s) |
+ 1
- 4
utils/google_utils.py
View File
| try: # GitHub | try: # GitHub | ||||
| url = 'https://github.com/ultralytics/yolov5/releases/download/v3.0/' + file | url = 'https://github.com/ultralytics/yolov5/releases/download/v3.0/' + file | ||||
| print('Downloading %s to %s...' % (url, weights)) | print('Downloading %s to %s...' % (url, weights)) | ||||
| if platform.system() == 'Darwin': # avoid MacOS python requests certificate error | |||||
| r = os.system('curl -L %s -o %s' % (url, weights)) | |||||
| else: | |||||
| torch.hub.download_url_to_file(url, weights) | |||||
| torch.hub.download_url_to_file(url, weights) | |||||
| assert os.path.exists(weights) and os.path.getsize(weights) > 1E6 # check | assert os.path.exists(weights) and os.path.getsize(weights) > 1E6 # check | ||||
| except Exception as e: # GCP | except Exception as e: # GCP | ||||
| print('Download error: %s' % e) | print('Download error: %s' % e) |
Loading…