feat: add functionality to zip and download 'runs' folder with saved …

…best models
healthonrails · Dec 6, 2024 · 1337222 · 1337222
1 parent 17f88a4
commit 1337222
Show file tree

Hide file tree

Showing 2 changed files with 44 additions and 5 deletions.
diff --git a/annolid/segmentation/yolos.py b/annolid/segmentation/yolos.py
@@ -175,5 +175,5 @@ def save_yolo_to_labelme(self, yolo_results, frame_shape, output_dir):
     video_path = os.path.expanduser("~/Downloads/IMG_0769.MOV")
 
     # Automatically find best.pt or use default
-    yolo_processor = InferenceProcessor(model_type="yolo")
+    yolo_processor = InferenceProcessor("yolo11n-seg.pt",model_type="yolo")
     yolo_processor.run_inference(video_path)
diff --git a/docs/tutorials/Annolid_Instance_Segmentation_on_YOLO11_Tutorial.ipynb b/docs/tutorials/Annolid_Instance_Segmentation_on_YOLO11_Tutorial.ipynb
@@ -48,6 +48,19 @@
       "execution_count": null,
       "outputs": []
     },
+    {
+      "cell_type": "code",
+      "source": [
+        "import shutil\n",
+        "import os\n",
+        "from google.colab import files"
+      ],
+      "metadata": {
+        "id": "fJwI06VXAd5l"
+      },
+      "execution_count": null,
+      "outputs": []
+    },
     {
       "cell_type": "markdown",
       "source": [
@@ -60,7 +73,6 @@
     {
       "cell_type": "code",
       "source": [
-        "from google.colab import files\n",
         "custom_dataset = files.upload()"
       ],
       "metadata": {
@@ -406,8 +418,6 @@
     {
       "cell_type": "code",
       "source": [
-        "import shutil\n",
-        "import os\n",
         "\n",
         "# Provide the path to the folder you want to zip\n",
         "folder_to_zip = output_dir  # Update with your folder path\n",
@@ -418,14 +428,43 @@
         "# Zip the folder\n",
         "shutil.make_archive(folder_to_zip, 'zip', folder_to_zip)\n",
         "\n",
-        "print(f\"Folder '{folder_to_zip}' has been zipped to '{output_zip_file}'.\")"
+        "print(f\"Folder '{folder_to_zip}' has been zipped to '{output_zip_file}'.\")\n",
+        "files.download(output_zip_file)"
       ],
       "metadata": {
         "id": "T-OfoVvzWA4K"
       },
       "execution_count": null,
       "outputs": []
     },
+    {
+      "cell_type": "markdown",
+      "source": [
+        "# Zip and download runs folder which continas the saved best model"
+      ],
+      "metadata": {
+        "id": "iUMFYuLk_waJ"
+      }
+    },
+    {
+      "cell_type": "code",
+      "source": [
+        "# Replace 'folder_name' with the name of your folder\n",
+        "folder_to_download = 'runs'\n",
+        "output_filename = 'runs.zip'\n",
+        "\n",
+        "# Compress the folder\n",
+        "shutil.make_archive(output_filename.replace('.zip', ''), 'zip', folder_to_download)\n",
+        "\n",
+        "# Download the zipped folder\n",
+        "files.download(output_filename)"
+      ],
+      "metadata": {
+        "id": "YuAW2iF8_jbq"
+      },
+      "execution_count": null,
+      "outputs": []
+    },
     {
       "cell_type": "markdown",
       "source": [