diff --git a/weights/download_weights.sh b/weights/download_weights.sh index 22c7712..d799230 100644 --- a/weights/download_weights.sh +++ b/weights/download_weights.sh @@ -1,7 +1,7 @@ #!/bin/bash # Download common models -python3 -c "from models import *; +python3 -c "from utils.google_utils import *; attempt_download('weights/yolov5s.pt'); attempt_download('weights/yolov5m.pt'); attempt_download('weights/yolov5l.pt')"