本项目将 timm/dpn131.mx_in1k 适配到华为昇腾 NPU (Ascend910B) 上运行。
使用 ModelScope snapshot_download 下载权重,timm.create_model(pretrained=False) 加载本地权重。
python inference.py推理结果示例(单张测试图片):
| Rank | Class | Probability |
|---|---|---|
| 1 | class_980 | 0.287 |
| 2 | class_812 | 0.1656 |
| 3 | class_974 | 0.0764 |
| 4 | class_657 | 0.0668 |
| 5 | class_820 | 0.0555 |
对单张测试图片进行 CPU 与 NPU 一致性验证:
| 指标 | 数值 |
|---|---|
| max_abs_error | 0.009371 |
| mean_abs_error | 0.001651 |
| relative_error | 0.1326% |
| cosine_similarity | 1.0 |
| threshold | 1.0% |
| 结果 | PASS |
| 指标 | 数值 |
|---|---|
| 平均延迟 | 30.62 ms |
| 吞吐量 | 32.66 images/sec |
本项目包含单图 smoke consistency 验证,非官方 ImageNet 完整验证集评测。
logs/inference.loglogs/accuracy.loglogs/benchmark.log| 项目 | 链接 |
|---|---|
| 原始权重模型地址 | https://modelscope.cn/models/timm/dpn131.mx_in1k |
| 提交项目地址 | https://gitcode.com/v50_/timm-dpn131.mx_in1k-NPU |
提交说明: 适配 timm/dpn131.mx_in1k 到昇腾 NPU (Ascend910)。使用 ModelScope snapshot_download 下载权重,timm.create_model(pretrained=False) 加载本地权重,包含推理验证、CPU-NPU 精度一致性检查和性能基准测试。精度: PASS (relative_error=0.1326%)。吞吐量: 32.66 images/sec
#NPU