akadmin/openwakeword
1
0
Fork 0
mirror of https://github.com/dscripka/openWakeWord.git synced 2026-08-27 18:17:20 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix link for audioset data

Browse source
This commit is contained in:
David Scripka 2024-02-19 07:24:07 -05:00
parent bcbfaabdc0
commit 65cf92737d
1 changed files with 502 additions and 492 deletions
Download patch file Download diff file Expand all files Collapse all files

44
notebooks/automatic_model_training.ipynb
View file

@ -207,7 +207,7 @@
"\n",
"fname = \"bal_train09.tar\"\n",
"out_dir = f\"audioset/{fname}\"\n",
"link = \"https://huggingface.co/datasets/agkphysics/AudioSet/resolve/main/\" + fname\n",
"link = \"https://huggingface.co/datasets/agkphysics/AudioSet/resolve/main/data\" + fname\n",
"!wget -O {out_dir} {link}\n",
"!cd audioset && tar -xvf bal_train09.tar\n",
"\n",
@ -416,6 +416,12 @@
},
{
"cell_type": "code",
"execution_count": null,
"id": "JSKWWLalnYzR",
"metadata": {
"id": "JSKWWLalnYzR"
},
"outputs": [],
"source": [
"# Step 4 (Optional): On Google Colab, sometimes the .tflite model isn't saved correctly\n",
"# If so, run this cell to retry\n",
@ -445,32 +451,39 @@
" return None\n",
"\n",
"convert_onnx_to_tflite(f\"my_custom_model/{config['model_name']}.onnx\", f\"my_custom_model/{config['model_name']}.tflite\")\n"
],
"metadata": {
"id": "JSKWWLalnYzR"
},
"id": "JSKWWLalnYzR",
"execution_count": null,
"outputs": []
]
},
{
"cell_type": "markdown",
"source": [
"After the model finishes training, the auto training script will automatically convert it to ONNX and tflite versions, saving them as `my_custom_model/<model_name>.onnx/tflite` in the present working directory, where `<model_name>` is defined in the YAML training config file. Either version can be used as normal with `openwakeword`. I recommend testing them with the [`detect_from_microphone.py`](https://github.com/dscripka/openWakeWord/blob/main/examples/detect_from_microphone.py) example script to see how the model performs!"
],
"id": "f9OyUW3ltOSs",
"metadata": {
"id": "f9OyUW3ltOSs"
},
"id": "f9OyUW3ltOSs"
"source": [
"After the model finishes training, the auto training script will automatically convert it to ONNX and tflite versions, saving them as `my_custom_model/<model_name>.onnx/tflite` in the present working directory, where `<model_name>` is defined in the YAML training config file. Either version can be used as normal with `openwakeword`. I recommend testing them with the [`detect_from_microphone.py`](https://github.com/dscripka/openWakeWord/blob/main/examples/detect_from_microphone.py) example script to see how the model performs!"
]
}
],
"metadata": {
"colab": {
"provenance": []
},
"kernelspec": {
"display_name": "Python 3",
"display_name": "Python 3 (ipykernel)",
"language": "python",
"name": "python3"
},
"language_info": {
"name": "python"
"codemirror_mode": {
"name": "ipython",
"version": 3
},
"file_extension": ".py",
"mimetype": "text/x-python",
"name": "python",
"nbconvert_exporter": "python",
"pygments_lexer": "ipython3",
"version": "3.9.12"
},
"toc": {
"base_numbering": 1,
@ -484,9 +497,6 @@
"toc_position": {},
"toc_section_display": true,
"toc_window_display": false
},
"colab": {
"provenance": []
}
},
"nbformat": 4,