耐能KL520 AI 芯片应用开发技能 ＜Part-009＞

        《耐能KL520的Python程式雕龍小技》

                     Part-009 
                By 神櫻 AI團隊 
                     2022/7/16

文章目录

• 说明
• 一、 范例
• 二、在 KL520 上运行(推论)
• 三、精进之路 

---

前言

由於在Pytorch 里執行時，輸出的格式，與KL520推理的輸出格式不同，所以要記得在host_lib 裡的App 必須Reshape 輸出的格式

一、 范例 

本範例輸入一張(3, 32, 32)的Png圖像。由這裡的AI模型進行卷積，然後將卷積得到的特徵表輸出。

Pytorch 程式碼： 

 

 

 

 這模型推論之後，輸出一個(1, 2, 4, 4)的陣列。 

如下： 

 將這個*.onnx檔案，經由耐能Toolchain進行優化轉換之後，就可以在KL520 Dongle 裡運行(推論)了。

二、 在 KL520 上運行(推論)

現在撰寫一支App 來引導模型進行推論。 

Python 程式碼： 

 

 

 

 

 

 

 此程式的模型推論，其結果輸出，如下：

其格式與在Pytorch裡輸出格式不同。 

其中的程式碼是：

 於是，就來改善它。 

三、 精進之路

只要增添一行指令即可。如下：

然後，在執行此程式，就輸出正確格式的資料了。如下：

 & 最后，感谢您阅读本篇文章。本篇文章摘自【AI学习杂志】。 

& 与神樱AI团队联络：misoo.tw@gmail.com。

                                         ◆◆◆

---

总结

更多资讯可私信，或者关注！

在个人主页，也可学习更多：

1、《关于NCS2+Openvino》： 、http://t.csdn.cn/452Tz

2、《建模_Unet_GAN》：http://t.csdn.cn/am37C

3、《彩绘GAN模型》：http://t.csdn.cn/OpfJ7

.............................................