ModelArts与Atlas 200 DK的云端协同——对视频中的进行行人检测和跟踪Demo(附代码)

举报
Tianyi_Li 发表于 2020/05/15 20:46:18 2020/05/15
【摘要】 基于Atlas 200 DK ,对视频中的进行行人检测和跟踪Demo,使用YOLOv3_ResNet进行行人检测,Sort算法对检测到的行人做跟踪,同时加入了行人人数统计,可统计当前画面行人数量(如视频中左上角第一行所示),自视频开始时的总人数(如视频中左上角第三行所示),以及通过画面中黄线的行人数量(如视频中左上角第二行所示),还可以简单描绘行人轨迹。

 首先给大家看一下最终展示效果:


                                         

     

知道大家最想看到的是代码吧,直接上代码链接:

链接:https://pan.baidu.com/s/143uZO0B3-qCXPVKoA1CwCQ 

提取码:ajwp


 PS:这一版代码很粗糙,有很多地方是当时为了调试而加入的代码或测试视频之类的,并没有做整理,非常抱歉,拖延症晚期,而且Atlas 200 DK开发板同学在用,我也无法进行调试整理,再次致歉,同时需要注意的是,这版代码的结果和上面视频有些出入,比如进行人数统计的时候,黄线会错位,可能看不见,这需要在代码中调整,我后期会整理的。但经过测试,正常运行是没问题的。不过需要安装scipy和filterpy。scipy安装方法可参考帖子:https://bbs.huaweicloud.cn/forum/thread-65980-1-1.html   安装filterpy的方法类似。


项目部署方法基本和官方案例头部姿势案例一致,不过最终是生成.avi视频和图片作为结果,不是推流到PC端展示哦。官方头部姿势检测案例的部署教程链接:https://gitee.com/Atlas200DK/sample-README/tree/master/sample-headposeestimation-python


这里特别感谢昵称为hiCrystal的同学,感谢她在这个案例中所做的测试和调试工作,非常感谢。

如果大家对这个案例有什么问题,或者运行出现问题,可以私信我哦,我会尽量和大家一起解决,但我能力有限,希望大家海涵。



        基于Atlas 200 DK 行人检测和跟踪Demo——ModelArts与Atlas 200 DK的云端协同实现,从创建数据集到模型训练,再到得到Ascend 310推理使用的.om模型都可在ModelArts上完成,得益于ModelArts提供的强大算力,可以让个人开发者很轻松地完成训练任务。同时,ModelArts也提供了很多很好的预置算法和AI市场中官方发布的算法,涵盖目前主流方向,大家有机会可以试试,免费订阅哦。

        这里使用自己基于公开数据集VOC 2007制作的行人数据集,由2473张图片组成,11592个行人框标注(数据集有点小,后期会考虑使用更大的数据集训练,提高性能),使用AI市场的yolov3-resnet18算法,在Ascend 910上训练了约5个小时,下载到本地,使用MindStudio转为.om模型,具体模型转换的细节主要是:

先要说明的是模型要求输入为RGB三通道的图像。大体要注意一下细节:

1. 色域转换(可选)。色域转换是为了得到模型要求的RGB格式,这一步可以在模型转换中用AIPP完成,也可以直接跳过,在代码中完成。

2. 归一化操作。即每个像素除以255,这里是乘以1/255,约为0.0039216,来完成归一化操作。

而在得到模型推理结果后,对推理结果进行解析的代码,则主要参考了AI市场中yolov3-resnet18算法的线上部署推理代码,实现行人检测。而其后面的行人跟踪,则是基于GitHub开源项目,使用Sort算法对检测到的行人做跟踪,同时加入了行人人数统计,可统计当前画面行人数量(如视频中左上角第一行所示),自视频开始时的总人数(如视频中左上角第三行所示),以及通过画面中黄线的行人数量(如视频中左上角第二行所示),还可以简单描绘行人轨迹。

        整体来说,这个Demo比较粗糙,主要是基于官方案例头部姿势识别做的修改,还有很多改进空间,希望大神不要见怪,我会尽力改进的,谢谢。目前代码有些乱,后期整理后放在GitHub和码云上,供大家参考,期待大家来watch、star、fork是的,是的,是的,我已经有好几个to do list了,可是最近有点忙,直接上传到GitHub和码云有不太行,因为代码太不优雅了,我无法忍受给别人看我不优雅的代码,我会尽快整理好,上传的,一定,一定,会在本帖或者再发帖子给出代码的,欢迎各位Atlas 200 DK或者HiLens小伙伴尝试呀,因为Atlas 200 DK或者HiLens都是基于Ascend 310的主打推理的设备,应该都可以的。



【版权声明】本文为华为云社区用户原创内容,转载时必须标注文章的来源(华为云社区)、文章链接、文章作者等基本信息, 否则作者和本社区有权追究责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@huaweicloud.com
  • 点赞
  • 收藏
  • 关注作者

评论(0

0/1000
抱歉,系统识别当前为高风险访问,暂不支持该操作

全部回复

上滑加载中

设置昵称

在此一键设置昵称,即可参与社区互动!

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。