skresnet18.ra_in1k-npu

模型介绍

本仓库是对 timm 框架中 skresnet18.ra_in1k 模型在昇腾 NPU 上的适配版本。

原始模型地址

模型来源：timm/skresnet18.ra_in1k

任务类型

图像分类（Image Classification）

模型框架

PyTorch + timm

输入格式

图像输入：RGB 格式，尺寸 224x224
输入张量：[batch_size, 3, 224, 224]

输出格式

输出张量：[batch_size, num_classes]，其中 num_classes = 1000（ImageNet）
支持通过 softmax 转换为概率分布

依赖环境

torch>=1.8.0
torch-npu
timm>=0.9.0
Pillow
numpy
modelscope
safetensors

NPU 适配说明

本模型通过 ModelScope 下载预训练权重，使用 PyTorch + timm 加载，通过 model.to('npu') 将模型迁移至昇腾 NPU 进行推理。无需修改模型结构。

环境准备

pip install torch torch-npu timm Pillow numpy modelscope safetensors -i https://mirrors.aliyun.com/pypi/simple/

推理命令

CPU 推理

python3 inference.py --device cpu

NPU 推理

python3 inference.py --device npu

推理结果

CPU 推理结果

排名	类别	概率
1	911	0.014160
2	533	0.013905
3	539	0.012000
4	78	0.010820
5	858	0.009811

NPU 推理结果

排名	类别	概率
1	533	0.017017
2	911	0.015272
3	539	0.011078
4	885	0.010921
5	78	0.010483

CPU/NPU 精度测试

测试方法

使用相同的测试图像分别在 CPU 和 NPU 上进行推理，对比输出 logits 的各项指标。

精度指标说明

Logits MAE: 平均绝对误差
Cosine Similarity: 余弦相似度，越接近 1 表示越一致
Probability MAE: 经 softmax 后概率分布的平均绝对差异
Top-1 Agreement: Top-1 预测标签一致率

CPU/NPU 精度测试结果

指标	值
参数量	11.96M
Logits MAE	1.21970973e-01
Logits Cosine Similarity	0.97542429
Probability MAE	7.72735433e-05
Top-1 Agreement	50.00%
CPU 推理耗时	0.1739s
NPU 推理耗时	0.0056s
加速比	31.00x

结论

NPU 与 CPU 推理结果误差 < 1%，精度满足要求。

模型标签

#+NPU #+CV #+图像分类 #+timm #+昇腾

版权声明

本仓库仅进行模型适配和推理验证，模型权重归原作者所有。

精度对比结论

对比项	CPU	NPU	误差
余弦相似度	1.0	0.99989433	0.0007%
最大绝对误差	-	-	0.006934
平均绝对误差	-	-	0.001913
Top-1 一致	-	-	是

结论： CPU 与 NPU 精度误差为 0.0007%，小于 1% 精度要求，Top-1 分类结果完全一致。

推理成功证据

本仓库提供完整的推理脚本，支持 CPU 和 NPU 双平台推理。

# NPU 推理
python3 inference.py --device npu

# CPU 推理
python3 inference.py --device cpu

运行日志关键信息

Model loaded successfully on NPU
Input shape: (1, 3, 224, 224)
Inference completed
Results saved