基于HiLens Studio開發行人檢測與跟蹤應用
986
2025-04-01
目前,HiLens Kit的AI芯片支持運行“.om”模型,“.om”模型可以通過Tensorflow或Caffe模型轉換而來。在HiLens Studio開發技能時,支持把模型導入進去,針對Tensorflow或Caffe模型,HiLens Studio平臺也支持將模型轉換成“.om”模型。
如果您是在ModelArts開發的模型,HiLens Studio支持一鍵導入ModelArts模型,并對非“om”格式的模型進行轉換,十分方便快捷。
ModelArts開發模型
首先需要在ModelArts在線訓練算法模型,詳情可參考《ModelArts文檔》。
如果要使用ModelArts的預置算法,當前華為HiLens平臺僅支持轉換如下預置算法:
yolov3_resnet18(檢測物體類別和位置)
ResNet_v1_50(圖像分類)
華為HiLens暫不支持導入ModelArts中“自動學習”訓練的模型。只能導入ModelArts中訓練的模型文件,不能導入ModelArts的模型。
導入(轉換)模型
在HiLens Studio界面左側,單擊。
頁面左側將顯示您在ModelArts訓練好的模型列表。
選擇待導入、轉換的模型,單擊“Operation”列的“Apply”。
如果是“om”格式的模型,待模型導入成功后,HiLens Studio右下角會提示“Model imported successfully.”,文件夾“model”下會顯示新導入的模型文件。
如果是非“om”格式的模型,會彈出“Model Convertion”對話框。
按表 Model Convertion參數說明填寫模型轉換的信息,單擊“OK”。
參數
說明
Model Path
待轉換的模型文件在技能項目文件中的位置。一般將模型導入至文件夾“save_model”。
Configuration
待轉換模型的配置文件在技能項目文件中的位置。例如“save_model/aipp_rgb.cfg”。
Output Path
模型轉換后輸出位置。
Type
模型轉換的類型,包括“TF-FrozenGraph-To-Ascend-HiLens”、“Caffe to Ascend”。
“TF-FrozenGraph-To-Ascend-HiLens”
支持將Tensorflow frozen graph模型轉換成可在ascend芯片上運行的模型。
“Caffe to Ascend”
支持將Caffe模型轉換成可在ascend芯片上運行的模型。
Advanced Options
當模型轉換類型為“TF-FrozenGraph-To-Ascend-HiLens”時,可填寫高級選項,包括張量形狀、轉換輸出節點等參數選項,詳情請見表 Advanced Options。
參數名稱
參數說明
Input Tensor Shape
輸入張量形狀。若在上文“模型來源”中,選擇需要轉換格式的模型(非om格式模型),并且轉換類型是“Tensorflow frozen graph 轉 Ascend”或“Tensorflow SavedModel 轉 Ascend”時,需要填寫輸入張量形狀。
張量形狀即模型輸入數據的shape,輸入數據格式為NHWC,如“input_name:1,224,224,3”,必填項?!癷nput_name”必須是轉換前的網絡模型中的節點名稱。當模型存在動態shape輸入時必須提供。例如“input_name1:?,h,w,c”,該參數必填,其中“?”為batch數,表示1次處理的圖片數量,需要根據實際情況填寫,用于將動態shape的原始模型轉換為固定shape的離線模型。
如果存在多個輸入,請以分號(;)隔開。
out_nodes
轉換輸出節點,即指定輸出節點,例如“node_name1:0;node_name1:1;node_name2:0”,其中“node_name”必須是模型轉換前的網絡模型中的節點名稱,冒號后的數字表示第幾個輸出,例如“node_name1:0”,表示節點名稱為“node_name1”的第0個輸出。
input_format
輸入數據格式,默認是“NHWC”,如果實際是“NCHW”的話,需要通過此參數指定“NCHW”。
net_format
優選數據格式,即指定網絡算子優先選用的數據格式,“ND(N=4)”和“5D”。僅在網絡中算子的輸入數據同時支持“ND”和“5D”兩種格式時,指定該參數才生效?!癗D”表示模型中算子按“NCHW”轉換成通用格式,“5D”表示模型中算子按華為自研的5維轉換成華為格式?!?D”為默認值。
fp16_high_precsion
生成高精度模型,指定是否生成高精度“FP16 Davinci”模型。
0為默認值,表示生成普通“FP16 Davinci”模型,推理性能更好。
1表示生成高精度“FP16 Davinci”模型,推理精度更好。
output_type
網絡輸出數據類型,“FP32”為默認值,推薦分類網絡、檢測網絡使用;圖像超分辨率網絡,推薦使用“UINT8”,推理性能更好。
更多操作
如果您的模型是在本地開發的,HiLens Studio也能支持導入和轉換模型。
詳情請參考導入/轉換本地開發模型。
還等什么,趕快來HiLens Studio大展身手吧!
AI開發平臺ModelArts EI智能體 華為HiLens
版權聲明:本文內容由網絡用戶投稿,版權歸原作者所有,本站不擁有其著作權,亦不承擔相應法律責任。如果您發現本站中有涉嫌抄襲或描述失實的內容,請聯系我們jiasou666@gmail.com 處理,核實后本網站將在24小時內刪除侵權內容。
版權聲明:本文內容由網絡用戶投稿,版權歸原作者所有,本站不擁有其著作權,亦不承擔相應法律責任。如果您發現本站中有涉嫌抄襲或描述失實的內容,請聯系我們jiasou666@gmail.com 處理,核實后本網站將在24小時內刪除侵權內容。