海思hi3516dv300芯片AI圖像模塊學(xué)習(xí)(十)海思SDK學(xué)習(xí)—SVP部分源碼
nnie目前包含的例子


一、cnn模型
1.配置輸入模型和網(wǎng)絡(luò)文件、配置參數(shù)

2.初始化
加載模型

重置系統(tǒng)和內(nèi)存


分配內(nèi)存

3.根據(jù)wk文件設(shè)置軟硬件參數(shù)

?4.保存模型運(yùn)行中間態(tài)會(huì)用到的內(nèi)存
stTskBuf 當(dāng)不再有任務(wù)使用時(shí)才能被釋放。



5.把RBG數(shù)據(jù)放進(jìn)模型
stInputDataIdx

?6..開始推理,分段推理?
stProcSegIdx 分段?//CNN不用分段配置0即可

7.獲得結(jié)果

cnn網(wǎng)絡(luò)結(jié)束,cnn輸出最終版結(jié)果不方便調(diào)試
segnet存在
SAMPLE_SVP_NNIE_PrintReportResult

SAMPLE_SVP_NNIE_PrintReportResult
函數(shù)可以把帶有report標(biāo)記層的結(jié)果打印
cnn固定輸入
rfcn含有從vpss中實(shí)時(shí)取幀的功能
配置vio


配置網(wǎng)絡(luò)參數(shù)

rcfn全流程
SAMPLE_SVP_NNIE_Rfcn_ViToVo:vi輸入、vpss取幀、nnie推理、vo接收并畫框、out輸出




rfcn要分段第一段??
u32SegIdx = 0; ---》SAMPLE_SVP_NNIE_Forward

第二段 不支持層實(shí)現(xiàn)

第三段
?//第二段輸出含有roi信息,調(diào)用SAMPLE_SVP_NNIE_ForwardWithBbox開始第二段推理
stProcSegIdx.u32SegIdx = 1?;——-》SAMPLE_SVP_NNIE_ForwardWithBbox

第四段

最后獲得結(jié)果

從網(wǎng)絡(luò)結(jié)果中拿出前幾的坐標(biāo)

根據(jù)輸出網(wǎng)絡(luò)坐標(biāo)在vo中調(diào)用isp的vgs開始畫框

yolo?v3與rfcn不同,不支持層在最后的YOLO層。
yolo?v3?例子直接輸入圖片,不需要vpss取幀,流程是初始化系統(tǒng)內(nèi)存加載模型填充數(shù)據(jù)推理自定義層


rfcn是初始化配置vio,配置nnie導(dǎo)入wk模型,配置軟硬件。從vpss取幀