administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2025-12-08 14:25:41 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Undo

Browse Source
This commit is contained in:
Nicolas Mowen 2025-09-25 09:08:43 -06:00
parent 2a40e3d9ab
commit ce624ab67d
1 changed files with 85 additions and 86 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
notebooks/YOLO_NAS_Pretrained_Export.ipynb
View File

@ -48,14 +48,13 @@
"source": [ "source": [
"# export the model for compatibility with Frigate\n", "# export the model for compatibility with Frigate\n",
"\n", "\n",
"model.export(\n", "model.export(\"yolo_nas_s.onnx\",\n",
" \"yolo_nas_s.onnx\",\n",
" output_predictions_format=DetectionOutputFormatMode.FLAT_FORMAT,\n", " output_predictions_format=DetectionOutputFormatMode.FLAT_FORMAT,\n",
" max_predictions_per_image=20,\n", " max_predictions_per_image=20,\n",
" num_pre_nms_predictions=300,\n", " num_pre_nms_predictions=300,\n",
" confidence_threshold=0.4,\n", " confidence_threshold=0.4,\n",
" input_image_shape=(320, 320),\n", " input_image_shape=(320,320),\n",
")" " )"
] ]
}, },
{ {
@ -68,7 +67,7 @@
"source": [ "source": [
"from google.colab import files\n", "from google.colab import files\n",
"\n", "\n",
"files.download(\"yolo_nas_s.onnx\")" "files.download('yolo_nas_s.onnx')"
] ]
} }
], ],