1、簡介
STM32 Model Zoo 是指 ST 在 github 上開源的 AI 相關(guān)模型倉庫,其包含的應(yīng)用包括:手勢識別,人體活動姿態(tài)識別,圖片分類,目標(biāo)檢測等相關(guān)應(yīng)用代碼,用戶可以使用其中的代碼進(jìn)行簡單的配置并進(jìn)行訓(xùn)練,然后使用 STM32Cube.AI 工具或 STM32CubeAI Developer Cloud得到模型的 Flash,Ram 等情況,并進(jìn)行模型的測試驗(yàn)證,然后部署到開發(fā)板中。本文將以手勢識別為例,介紹 Model Zoo 的使用。
2、環(huán)境安裝
詳細(xì)的安裝步驟可以參考 ST 的 wiki 頁面:https://wiki.stmicroelectronics.cn/stm32mcu/index.php?title=AI:How_to_install_STM32_model_zoo
3、手勢識別應(yīng)用
手勢識別應(yīng)用位于 stm32ai-modelzoo 的 hand_posture 目錄下,使用 ST 的 TOF 傳感器 VL53L5CX 或 VL53L8CX 采集數(shù)據(jù)。用于識別靜態(tài)的手勢,比如 like, dislike, love等手勢。其 demo 應(yīng)用部署在 NUCLEO-F401RE 和 X-NUCLEO-53LxA1 擴(kuò)展板組合的開發(fā)板上。
總結(jié)
本文介紹了 STM32 model-zoo 的基本內(nèi)容,并以手勢識別為例,介紹了具體的使用方法,包括模型訓(xùn)練,如何查看模型訓(xùn)練后的結(jié)果以及如何部署模型。對于其他 modelzoo 內(nèi)的應(yīng)用案例,感興趣的話可以去 github 上下載整個(gè)倉庫,然后進(jìn)行閱讀并使用。