diff --git a/models.py b/models.py index b3670b62..6401ab6c 100755 --- a/models.py +++ b/models.py @@ -433,6 +433,8 @@ def attempt_download(weights): gdrive_download(id='16lYS4bcIdM2HdmyJBVDOvt3Trx6N3W2R', name=weights) elif file == 'yolov3.weights': gdrive_download(id='1uTlyDWlnaqXcsKOktP5aH_zRDbfcDp-y', name=weights) + elif file == 'yolov3-tiny.weights': + gdrive_download(id='1CCF-iNIIkYesIDzaPvdwlcf7H9zSsKZQ', name=weights) elif file == 'yolov3-spp.pt': gdrive_download(id='1f6Ovy3BSq2wYq4UfvFUpxJFNDFfrIDcR', name=weights) elif file == 'yolov3.pt':