最近几天跟着AI界的大V江大白体验了一把基于Yolov5框架实现人流统计,觉得现在实现AI真的是触手可及,至少之前没有这么觉得过。个人转行进入教育行业将近4年,在此期间虽然也自娱自乐搞一些AI项目,但是部署到现实场景的还是少之甚少,特别是这次用一部Android手机就能实现人流统计的部署让我发现了更多好玩的事情,也期待能做出更多有意义的事情。言归正传,今天就来一个粗略的复盘,因为最近比较忙,待过了这阵儿再挤时间详细梳理一下。
上面的截图是之后会梳理的部分,今天先说个大概:⼈体检测+⼈体追踪+线段统计的⽅式,针对视频实现⼈流量统计,并通过喵提醒告知⼈流统计的数量。
一、准备工作
框架:Yolov5
平台:Aidlux
Aidlux是什么?没听说过是不是,我也是第一次用。喏,介绍附在这里了。
AIdlux系统是基于ARM架构的跨生态(Android/鸿蒙+Linux)一站式AIOT应用开发平台。实际应用到现场的时候,通常会以几种形态:GPU服务器、嵌入式设 备(比如Android手机、人脸识别闸机等)、边缘设备。GPU服务器等,而Android 嵌入式设备的底层芯片,通常是ARM架构。而Linux底层也是ARM架构,并且Android又是 基于Linux内核开发的操作系统,两者可以共享Linux内核。因此就产生了从底层开发一套应用系统的方式,在此基础上同时带来原生Android和原生Linux使用体验。所以在基于Linux系统和ARM架构上,我们可以在安卓手机上直接下载Aidlux软件,使用安卓手机充当边缘设备,进行代码的调试与案例的实现。感兴趣的老铁可以移步这里探秘(https://community.aidlux.com/)哈。
今天我们的任务就是通过下面的简易图来实现。
工欲善其事必先利其器,我们需要做一下准备。
准备一台笔记本、一部Android手机,需要网络或者手机热点即可。
笔记本需要我们安装编程调试工具VSCode,官网下载安装即可;
安卓手机需要安装AidLux软件,打开应用市场搜索AidLux下载安装即可,注意我们的Android手机需要满足以下的条件:
●Android 版本≥6.0.1
●cpu 支持arm64-v8a架构
●手机存储空间充足
●1.0以上版本需要安装空间>600M
●初始化空间>1.7G
二、目标检测(yolov5⼈体检测模型)
PC端
Android手机端
在我们的模型中,设定标签为person,上图中绿色框圈出来的部分就是检测到的人,我们对检测结果标定了标签和置信度(检测结果是人的概率)。
代码比较长,这里就不贴了,需要源码的可以后台留言【人流统计】。
三、目标追踪(