Mirrors/blakeblackshear.frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2025-07-30 13:48:07 +02:00
Code Issues Projects Releases Wiki Activity

Merge ce66486e10 into 0c92c3ccfa

Browse Source
This commit is contained in:
Manuel López Antequera 2025-07-22 11:30:44 -03:00 committed by GitHub
commit 4a3a7e1db9
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", "\n",
"files.download('yolo_nas_s.onnx')" "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": { "metadata": {