From 27138c2ebb00d3a2c94245d1b959328b414635e8 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Tue, 10 Nov 2020 14:28:45 +0100
Subject: [PATCH] Update download_weights.sh (#1338)

---
 weights/download_weights.sh | 16 ++++++++--------
 1 file changed, 8 insertions(+), 8 deletions(-)

diff --git a/weights/download_weights.sh b/weights/download_weights.sh
index 206b700..c33f75e 100755
--- a/weights/download_weights.sh
+++ b/weights/download_weights.sh
@@ -1,10 +1,10 @@
 #!/bin/bash
-# Download common models
+# Download latest model from https://github.com/ultralytics/yolov5/releases
 
-python -c "
-from utils.google_utils import *;
-attempt_download('weights/yolov5s.pt');
-attempt_download('weights/yolov5m.pt');
-attempt_download('weights/yolov5l.pt');
-attempt_download('weights/yolov5x.pt')
-"
+python - <<EOF
+from utils.google_utils import attempt_download
+
+for x in ['s', 'm', 'l', 'x']:
+    attempt_download(f'yolov5{x}.pt')
+
+EOF