Update RubikPi Collab Notebook with working Device (#2339)

The RubikPi collab notebook included a device for creating a quantized
model that was deprecated by the Qualcomm team. I've included a separate
cell with a command to check available devices and updated the command
to a known working state.
This commit is contained in:
John Fogarty
2026-01-27 23:11:25 -08:00
committed by GitHub
parent 7cc22e52ea
commit 1b5f4fa802

View File

@@ -39,20 +39,35 @@
"execution_count": null,
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/",
"height": 1000
"base_uri": "https://localhost:8080/"
},
"id": "aX3JcSFKp58s",
"outputId": "f2cdadd2-c448-4d8c-c681-c19decef7f3e"
"outputId": "7fffa581-fb85-4808-84d6-b737142011b2"
},
"outputs": [],
"source": [
"# This installs Python package\n",
"!pip install qai-hub-models[yolov8_det]\n",
"# sets up AI Hub enviroment\n",
"!qai-hub configure --api_token <YOUR_API_TOKEN>\n",
"!qai-hub configure --api_token [YOUR API KEY]\n",
"# Converts the model to be ran on RB3Gen2\n",
"!yes | python -m qai_hub_models.models.yolov8_det.export --quantize w8a8 --device=\"RB3 Gen 2 (Proxy)\" --ckpt-name /PATH/TO/WEIGHTS --device-os linux --target-runtime tflite --output-dir .\n"
"!yes | python -m qai_hub_models.models.yolov8_det.export --quantize w8a8 --device=\"Dragonwing RB3 Gen 2 Vision Kit\" --ckpt-name /content/PyTorch_model.pt --device-os 1.6 --target-runtime tflite --output-dir .\n"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/"
},
"id": "jl9M3nPDqlue",
"outputId": "2df2c1af-c4ee-4cbf-dad6-7ec3afd28337"
},
"outputs": [],
"source": [
"# check valid devices / os types\n",
"!qai-hub list-devices"
]
},
{