From fee83c1634ddce6a95f46e8fab85699cfa069c73 Mon Sep 17 00:00:00 2001
From: Cristi Fati <fati_utcluj@yahoo.com>
Date: Sat, 23 Oct 2021 15:20:26 +0300
Subject: [PATCH] Weights download script minor improvements (#5213)

* Add nano model to the download list

* Add possibility to also download the "*6" models

* Cleanup

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
---
 data/scripts/download_weights.sh | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/data/scripts/download_weights.sh b/data/scripts/download_weights.sh
index b4b0ccd..e9fa653 100755
--- a/data/scripts/download_weights.sh
+++ b/data/scripts/download_weights.sh
@@ -11,7 +11,10 @@
 python - <<EOF
 from utils.downloads import attempt_download
 
-for x in ['s', 'm', 'l', 'x']:
+models = ['n', 's', 'm', 'l', 'x']
+models.extend([x + '6' for x in models])  # add P6 models
+
+for x in models:
     attempt_download(f'yolov5{x}.pt')
 
 EOF