在小藤上实现无人机人群密度估计
【摘要】 在小藤上实现无人机人群密度估计
登录开发板：
cd ~/ascend_community_projects/UAV_CrowdCounting
获取onnx模型：
cd models
wget https://mindx.sdk.obs.cn-north-4.myhuaweicloud.com/ascend_community_projects/UAV_CrowdCounting/UAV_CrowdCounting.onnx --no-check-certificate
开始模型转换，将onnx模型转为om模型：
source /usr/local/Ascend/ascend-toolkit/set_env.sh
source /home/HwHiAiUser/mxVision-5.0.RC2/set_env.sh
atc --model=./UAV_CrowdCounting.onnx --framework=5 --output=./uav_crowdcounting_norm --soc_version=Ascend310B1 --input_shape="input:1,3,512,640" --input_format=NCHW --insert_op_conf=./aipp.cfg
下载测试数据集
mkdir ~/ascend_community_projects/UAV_CrowdCounting/dataset
cd ~/ascend_community_projects/UAV_CrowdCounting/dataset
wget https://mindx.sdk.obs.myhuaweicloud.com/ascend_community_projects/UAV_CrowdCounting/VisDrone2021.zip --no-check-certificate
解压 unzip VisDrone2021.zip
随机取出一张图片，改名为test.jpg，复制到 ~/ascend_community_projects/UAV_CrowdCounting/data目录下
图片是这个样子的：
（张小白总算知道了为啥这个样例只能用于无人机）
cd ~/ascend_community_projects/UAV_CrowdCounting/
执行推理：
. run.sh
抱歉，忘记改名了：
预计有14个人。
结果文件为：vis_img.jpg
下载下来看看：
确实标记出来的好像是每个人的位置。
执行精度测试：
cd ~/ascend_community_projects/UAV_CrowdCounting/
. eval.sh
看来原因是我把文件放到了dataset目录，它却想到data目录下读文件。
把目录移动过去:
重新执行精度测试：
cd ~/ascend_community_projects/UAV_CrowdCounting/
. eval.sh
均方差MSE为17.97，不大于35，绝对误差MAE 10.20，不大于20，FPS为14.17,不大于20。满足精度要求。
（全文完，谢谢阅读）
【版权声明】本文为华为云社区用户原创内容，转载时必须标注文章的来源（华为云社区）、文章链接、文章作者等基本信息， 否则作者和本社区有权追究责任。如果您发现本社区中有涉嫌抄袭的内容，欢迎发送邮件进行举报，并提供相关证据，一经查实，本社区将立刻删除涉嫌侵权内容，举报邮箱：
cloudbbs@huaweicloud.com
- 点赞
- 收藏
- 关注作者
评论（0）