XiaYuanOwO/ascend-resnet34-a1-in1k-model
模型介绍文件和版本Pull Requests讨论分析
下载使用量0

面向昇腾NPU的ResNet-34图像分类模型

#+NPU

NPU Python PyTorch

模型概述

模型名称: resnet34.a1_in1k

网络架构: ResNet-34(含34层的残差网络)

参数量: ~21.8M

预训练数据集: ImageNet-1k(120万张图像,1000个类别)

任务类型: 图像分类

模型描述: 采用AutoAugment(a1)预处理的ResNet-34,在ImageNet-1k上进行预训练。该模型利用残差连接实现更深网络的训练,并在训练过程中应用AutoAugment学习到的增强策略以提升精度。

硬件环境

目标平台: 昇腾NPU(华为Atlas)

运行时: torch_npu(昇腾PyTorch插件)

性能表现

推理性能(单张图像)

设备延迟吞吐量相对CPU加速比
NPU2.45 ms408 FPS1.23x
CPU3.01 ms332 FPS1.00x

Top-5预测结果对比

排名NPU类别NPU概率CPU类别CPU概率是否匹配
12590.84722590.8471是
22810.05232810.0524是
32820.04122820.0411是
42880.01582880.0159是
52850.01242850.0123是

Top-5预测匹配度:100%

模型适配总结

属性值
状态SUCCESS
使用预训练是
使用本地权重是
1%内匹配度是

使用方法

安装

pip install -r requirements.txt

运行推理

python inference.py

输出文件

  • logs/run_npu.log - NPU推理性能日志
  • logs/accuracy_compare.log - CPU与NPU预测结果对比
  • logs/summary.json - 适配结果摘要

环境要求

  • Python 3.8及以上版本
  • PyTorch 2.0及以上版本
  • timm(PyTorch图像模型库)
  • torchvision
  • torch_npu(昇腾PyTorch插件)
  • CANN(神经网络计算架构)

许可证

本模型遵循其来源(timm/ImageNet)的许可证。

致谢

  • 模型来源于timm库
  • 预训练权重来源于ImageNet-1k
  • 通过torch_npu获得昇腾NPU支持