0bc1db58d82c2482bfac1e32a3a43cfd5a533da2,utils/google_utils.py,,attempt_download,#,19

Before Change

def attempt_download(weights):
    // Attempt to download pretrained weights if not found locally
    weights = str(weights).strip().replace(""", "")
    file = Path(weights).name.lower()

    msg = weights + " missing, try downloading from https://github.com/ultralytics/yolov3/releases/"
    response = requests.get("https://api.github.com/repos/ultralytics/yolov3/releases/latest").json()  // github api
    assets = [x["name"] for x in response["assets"]]  // release assets ["yolov3.pt", "yolov3-spp.pt", "yolov3-tiny.pt"]

After Change

    // Attempt file download if does not exist
    file = Path(str(file).strip().replace(""", "").lower())

    if not file.exists():
        response = requests.get("https://api.github.com/repos/ultralytics/yolov3/releases/latest").json()  // github api
        assets = [x["name"] for x in response["assets"]]  // release assets ["yolov3.pt", "yolov3-spp.pt", ...]
        name = file.name

        if name in assets:
            msg = f"{file} missing, try downloading from https://github.com/ultralytics/yolov3/releases/"
            redundant = False  // second download option
            try:  // GitHub
                tag = response["tag_name"]  // i.e. "v1.0"
                url = f"https://github.com/ultralytics/yolov3/releases/download/{tag}/{name}"
                print(f"Downloading {url} to {file}...")
                torch.hub.download_url_to_file(url, file)
                assert file.exists() and file.stat().st_size > 1E6  // check
            except Exception as e:  // GCP
                print(f"Download error: {e}")
                assert redundant, "No secondary mirror"
                url = f"https://storage.googleapis.com/ultralytics/yolov3/ckpt/{name}"
                print(f"Downloading {url} to {file}...")
                os.system(f"curl -L {url} -o {file}")  // torch.hub.download_url_to_file(url, weights)
            finally:
                if not file.exists() or file.stat().st_size < 1E6:  // check
                    file.unlink(missing_ok=True)  // remove partial downloads
                    print(f"ERROR: Download failure: {msg}")
                print("")
                return


def gdrive_download(id="16TiPfZj7htmTyhntwcZyEEAejOUxuT6m", file="tmp.zip"):
    // Downloads a file from Google Drive. from yolov3.utils.google_utils import *; gdrive_download()
    t = time.time()
    file = Path(file)

Instances