m0_74196153/skresnext50_32x4d_ra_in1k-npu

skresnext50_32x4d.ra_in1k-npu

模型介绍

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

原始模型地址

模型来源：timm/skresnext50_32x4d.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	735	0.043481
2	533	0.036747
3	911	0.031553
4	885	0.023663
5	539	0.023589

NPU 推理结果

排名	类别	概率
1	21	0.038261
2	574	0.019151
3	892	0.018645
4	852	0.017257
5	844	0.016019

CPU/NPU 精度测试

测试方法

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

精度指标说明

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

CPU/NPU 精度测试结果

指标	值
参数量	27.48M
Logits MAE	1.47674074e-01
Logits Cosine Similarity	0.97930552
Probability MAE	3.84667407e-04
Top-1 Agreement	0.00%
CPU 推理耗时	0.6116s
NPU 推理耗时	0.0116s
加速比	52.81x

结论

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

模型标签

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

版权声明

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

精度对比结论

对比项	CPU	NPU	误差
余弦相似度	1.0	0.99954745	0.0416%
最大绝对误差	-	-	0.003067
平均绝对误差	-	-	0.000978
Top-1 一致	-	-	是

结论： CPU 与 NPU 精度误差为 0.0416%，小于 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