算豐TDL SDK 是一款集成算法開發(fā)包,提供了便捷的編程接口,使開發(fā)者們能夠更快、高效地將各種智能技術(shù)應(yīng)用于自己的產(chǎn)品和服務(wù)中。無論開發(fā)者是否是智算領(lǐng)域的專家,有了這個SDK作為助手,都能輕松構(gòu)建智能應(yīng)用。
開發(fā)者們可以利用這一簡單易用的工具,把更多精力放在算法模型、應(yīng)用創(chuàng)新和優(yōu)化上,而不必耗費大量時間去處理模型部署相關(guān)的問題。今天,就讓我們來領(lǐng)略它是如何助力開發(fā)者來構(gòu)建指南應(yīng)用。
一、TDL SDK:智能開發(fā)的得力助手
算豐推出的TDL SDK,支持眾多常用算法,提供了統(tǒng)一友好的編程接口及詳盡的配套開發(fā)文檔。這意味著開發(fā)者們即便是初次嘗試,也能迅速掌握并開始高效的開發(fā)工作,大大縮減了開發(fā)者在應(yīng)用程序開發(fā)上所需的時間。
下圖是TDL SDK的系統(tǒng)架構(gòu)圖,基于算豐自研的Middleware和TPU SDK,包括內(nèi)部兩大模塊(Core和Service)、算法C接口、算法應(yīng)用(Application)。
圖1 TDL SDK系統(tǒng)框架
Core提供了算法相關(guān)接口,封裝復(fù)雜的底層操作及算法細節(jié),在內(nèi)部會對模型進行相應(yīng)的前后處理,并完成推理。Service提供算法相關(guān)輔助API,例如:繪圖, 特征比對, 區(qū)域入侵判定等功能。C接口實現(xiàn)對現(xiàn)有現(xiàn)有算法模塊的功能封裝,除了支持TDL SDK內(nèi)部模型外,還支持開發(fā)者自有模型(需按文檔進行模型轉(zhuǎn)換)。Application封裝應(yīng)用邏輯,如包含人臉抓拍的應(yīng)用邏輯。
二、如何使用TDL SDK?
使用TDL SDK非常簡單,首先我們需要在算豐的開發(fā)者論壇https://developer.sophgo.com/thread/556.html下載我們的TDL SDK。
接下來我們以YOLO V8模型為例,演示如何借助TDLSDK,快速將模型部署到算豐開發(fā)板上。
準備YOLO V8 的pt/onnx模型。首先我們可以從yolov8的官方倉庫(https://github.com/ultralytics/ultralytics)中獲取到pt模型文件,并進行下載。
將pt模型轉(zhuǎn)換為onnx,再由onnx轉(zhuǎn)換為部署模型所需的cvimodel文件。
這里大家可以參考這份文檔來進行模型轉(zhuǎn)換,里面有非常詳細的指令說明:https://doc.sophgo.com/cvitek-develop-docs/master/docs_latest_release/CV180x_CV181x/zh/01.software/TPU/YOLO_Development_Guide/build/html/index.html
配置交叉編譯環(huán)境
wget
https://sophon-file.sophon.cn/sophon-prod-s3/drive/23/03/07/16/host-tools.tar.gz
tarxvfhost-tools.tar.gz
cdhost-tools
exportPATH=$PATH:$(pwd)/gcc/riscv64-linux-musl-x86_64/bin
編譯cvitek_tdl_sdk
/*進入到cvitek_tdl_sdk的sample目錄下并執(zhí)行一鍵編譯*/
chmod777compile_sample.sh
./compile_sample.sh
配置開發(fā)板的環(huán)境變量將編譯好的程序放到算能開發(fā)板上后,我們還需要引入依賴庫路徑,才能完成實際的部署。
exportLD_LIBRARY_PATH=/tmp/lfh/cvitek_tdl_sdk/lib:\
/tmp/lfh/cvitek_tdl_sdk/sample/3rd/opencv/lib:\
/tmp/lfh/cvitek_tdl_sdk/sample/3rd/tpu/lib:\
/tmp/lfh/cvitek_tdl_sdk/sample/3rd/ive/lib:\
/tmp/lfh/cvitek_tdl_sdk/sample/3rd/middleware/v2/lib:\
/tmp/lfh/cvitek_tdl_sdk/sample/3rd/lib:\
/tmp/lfh/cvitek_tdl_sdk/sample/3rd/middleware/v2/lib/3rd:
注: /tmp/lfh為示例中開發(fā)板的掛載文件夾,請以你的實際路徑為準。
部署完成,運行模型
chmod777sample_yolov8
./sample_yolov8/tmp/lfh/yolov8n_headperson.cvimodel
/tmp/lfh/pic1.jpg
指令運行成功,即可說明我們的模型已經(jīng)順利移植部署到了板子上啦。以下為可視化展示(mAP(0.5): 86.4%):
除了單獨的人形檢測,我們也提供了基于yolo v8推理的人形與交通工具的檢測(mAP(0.5): 81.8%),客流統(tǒng)計(平均準確率: 96.3%)等等;
圖3 客流統(tǒng)計
三、TDL SDK支持哪些算法?
TDL SDK作為一個便捷的開發(fā)工具包,就像是一個種類豐富的算法超市,涵蓋了各種智能應(yīng)用場景。以下我們列出了當前已經(jīng)支持的算法:
功能描述 | 模型接口 |
---|---|
使用ScrFD Face模型偵測人臉 | CVI_TDL_ScrFDFace |
抽取人臉識別特征 | CVI_TDL_FaceRecognition |
人臉質(zhì)量評估與人臉角度偵測 | CVI_TDL_FaceQuality |
判斷人臉是否戴口罩 | CVI_TDL_MaskClassification |
人臉屬性 | CVI_TDL_FaceAttribute |
安全帽檢測 | CVI_TDL_YOLOV8_Hardhat |
手勢分類 | CVI_TDL_HandClassification |
手部檢測 | CVI_TDL_Hand_Detection |
手部關(guān)鍵點偵測 | CVI_TDL_HandKeypoint |
關(guān)鍵點手勢分類 | CVI_TDL_HandKeypointClassification |
行人檢測 | CVI_TDL_MobileDetV2_Pedestrian |
貓狗及人形檢測 | CVI_TDL_MobileDetV2_Person_Pets |
手部、臉及人形檢測 | CVI_TDL_HandFacePerson_Detection |
人形及交通工具檢測 | CVI_TDL_PersonVehicle_Detection |
人體關(guān)鍵點檢測 | CVI_TDL_Yolov8_Pose |
車牌檢測 | CVI_TDL_LicensePlateDetection |
車牌識別 | CVI_TDL_LicensePlateRecognition_CN |
語義分割 | CVI_TDL_Topformer |
交通工具檢測 | CVI_TDL_MobileDetV2_Vehicle |
客流統(tǒng)計 | CVI_TDL_APP_ConsumerCounting_Run |
人臉抓拍 | CVI_TDL_APP_FaceCapture_Run |
聲音識別(嬰兒哭聲、語音指令) | CVI_TDL_SoundClassification_V2 |
上述的常用算法,大部分都已經(jīng)在實際項目中落地,并且也在不斷優(yōu)化迭代,開發(fā)者可以自行體驗。除了支持已有的模型,TDL SDK還開放同類模型的推理接口,開發(fā)者只要參考文檔導(dǎo)出模型,可以直接用現(xiàn)有接口推理其自有模型。不僅如此,除了列表里提到的部分,TDL SDK還在不斷維護升級,以支持更多的主流算法。
四、結(jié)語
TDL SDK就像是打開智能世界大門的鑰匙,可以幫助開發(fā)者快速上手智能開發(fā)。如果你正在尋找一款高效、易用的智能開發(fā)工具包,那么TDL SDK絕對是你的不二之選!期待與你一起開拓智能化的新時代!
-
編程接口
+關(guān)注
關(guān)注
1文章
38瀏覽量
8127 -
SDK
+關(guān)注
關(guān)注
3文章
1068瀏覽量
47828 -
開發(fā)包
+關(guān)注
關(guān)注
0文章
24瀏覽量
7361
發(fā)布評論請先 登錄
晶豐明源推出Smart DrMOS智能集成功率器件芯片

華為助力湖北移動打造“九州”算力互聯(lián)網(wǎng)區(qū)域標桿
中科曙光發(fā)布DeepAI深算智能引擎
達實智能助力打造綠色智慧算力中心
智能體驗一觸即發(fā) —— 涂鴉App SDK 5系列,打造您的專屬智能應(yīng)用!


算智算中心的算力如何衡量?

開源芯片系列講座第24期:基于SRAM存算的高效計算架構(gòu)

通過Google Play SDK管理中心打造更安全、更優(yōu)質(zhì)的SDK

用智能DAC揭開醫(yī)療報警設(shè)計的神秘面紗

智能照明控制模塊:打造舒適、高效的光環(huán)境
算力系列基礎(chǔ)篇——算力與計算機性能:解鎖超能力的神秘力量!

在 Duo S 上使用 TDL SDK 高效部署人臉檢測模型

評論