fbnetv3_d.ra2_in1k NPU 推理适配
模型介绍
FBNetV3-D 是 Facebook 提出的 FBNetV3 系列神经网络架构搜索模型，是 FBNetV3 的中等规模变体。该模型在 ImageNet-1k 数据集上训练，通过 NAS 技术搜索高效 CNN 架构，在精度和计算量之间取得良好平衡。
- 原始模型地址: https://huggingface.co/timm/fbnetv3_d.ra2_in1k
- ModelScope 地址: https://www.modelscope.cn/models/timm/fbnetv3_d.ra2_in1k
- 任务类型: 图像分类 (Image Classification)
- 模型框架: PyTorch + timm
模型信息
| 属性 | 值 |
|---|---|
| 架构类型 | MobileNetV3 |
| 输入尺寸 | 224 × 224 |
| 类别数 | 1000 |
| 预处理均值 | [0.485, 0.456, 0.406] |
| 预处理标准差 | [0.229, 0.224, 0.225] |
NPU 适配说明
此模型基于 torch_npu 在华为昇腾 NPU 上完成推理适配。
环境准备
pip install -r requirements.txt推理命令
# CPU
python3 inference.py --device cpu
# NPU
python3 inference.py --device npu推理结果
CPU
Avg: 49.78 ms (10 runs)
Top-5: Class 21 (0.106597), Class 92 (0.050216), Class 127 (0.044470), Class 22 (0.037443), Class 23 (0.037318)NPU
Avg: 11.37 ms (10 runs)
Top-5: Class 21 (0.106680), Class 92 (0.050331), Class 127 (0.044525), Class 22 (0.037564), Class 23 (0.037404)性能对比
| 设备 | 耗时 (ms) | 加速比 |
|---|---|---|
| CPU | 49.78 | 1.0× |
| NPU | 11.37 | 4.4× |
CPU/NPU 精度测试
精度测试结果
| 指标 | 值 |
|---|---|
| Logits 余弦相似度 | 0.99999955 |
| Probabilities 最大绝对误差 | 0.00012109 |
| Probabilities 平均绝对误差 | 0.00000067 |
| Top-1 类别一致率 | 100.00% |
精度测试结论
NPU 与 CPU 推理误差为 0.01%（概率最大绝对误差），符合精度误差小于 1% 的要求。
推理截图
推理成功证据
以下日志展示了 NPU 推理成功的关键信息：
CPU Top-1: class 21, prob 0.106597
NPU Top-1: class 21, prob 0.106680
Top-1 agree: 100.00%
Top-5:模型标签
- #+NPU
- #+CV
- #+图像分类
- #+昇腾
- #+FBNetV3