1200字范文,内容丰富有趣,写作的好帮手!
1200字范文 > 利用Vitis开发基于ZCU106的神经网络加速器(二)——DPU编译及Demo

利用Vitis开发基于ZCU106的神经网络加速器(二)——DPU编译及Demo

时间:2019-04-27 23:00:11

相关推荐

利用Vitis开发基于ZCU106的神经网络加速器(二)——DPU编译及Demo

前言

上一篇文章讲到了如何生成可供Vitis使用的XRT platform。这次我们使用生成的platform来开发一个完整的加速器Demo并在ZCU06上跑通。

DPU概述

DPU是Vitis AI中官方提供的FPGA神经网络加速器,其架构如下图:

DP(Deep Learning Processor Unit)是专用于卷积神经网络的可配置计算引擎。 引擎中使用的并行度是设计参数和应用程序。 它包含一组高度优化的指令,并支持大多数卷积神经网络,例如VGG,ResNet,GoogleNet,YOLO,SSD,MobileNet,FPN等。详情可见Github。

编译安装过程

环境要求:

除与上一篇文章中相同的环境要求外,还需要下载xilinx-zynqmp-common-v.1硬件平台。

下载完成后,解压相关内容,正式进入编译过程。

tar xzf xilinx-zynqmp-common-v.1.tar.gzgunzip xilinx-zynqmp-common-v.1/rootfs.ext4.gz

首先下载Vitis AI到你的工作目录,建议下载适配你的Vitis和petalinux版本的release版本,我下载的是Vitis AI 1.2.1,master分支下的代码仍处于开发中,使用可能会有未知的问题。

下载完成后,进入DPU-PRD子文件夹,目前我们仅会用到此文件夹下的内容。

cd <Vitis_AI_HOME>/DPU-TRD

可以看到有三个文件夹,app是我们的Demo程序Restnet50,dpu_ip包含dpu的ip核,而prj则是我们要编译的工程目录。

prj目录中有Vitis和Vivado两个文件夹,使用不同工具开发的开发者可以选择进入不同的文件夹。我们进入Vitis

cd prj/Vitis

修改Makefile

先进行备份

cp Makefile Makefile.bak

修改Makefile,将XOCC_OPTS的–config修改为prj_config_104_2dpu,ZCU106使用的FPGA芯片与ZCU104相同,因此我们使用ZCU104的配置。

#XOCC_OPTS = -t ${TARGET} --platform ${SDX_PLATFORM} --save-temps --config ${DIR_PRJ}/config_file/prj_config --xp param:compiler.userPostSysLinkOverlayTcl=${DIR_PRJ}/syslink/strip_interconnects.tcl XOCC_OPTS = -t ${TARGET} --platform ${SDX_PLATFORM} --save-temps --config ${DIR_PRJ}/config_file/prj_config_104_2dpu --xp param:compiler.userPostSysLinkOverlayTcl=${DIR_PRJ}/syslink/strip_interconnects.tcl

初始化编译环境:

export TRD_HOME =<Vitis AI path>/DPU-TRDsource <vitis install path>/Vitis/.1/settings64.shsource opt/xilinx/xrt/setup.shexport EDGE_COMMON_SW=<mpsoc common system>/xilinx-zynqmp-common-v.1 export SDX_PLATFORM=<zcu106 base platform path>/xilinx_zcu106_base_20_1/xilinx_zcu106_base_20_1.xpfm

其中,EDGE_COMMON_SW为之前下载的xilinx-zynqmp-common-v.1路径,SDX_PLATFORM为上一篇文章中生成的platform路径。

编译

make KERNEL=DPU_SM DEVICE=zcu106

这个过程会比较漫长,我在自己的机器上跑了3小时左右。编译成功后,生成的文件放置到$TRD_HOME/prj/Vitis/binary_container_1/sd_card目录下。

烧录SD卡。

这里要用到一个软件,balena etcher,下载地址点击这里,linux版本下载完成后无需安装,解压后即可使用。这个软件可以直接烧录binary_container_1目录下的sd_card.img镜像文件至SD卡中,该镜像文件中包含了所有我们需要的数据。

软件使用方式非常简单,我们选择sd_card.img,下一步选择要烧录的SD卡(当然此时SD卡需要连接到电脑),再点击flash即可。

软件会自动将SD卡分区,分为一个boot区,包含系统启动所需文件,以及rootfs文件系统,即系统启动后我们的/目录。

烧录好后我们将TRD_HOME/app文件夹拷贝到rootfs分区的home/root下,系统启动后运行测试。

启动系统。

我们的SD卡烧录好了,接下来就是启动ZCU106并进行调试。我们将板子的启动模式调整为从SD卡启动,对应的板子上的开关为off,off,off,on。插入SD卡,先不打开电源,接下来我们电脑打开串口调试工具,来观察板子的启动情况。

板子的UART接口通过USB UART转接线连接到ubutnu,ubuntu打开minicom,如果没有minicom使用sudo apt install minicom进行安装。

我们将minicom的配置调整为波特率115200,8数据位,1校检位,端口一般为/dev/ttyUSB0(如果不是需要自己用lsusb命令等找到板子连接的USB口)。注意将硬件控制和软件控制均设置为no,否则可能出现无法在minicom中输入命令的情况。

我们打开板子的电源,如果没有问题的话,你将会看到Xilinx ZYNQ MP First Stage Boot Loader的字样,然后进行启动相关信息的打印,当最终看到root@zynqmp-commonn-_1的字样,那么恭喜,说明系统启动成功了。

我们为板子插上网线,minicom中输入ifconfig,得到板子的ip地址,以后就可以在电脑上通过ssh来登录板子了。当然这需要我们的板子和ubuntu在一个局域网中,或者能互相ping通。

测试resnet50

我们先前已经将app文件copy到root目录下,我们cd进入app文件夹。依次执行:

tar xzf resnet50.tar.gzenv LD_LIBRARY_PATH=samples/lib XLNX_VART_FIRMWARE=/media/sd-mmcblk0p1/dpu.xclbin samples/bin/resnet50 img/bellpeppe-994958.JPEG

看到:

score[945] = 0.992235text: bell pepper,score[941] = 0.00315807 text: acorn squash,score[943] = 0.00191546 text: cucumber, cuke,score[939] = 0.000904801 text: zucchini, courgette,score[949] = 0.00054879 text: strawberry,

说明DPU运行正常。

不过我个人这里遇到了问题,程序能够正常运行,但是分类结果不对,我的结果是:

score[816] = 0.00531614 text: spindle,score[723] = 0.00414021 text: pinwheel,score[612] = 0.00414021 text: jinrikisha, ricksha, rickshaw,score[592] = 0.00414021 text: hard disc, hard disk, fixed disk,score[369] = 0.00414021 text: siamang, Hylobates syndactylus, Symphalangus syndactylus,

至今这个问题还没解决,有运行得到正常结果的同学可以给我评论,我们交流一下,万分感谢。

遇到的问题

我遇到的问题大部分都是XX文件找不到,这时候需要我们自己找到该文件,然后在Makefile中修改该文件的位置,一般都能解决问题。

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。