cec59f12c8b0800d00949ea15511469f667d45dd,models.py,,attempt_download,#,453

Before Change

def attempt_download(weights):
    // Attempt to download pretrained weights if not found locally
    weights = weights.strip()
    msg = weights + " missing, try downloading from https://drive.google.com/open?id=1LezFG5g3BCW6iYaV89B2i64cqEUZD7e0"

    if len(weights) > 0 and not os.path.isfile(weights):
        d = {"yolov3-spp.weights": "16lYS4bcIdM2HdmyJBVDOvt3Trx6N3W2R",
             "yolov3.weights": "1uTlyDWlnaqXcsKOktP5aH_zRDbfcDp-y",
             "yolov3-tiny.weights": "1CCF-iNIIkYesIDzaPvdwlcf7H9zSsKZQ",
             "yolov3-spp.pt": "1f6Ovy3BSq2wYq4UfvFUpxJFNDFfrIDcR",
             "yolov3.pt": "1SHNFyoe5Ni8DajDNEqgB2oVKBb_NoEad",
             "yolov3-tiny.pt": "10m_3MlpQwRtZetQxtksm9jqHrPTHZ6vo",
             "darknet53.conv.74": "1WUVBid-XuoUBmvzBVUCBl_ELrzqwA8dJ",
             "yolov3-tiny.conv.15": "1Bw0kCpplxUqyRYAJr9RY9SGnOJbo9nEj",
             "yolov3-spp-ultralytics.pt": "1UcR-zVoMs7DH5dj3N1bswkiQTA4dmKF4"}

        file = Path(weights).name
        if file in d:
            r = gdrive_download(id=d[file], name=weights)
        else:  // download from pjreddie.com
            url = "https://pjreddie.com/media/files/" + file
            print("Downloading " + url)
            r = os.system("curl -f " + url + " -o " + weights)

After Change

def attempt_download(weights):
    // Attempt to download pretrained weights if not found locally
    weights = weights.strip().replace(""", "")
    msg = weights + " missing, try downloading from https://drive.google.com/open?id=1LezFG5g3BCW6iYaV89B2i64cqEUZD7e0"

    if len(weights) > 0 and not os.path.isfile(weights):
        d = {"yolov3-spp.weights": "16lYS4bcIdM2HdmyJBVDOvt3Trx6N3W2R",
             "yolov3.weights": "1uTlyDWlnaqXcsKOktP5aH_zRDbfcDp-y",
             "yolov3-tiny.weights": "1CCF-iNIIkYesIDzaPvdwlcf7H9zSsKZQ",
             "yolov3-spp.pt": "1f6Ovy3BSq2wYq4UfvFUpxJFNDFfrIDcR",
             "yolov3.pt": "1SHNFyoe5Ni8DajDNEqgB2oVKBb_NoEad",
             "yolov3-tiny.pt": "10m_3MlpQwRtZetQxtksm9jqHrPTHZ6vo",
             "darknet53.conv.74": "1WUVBid-XuoUBmvzBVUCBl_ELrzqwA8dJ",
             "yolov3-tiny.conv.15": "1Bw0kCpplxUqyRYAJr9RY9SGnOJbo9nEj",
             "yolov3-spp-ultralytics.pt": "1UcR-zVoMs7DH5dj3N1bswkiQTA4dmKF4"}

        file = Path(weights).name
        if file in d:
            r = gdrive_download(id=d[file], name=weights)
        else:  // download from pjreddie.com
            url = "https://pjreddie.com/media/files/" + file
            print("Downloading " + url)
            r = os.system("curl -f " + url + " -o " + weights)

Instances