Resnext101 - 昇腾 NPU 适配

模型介绍

本仓库为基于 Timm 的 Image Classification 模型，已适配华为昇腾 Ascend NPU，支持在 CPU 和 NPU 两种设备上运行推理。

• 任务类型: Image Classification
• 模型框架: PyTorch
• 硬件支持: CPU / Ascend NPU

依赖环境

pytorch
torch_npu
opencv-python
numpy

NPU 适配说明

1. 使用 torch_npu 将模型加载到昇腾 NPU 设备
2. 支持 CPU 和 NPU 双平台推理
3. 使用 torch.npu.synchronize() 确保正确的性能测量

推理命令

CPU 推理

python3 inference.py --device cpu

NPU 推理

python3 inference.py --device npu

推理结果

推理完成后输出结果和性能数据。

推理成功证据

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

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

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

推理完成后会输出推理结果和耗时，表明模型在 NPU 上推理成功。

文件说明

精度结论

基于现有评测数据，CPU 与 NPU 的 平均绝对误差 精度误差为 0.0637%，小于 1% 的精度要求。

模型标签

#+NPU #+昇腾 #+CV #+PyTorch