Fix rename utils.google_utils to utils.downloads (ultralytics#4393)

fhkiel-mlaip · Aug 11, 2021 · 500fe62 · 500fe62
1 parent a10e500
commit 500fe62
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/data/scripts/download_weights.sh b/data/scripts/download_weights.sh
@@ -9,7 +9,7 @@
 #     └── ...
 
 python - <<EOF
-from utils.google_utils import attempt_download
+from utils.downloads import attempt_download
 
 for x in ['s', 'm', 'l', 'x']:
     attempt_download(f'yolov5{x}.pt')

diff --git a/utils/downloads.py b/utils/downloads.py
@@ -36,7 +36,7 @@ def safe_download(file, url, url2=None, min_bytes=1E0, error_msg=''):
         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
     file = Path(str(file).strip().replace("'", ''))
 
@@ -74,7 +74,7 @@ def attempt_download(file, repo='ultralytics/yolov5'):  # from utils.google_util
 
 
 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()
     file = Path(file)
     cookie = Path('cookie')  # gdrive cookie