administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-04-27 17:17:40 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Update YOLO_NAS_Pretrained_Export.ipynb

Browse Source
This commit is contained in:
Manuel López Antequera 2025-07-04 21:34:28 +02:00 committed by GitHub
parent 4339f26ded
commit ce66486e10
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 14 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
notebooks/YOLO_NAS_Pretrained_Export.ipynb
View File

@ -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": {