Ver código fonte

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

modifyDataloader
Glenn Jocher GitHub 3 anos atrás
pai
commit
3e7c59ad3b
Nenhuma chave conhecida encontrada para esta assinatura no banco de dados ID da chave GPG: 4AEE18F83AFDEB23
2 arquivos alterados com 3 adições e 3 exclusões
  1. +1
    -1
      data/scripts/download_weights.sh
  2. +2
    -2
      utils/downloads.py

+ 1
- 1
data/scripts/download_weights.sh Ver arquivo

@@ -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')

+ 2
- 2
utils/downloads.py Ver arquivo

@@ -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

Carregando…
Cancelar
Salvar