华为海思AI芯片HI3516DV300上SSD代码的移植和优化
前一段时间一直在做海思AI芯片 HI3516DV300上SSD代码的优化工作,这里总结一下优化的一些心得体会。之所以要对海思AI芯片 HI3516DV300上SSD代码做优化,是因为SSD有一部分是运行在CPU上,而HI3516DV300上的CPU性能比较差,网络只要稍微复杂一点,CPU部分运行就会很慢。下表给出了一些模型在芯片上运行的时间,可以看到CPU部分运行的时间占据了大部分。
注意:
这里的Forward对应SAMPLE_SVP_NNIE_Forward()函数,这个函数基本上是运行在NNIE上的,GetResult对应SAMPLE_SVP_NNIE_Ssd_GetResult函数,这个函数是运行在CPU上的,后面直接使用GetResult表示这个函数
文章目录
- DetectionOutForward的优化
- NMS的优化
- Softmax的优化
- 结束语
DetectionOutForward的优化
首先分析性能瓶颈。通过对GetResult函数分析发现主要耗时在两个函数中:SVP_NNIE_Ssd_SoftmaxForward和SVP_NNIE_Ssd_DetectionOutForward,后面直接使用SoftmaxForward和DetectionOutForward表示。
进一步对DetectionOutForward函数进行分析,其中DetectionOutForward主要包含了两个步骤,这两个部分分别对应了两个for循环:
- decode box: 将网络输出的预测值转换为实际的坐标值,这一部分记为Loop1
- NMS: 对每一类目标的前topk个置信度最高的bbox做NMS,这一部分记为Loop2
发现主要耗时在Loop1中
Loop1对应了如下代码:
分析Loop1代码,Loop1主要功能是对所有anchor的预测值进行解码转换为anchor在图像中实际的坐标值。
对每一个anchor都需要调用两次exp()函数(上图中红框标出的部分),这个函数是一个非常耗时的函数,45120个anchor一共调用90240次exp(),通过测试发现90240次exp()调用需要耗时21ms,如果能够减少exp()的调用次数,将会大大减少时间。
由于实际做NMS的时候,我们只对前topk个置信度最高的bbox做NMS,也就是说只会用到前topk个bbox的真实坐标值,其他bbox的坐标值用不到,所以我们只需要计算出前topk个bbox的真实坐标值就可以了,这里topK设置为400,也就是说只需要执行800次exp操作就可以,而800次exp操作只需要0.2ms,可以大大减少时间。
但是最后做NMS的时候,使用的是图1中蓝框中的4个变量,而如果要计算出真实坐标值,就必须要知道网络输出的预测值和priorbox的坐标,也就是如果要计算出前topk个bbox的真实坐标值,就必须要知道每个bbox对应的8个变量,图1中红框和绿框中的8个值。
图1
8个值:
ps32PriorBoxes[jSAMPLE_SVP_NNIE_COORDI_NUM]
ps32PriorBoxes[jSAMPLE_SVP_NNIE_COORDI_NUM+1]
ps32PriorBoxes[jSAMPLE_SVP_NNIE_COORDI_NUM+2]
ps32PriorBoxes[jSAMPLE_SVP_NNIE_COORDI_NUM+3]
ps32LocPreds[jSAMPLE_SVP_NNIE_COORDI_NUM]
ps32LocPreds[jSAMPLE_SVP_NNIE_COORDI_NUM+1]
ps32LocPreds[jSAMPLE_SVP_NNIE_COORDI_NUM+2]
ps32LocPreds[jSAMPLE_SVP_NNIE_COORDI_NUM+3]
下面的问题就是如何将每个bbox对应的这8个变量编码到4个变量中,通过对每个bbox对应的8个变量分析发现:
- 由于网络输入大小是512,也就是图像中的坐标在[0,512]之间,也就是ps32PriorBoxes的值也在[0,512]之间,也就表示ps32PriorBoxes的值可以通过3位数来表示
- ps32LocPreds的值有正有负,而且值的范围可以使用5位数表示
基于上面的发现,设计了以下编码方式:
ps32LocPreds * 10000+ps32PriorBoxes * 10+flag
其中flag为标记位,当ps32LocPreds值为负数时,flag为0,当ps32LocPreds值为正数时,flag为1
而且编码之后的值全部转换为正数,便于解码。
这个编码的含义就是:最后一位是标记位,倒数第2到第4位是priorbox的值,剩下的位数是位置预测值
示例:
-3852,125→38521250
5017,96→50170961
-13590,145→135901450
最后,在NMS的时候,进行解码就可以计算出真实坐标值了,比如对每个bbox编码之后的4个值的第一个值进行解码:
这样就可以得到需要的8个值,最后根据这8个值就可以计算出该bbox的真实坐标值了
使用优化后的代码进行速度测试
行人检测器Loop1经过优化后从原来的34ms提高到了7ms,速度得到了明显提高,由于Loop2增加了解码的时间,所以Loop2会略微变慢,但是总体时间还是大大降低了。
优化之后对精度是完全没有影响的。
NMS的优化
AI芯片中官方给的SSD代码在对每一类做NMS的时候,实现上有点bug,官方代码中,当类别为背景的时候,也做了NMS。
当类别为背景的时候,是不需要对所有的预测框做NMS的,这样可以节省很多时间。
修改后的代码:
优化后的时间:
我们可以看到优化后Loop2的时间大大降低,在CPU上运行的时间基本上降低为原来的一半了。
Softmax的优化
上面分析了GetResult函数主要耗时在两个函数中:SoftmaxForward和DetectionOutForward。上面对DetectionOutForward进行了优化,那么SoftmaxForward是否可以优化呢?
分析了SoftmaxForward函数,发现计算所有anchor的置信度的时候,45120个anchor也需要调用90240次exp(),需要耗时21ms。
我们知道softmax概率的计算公式如下:
在对softmax进行优化的时候,我做了一个实验,直接使用fi表示softmax置信度,然后在多个模型中做了测试:
数据集A:
数据集B:
我发现一个很奇怪的现象:直接使用fi代替softmax概率,大部分模型精度基本上没有损失。同时也观察了一系列模型的结果,发现softmax概率大的目标往往fi值也大。 目前还不知道具体原因。有知道其中原因的朋友欢迎留言讨论。
由于去掉了softmax的计算,所以时间会大大减少,上表中可以看到去掉了softmax计算后,从原来的77ms直接降低到了18ms,速度提高了4倍多。但是由于目前对这种现象的原因还不明确,所以暂时是没有使用的。
结束语
通过本次优化,深刻体会到了算法性能的重要性,PC上很快的一个操作,在其他平台很可能就是性能的瓶颈。同时也体会到了深入理解算法的重要性以及将速度做到极致的那种成就感。
非常感谢您的阅读,如果您觉得这篇文章对您有帮助,欢迎扫码进行赞赏。
华为海思AI芯片HI3516DV300上SSD代码的移植和优化相关推荐
- 海思AI芯片(Hi3519A/3559A)方案学习(十七)开发板上运行yolo3模型的代码分析
前言 前面的博客系列 已经介绍了如何将caffemodel转换成wk文件,如何将jpg文件转成bgr格式数据以及如何在PC上仿真模型推理等,基于这些基础,本文来结合代码分析如何在板子上推理yolov3 ...
- 海康摄像头 海思AI芯片(Hi35XX): 图像jpg转.bgr
前言 把一张jpg图片转成bgr(注意顺序是bgr)二进制文件,最近在弄华为Hi3516DV300开发板上的一些深度学习的样例程序,换言之,35xx nnie SDK只支持BGR以及yuv格式的数据输 ...
- 润和软件:公司深度参与了华为海思系列芯片研发
[TechWeb]6月11日,今日润和软件在互动平台与投资人互动时表示,华为是润和HiHope最重要的生态合作伙伴,双方已联合发布了HiKey960.HiKey970.Poplar.HiHope-Hi ...
- 海思AI芯片(Hi3519A/3559A)方案学习(十四)JPEG图片转换成bgr文件
原文:https://blog.csdn.net/avideointerfaces/article/details/89931156 前言 在系列文章海思AI芯片(Hi3519A/3559A)方案学习 ...
- 海思AI芯片(Hi3519A/3559A)方案学习(一)资料以及术语介绍
芯片系列介绍 海思AI芯片Hi35xx系列具体包括的型号Hi3559A V100, Hi3559C V100, Hi3519A V100, Hi3516C V500以及Hi3516D V300.下面要 ...
- arm 升腾310_简单说说华为海思的芯片产品线
很多人对于海思芯片有一定的了解,但是对于海思芯片产品线究竟有哪些场景缺乏整体的框架和理解,小编就简单介绍下华为海思芯片的产品线有哪些: 1.手机SoC芯片 麒麟系列:麒麟910 .麒麟920 .麒麟9 ...
- 高清 网络摄像机 华为海思3510芯片
给大家介绍一款,采用海思3510芯片 的网络摄像机 IP camera. 海思半导体有限公司成立于2004年10月,前身是创建于1991年的华为集成电路设计中心.海思公司总部位于深圳,在北京.上海.美 ...
- 华为海思Hi1620芯片发布在即 7nm制程ARM架构最高可达3.0GHz
一夜间,华为海思"备胎转正",不仅周边的"备胎股"受到广泛关注.5月23日,普华基础软件在公众号透露"海思hi1620系列即将发布"的消息. ...
- 海思AI芯片(Hi35XX): 图像jpg转.bgr
前言 把一张jpg图片转成bgr(注意顺序是bgr)二进制文件,最近在弄华为Hi3516DV300开发板上的一些深度学习的样例程序,换言之,35xx nnie SDK只支持BGR以及yuv格式的数据输 ...
最新文章
- (C#)如何利用Graphics画出一幅图表
- 8086汇编学习小记-王爽汇编语言实验12
- c#sdf数据库连接_如何连接并处理 sdf 数据库文件(便捷数据库处理)
- Hadoop pipes编程
- java双机调度_Haproxy+keepalive-群集架构实验
- springboot多数据源动态数据源(主从)
- 计算机hub体系部件,原来如此!USB Hub接口为啥都是4个7个或10个?
- mybatis-plus乐观锁
- git 怎么拉取线上代码到本地进行合并_android studio如何使用git提交、拉取、合并代码的操作...
- 集体智慧编程——搜索与排名
- pip 安装第三方库时各种报错It is a distutils installed project and thus we cannot accurately determine
- R语言--Cox模型校准曲线原理(二)算法
- Win10 锁屏之后无法唤起 出现假死解决办法
- 入门Web前端开发需要学习哪些技术?薪资高吗?
- 香港科技大学陈浩教授招收AI全奖博士
- 【愚公系列】2022年01月 Java教学课程 80-软件开发介绍
- 关于Ajax请求问题的总结
- make menuconfig问题
- 《信息简史》第七章人物、事件、名言收集小感
- 面向企业的终极物联网实施指南
热门文章
- C语言用switch结构编程完成:输入月份数字,输出对应的季节,12-2月是冬季,3-5月是春季,6-8月是夏季,9-11月是秋季。
- error: reference to 'max' is ambiguous
- 哈工大《信息内容安全》课程知识要点和难点
- MATLAB中的eps的详细解释及用法
- 微处理器(Microprocessor)与微控制器(Microcontroller) 的区别
- 大厂的优惠券系统是如何设计的?
- 哈佛管理大师学习笔记
- iOS强制关闭暗黑模式
- 刘奇:能否掌控复杂性,决定着分布式数据库的生死存亡
- html 判断iphonex,iPhoneX和iPhoneXS区别