将 timm/swin_tiny_patch4_window7_224.ms_in22k_ft_in1k 图片分类模型适配到单卡昇腾 NPU(Ascend910)。
timm.create_model(pretrained=False) + ModelScope snapshot_download 本地权重# 安装依赖
pip install -r requirements.txt
# 运行推理
python inference.py推理结果(单张测试图片):
python eval_accuracy.py对单张测试图片进行 CPU 与 NPU 一致性验证:
| 指标 | 数值 |
|---|---|
| max_abs_error | 0.007288 |
| mean_abs_error | 0.001283 |
| relative_error | 0.2207% |
| cosine_similarity | 0.999999 |
| threshold | 1.0% |
| 结果 | PASS |
| 指标 | 值 |
|---|---|
| avg | 9.24 ms |
| min | 9.16 ms |
| max | 9.35 ms |
| p50 | 9.22 ms |
| p90 | 9.34 ms |
| p95 | 9.34 ms |
| 吞吐量 | 108.27 img/s |
Smoke consistency check(非官方精度评测),详见 logs/accuracy.log。
详见 screenshots/self_verification.png
#NPU #Ascend #image-classification #timm #swin-transformer