概述
本应用以昇腾 Atlas 500 A2、Atlas 300I Pro、Atlas 300V Pro为主要的硬件平台,使用Vision SDK实现了姿态估计网络(PoseEstNet)推理,用于检测并预测车辆36个关键点坐标以及生成热点图。
功能模块介绍
本案例流程如下:
1. 车辆检测:检测出图像中的车辆对象,得到坐标框
2. 抠图缩放:将图像中的车辆按坐标框抠图,并缩放成PoseEstNe模型要求输入大小
3. PoseEstNet模型推理:推理车辆图像
4. 模型后处理:将推理结果解析为36个关键点坐标