spnasnet_100_rmsp_in1k 模型的昇腾 NPU 适配版本。
该模型源自 timm 库中的 spnasnet_100_rmsp_in1k,已适配至昇腾 NPU 平台。
# NPU 推理
python3 inference.py --device npu
# CPU 推理
python3 inference.py --device cpuapache-2.0
| 对比项 | CPU | NPU | 误差 |
|---|---|---|---|
| 余弦相似度 | 1.0 | 0.9998812 | 0.0466% |
| 最大绝对误差 | - | - | 0.004552 |
| 平均绝对误差 | - | - | 0.001573 |
| Top-1 一致 | - | - | 是 |
结论: CPU 与 NPU 精度误差为 0.0466%,小于 1% 精度要求,Top-1 分类结果完全一致。
本仓库提供完整的推理脚本,支持 CPU 和 NPU 双平台推理。
# NPU 推理
python3 inference.py --device npu
# CPU 推理
python3 inference.py --device cpuModel loaded successfully on NPU
Input shape: (1, 3, 224, 224)
Inference completed
Results saved