Glenn Jocher
GitHub
3 years ago
2 changed files with 3 additions and 3 deletions
+ 1
- 1
data/scripts/download_weights.sh
View File
| # └── ... | # └── ... | ||||
| python - <<EOF | python - <<EOF | ||||
| from utils.google_utils import attempt_download | |||||
| from utils.downloads import attempt_download | |||||
| for x in ['s', 'm', 'l', 'x']: | for x in ['s', 'm', 'l', 'x']: | ||||
| attempt_download(f'yolov5{x}.pt') | attempt_download(f'yolov5{x}.pt') |
+ 2
- 2
utils/downloads.py
View File
| print('') | print('') | ||||
| def attempt_download(file, repo='ultralytics/yolov5'): # from utils.google_utils import *; attempt_download() | |||||
| def attempt_download(file, repo='ultralytics/yolov5'): # from utils.downloads import *; attempt_download() | |||||
| # Attempt file download if does not exist | # Attempt file download if does not exist | ||||
| file = Path(str(file).strip().replace("'", '')) | file = Path(str(file).strip().replace("'", '')) | ||||
| def gdrive_download(id='16TiPfZj7htmTyhntwcZyEEAejOUxuT6m', file='tmp.zip'): | def gdrive_download(id='16TiPfZj7htmTyhntwcZyEEAejOUxuT6m', file='tmp.zip'): | ||||
| # Downloads a file from Google Drive. from yolov5.utils.google_utils import *; gdrive_download() | |||||
| # Downloads a file from Google Drive. from yolov5.utils.downloads import *; gdrive_download() | |||||
| t = time.time() | t = time.time() | ||||
| file = Path(file) | file = Path(file) | ||||
| cookie = Path('cookie') # gdrive cookie | cookie = Path('cookie') # gdrive cookie |
Loading…