mirror of
https://github.com/PhotonVision/photonvision
synced 2026-06-19 00:41:41 +00:00
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:
@@ -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"
|
||||
]
|
||||
},
|
||||
{
|
||||
|
||||
Reference in New Issue
Block a user