w
gcw_uQ09W7jl/jp_lan-cv_vitg_classification_dinov2-NPU
模型介绍文件和版本Pull Requests讨论分析
下载使用量0

jp_lan/cv_vitg_classification_dinov2 on Ascend NPU

1. 简介

本项目将 DINOv2 ViT-g/14 图像分类模型 适配到单卡昇腾 NPU(Ascend910)上运行。

模型由 DINOv2 ViT-g/14 主干网络和线性分类头组成,在 ImageNet-1k 上达到 86.5% 的 top-1 准确率。适配过程使用 ModelScope 本地权重加载,不依赖 HuggingFace 自动下载。

2. 验证环境

项目版本
NPUAscend910
CANN8.5.1
Python3.11.14
PyTorch2.x
torch_npu已安装

3. 推理运行

pip install -r requirements.txt
python inference.py

推理结果 (NPU Top-5):

排名类别概率
Top-1house finch, linnet, Carpodacus mexicanus0.9803
Top-2brambling, Fringilla montifringilla0.0148
Top-3goldfinch, Carduelis carduelis0.0017
Top-4jay0.0006
Top-5indigo bunting, indigo finch, indigo bird, Passerina cyanea0.0003

日志保存在 logs/inference.log。

4. 精度验证

对单张测试图片进行 CPU 与 NPU 一致性验证:

指标数值
max_abs_error0.002928
mean_abs_error0.000613
relative_error0.0534%
cosine_similarity1.000000
threshold1.0%
结果PASS

CPU Top-1: house finch, linnet, Carpodacus mexicanus NPU Top-1: house finch, linnet, Carpodacus mexicanus Top-1 match: True

5. 性能参考

指标数值
avg_time_ms23.78
min_time_ms23.59
max_time_ms25.00
p50_time_ms23.63
p90_time_ms23.86
p95_time_ms24.43
throughput_images_per_sec42.05

6. 精度评测说明

本项目包含单图 smoke consistency 验证,非官方 ImageNet 完整验证集评测。详细指标见第 4 节。

7. 自验证截图

见 screenshots/self_verification.txt 和 screenshots/self_verification.png。

8. 日志文件

  • logs/inference.log — 推理输出
  • logs/accuracy.log — 精度验证结果
  • logs/benchmark.log — 性能基准测试结果
  • logs/env_check.log — NPU 环境检查

9. 注意事项

  • 模型权重文件(*.pth)不提交到 Git,通过 .gitignore 排除
  • 首次运行会自动从 ModelScope 缓存加载权重,无需手动下载
  • 预处理使用 BGR 格式、224x224 分辨率、ImageNet 标准归一化

10. 标签

#NPU #Ascend #DINOv2 #ImageClassification