YOLOv5 v7.0 release updates (#10245)

* YOLOv5 v7.0 splash image update * Update tutorial.ipynb Signed-off-by: Glenn Jocher <[email protected]> * Update tutorial.ipynb Signed-off-by: Glenn Jocher <[email protected]> * Update tutorial.ipynb Signed-off-by: Glenn Jocher <[email protected]> * readme segmentation section * readme segmentation section * readme segmentation section * readme segmentation section * readme segmentation section * Update README.md Signed-off-by: Glenn Jocher <[email protected]> * Update README.md Signed-off-by: Glenn Jocher <[email protected]> * Update tutorial.ipynb Signed-off-by: Glenn Jocher <[email protected]> * Update README.md Signed-off-by: Glenn Jocher <[email protected]> * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Update README.md Signed-off-by: Glenn Jocher <[email protected]> * Update README.md Signed-off-by: Glenn Jocher <[email protected]> * Update download URLs to 7.0 assets Signed-off-by: Glenn Jocher <[email protected]> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
ultralytics · Nov 22, 2022 · 915bbf2 · 915bbf2
1 parent be348cc
commit 915bbf2
Show file tree

Hide file tree

Showing 7 changed files with 107 additions and 31 deletions.
diff --git a/.github/README_cn.md b/.github/README_cn.md
@@ -1,7 +1,7 @@
 <div align="center">
   <p>
     <a align="center" href="https://ultralytics.com/yolov5" target="_blank">
-      <img width="850" src="https://github.com/ultralytics/assets/blob/master/yolov5/v62/splash_readme.png"></a>
+      <img width="850" src="https://raw.githubusercontent.com/ultralytics/assets/master/yolov5/v70/splash.png"></a>
   </p>
 
   [English](../README.md) | 简体中文

diff --git a/README.md b/README.md
diff --git a/classify/tutorial.ipynb b/classify/tutorial.ipynb
@@ -9,7 +9,7 @@
         "<div align=\"center\">\n",
         "\n",
         "  <a href=\"https://ultralytics.com/yolov5\" target=\"_blank\">\n",
-        "    <img width=\"1024\", src=\"https://github.com/ultralytics/assets/raw/master/yolov5/v62/splash_readme.png\"></a>\n",
+        "    <img width=\"1024\", src=\"https://raw.githubusercontent.com/ultralytics/assets/master/yolov5/v70/splash.png\"></a>\n",
         "\n",
         "\n",
         "<br>\n",
@@ -1452,7 +1452,8 @@
     "accelerator": "GPU",
     "colab": {
       "name": "YOLOv5 Classification Tutorial",
-      "provenance": []
+      "provenance": [],
+      "toc_visible": true
     },
     "kernelspec": {
       "display_name": "Python 3 (ipykernel)",

diff --git a/data/scripts/download_weights.sh b/data/scripts/download_weights.sh
@@ -11,11 +11,12 @@
 python - <<EOF
 from utils.downloads import attempt_download
 
-p5 = ['n', 's', 'm', 'l', 'x']  # P5 models
+p5 = list('nsmlx')  # P5 models
 p6 = [f'{x}6' for x in p5]  # P6 models
 cls = [f'{x}-cls' for x in p5]  # classification models
+seg = [f'{x}-seg' for x in p5]  # classification models
 
-for x in p5 + p6 + cls:
+for x in p5 + p6 + cls + seg:
     attempt_download(f'weights/yolov5{x}.pt')
 
 EOF
diff --git a/segment/tutorial.ipynb b/segment/tutorial.ipynb
@@ -9,7 +9,7 @@
         "<div align=\"center\">\n",
         "\n",
         "  <a href=\"https://ultralytics.com/yolov5\" target=\"_blank\">\n",
-        "    <img width=\"1024\", src=\"https://github.com/ultralytics/assets/raw/master/yolov5/v62/splash_readme.png\"></a>\n",
+        "    <img width=\"1024\", src=\"https://raw.githubusercontent.com/ultralytics/assets/master/yolov5/v70/splash.png\"></a>\n",
         "\n",
         "\n",
         "<br>\n",

diff --git a/tutorial.ipynb b/tutorial.ipynb
@@ -369,7 +369,7 @@
         "<div align=\"center\">\n",
         "\n",
         "  <a href=\"https://ultralytics.com/yolov5\" target=\"_blank\">\n",
-        "    <img width=\"1024\", src=\"https://github.com/ultralytics/assets/raw/master/yolov5/v62/splash_readme.png\"></a>\n",
+        "    <img width=\"1024\", src=\"https://raw.githubusercontent.com/ultralytics/assets/master/yolov5/v70/splash.png\"></a>\n",
         "\n",
         "\n",
         "<br>\n",

diff --git a/utils/downloads.py b/utils/downloads.py
@@ -59,14 +59,14 @@ def safe_download(file, url, url2=None, min_bytes=1E0, error_msg=''):
         LOGGER.info('')
 
 
-def attempt_download(file, repo='ultralytics/yolov5', release='v6.2'):
-    # Attempt file download from GitHub release assets if not found locally. release = 'latest', 'v6.2', etc.
+def attempt_download(file, repo='ultralytics/yolov5', release='v7.0'):
+    # Attempt file download from GitHub release assets if not found locally. release = 'latest', 'v7.0', etc.
     from utils.general import LOGGER
 
     def github_assets(repository, version='latest'):
-        # Return GitHub repo tag (i.e. 'v6.2') and assets (i.e. ['yolov5s.pt', 'yolov5m.pt', ...])
+        # Return GitHub repo tag (i.e. 'v7.0') and assets (i.e. ['yolov5s.pt', 'yolov5m.pt', ...])
         if version != 'latest':
-            version = f'tags/{version}'  # i.e. tags/v6.2
+            version = f'tags/{version}'  # i.e. tags/v7.0
         response = requests.get(f'https://api.github.com/repos/{repository}/releases/{version}').json()  # github api
         return response['tag_name'], [x['name'] for x in response['assets']]  # tag, assets