本文档记录 DamoYOLO TinyNAS 人体检测模型在昇腾 NPU (Ascend 910B3) 上的迁移适配与精度验证结果。
与 cv_tinynas_head-detection_damoyolo(人头检测)共享相同的 DamoYOLO + TinyNAS 架构,差异在于训练数据——本模型针对全身人体进行训练。适配方式完全复用。
相关获取地址:
| 组件 | 版本 |
|---|---|
CANN | 8.5.1 |
torch | 2.8.0 |
torch_npu | 2.8.0.post4 |
modelscope | 1.37.0 |
timm | latest |
Ascend 910B3,单卡推理pip install torch==2.8.0 torch_npu==2.8.0.post4 modelscope opencv-python-headless numpy timm thop easydict matplotlib
modelscope download --model damo/cv_tinynas_human-detection_damoyolo --local_dir ./model
python inference.py --image scene.jpg --device npu[1, 128, 80, 80]| 指标 | CPU | NPU (Ascend 910B3) |
|---|---|---|
| 平均耗时 (完整模型) | 4202 ms | 27 ms |
| 中位耗时 | 4199 ms | 28 ms |
| 加速比 | — | 156x |
| 指标 | 数值 |
|---|---|
| 特征图形状 | [1, 128, 80, 80] |
| 鲁棒平均相对误差 | 0.324% |
| 精度要求 | 平均相对误差 < 1% |
| 精度评测 | 通过 |
与同系列 DamoYOLO 模型(人头检测、安全帽检测、无人机检测)共享相同架构,适配代码完全复用,仅需修改 --model_dir 参数。