Merge ce66486e10 into 0c92c3ccfa

notebooks/YOLO_NAS_Pretrained_Export.ipynb

@@ -69,6 +69,20 @@
         "\n",
         "files.download('yolo_nas_s.onnx')"
       ]
+    },
+    {
+      "cell_type": "code",
+      "source": [
+        "labels = model.get_class_names()\n",
+        "with open(\"coco-80.txt\", \"w\") as f:\n",
+        "  f.write(\"\\n\".join(labels))\n",
+        "files.download(\"coco-80.txt\")"
+      ],
+      "metadata": {
+        "id": "KEhxK_lzvlqk"
+      },
+      "execution_count": null,
+      "outputs": []
     }
   ],
   "metadata": {