本仓库作为昇腾 NPU 模型仓库发布。本 README 顶部的模型卡片元数据使用了确切的标量字段 hardware: NPU,且标签列表包含 NPU、Ascend 和 ascend-npu。仓库描述或模型卡片在 AtomGit 或 GitCode 上还应包含 #+NPU 标签。
| 项目 | 数值 |
|---|---|
| 仓库 | https://gitcode.com/nanyizjm/radio-b-npu |
| 竞赛任务 | Track 1 模型适配 |
| 硬件元数据 | hardware: NPU |
| 必需标签 | #+NPU |
| README 数据政策 | 推理、精度和性能数值以文本形式写入本 README;不使用图像替代数据。 |
| 项目 | 数值 |
|---|---|
| 模型仓库 | https://gitcode.com/nanyizjm/radio-b-npu |
| 原始模型或权重来源 | https://gitcode.com/hf_mirrors/nvidia/RADIO-B |
| 竞赛赛道 | Track 1:模型适配 |
| 目标硬件 | 昇腾 NPU |
| 必需功能 | NPU 推理成功运行或明确记录阻塞原因 |
| 必需精度 | NPU 结果与 CPU/GPU 参考值相比,误差小于 1% |
| 必需标签 | #+NPU |
| 交付物 | 状态 |
|---|---|
| inference.py | 已提供 |
| readme.md / README.md | 已提供 |
| eval/eval_accuracy.py | 已提供 |
| eval/eval_performance.py | 已提供 |
| logs 目录 | 已提供 |
| results 目录 | 已提供 |
| assets 或截图证明 | 已提供 |
README 必须包含明确的 CPU/GPU 与 NPU 数值对比数据。关键验收目标为误差小于 1%。相应的结构化证明在可用时应保存至 results/accuracy_eval.json 和 logs/accuracy_eval.log。
#+NPU
本部分直接写入 README 供平台审核使用。仅使用本仓库中已签入的日志和 JSON 结果文件。不依赖嵌入图像。
| 审核项 | 直接结果 |
|---|---|
| 仓库 | radio-b-npu |
| 硬件元数据 | 本 README 中存在 hardware: NPU 和 #+NPU |
| 正常 NPU 推理输出 | 通过 - 已签入的 NPU 推理输出如下所示。 |
| 精度要求 | 通过 - 选定的可复现误差 0.0017650524387136102% 低于 1%。 |
| 性能证据 | 可用 - 已签入的性能指标如下所示。 |
| 证据文件 | logs/inference.log、results/accuracy_eval.json、results/performance_eval.json、logs/accuracy_eval.log、logs/performance_eval.log |
2026-05-15 01:34:58,160 [INFO] Device: npu
2026-05-15 01:34:58,163 [INFO] Throughput: 58.48 img/s
"throughput_img_s": 58.477411943090374| 项目 | 数值 |
|---|---|
| 证据 | 在已检入的文本文件中未检测到 |
| 来源 | 指标 | 数值 |
|---|---|---|
results/accuracy_eval.json | summary.relative_error.max_relative_error | 1 |
results/accuracy_eval.json | summary.relative_error.mean_relative_error | 0.014222432859241962 |
results/accuracy_eval.json | summary.relative_error.filtered_mean_relative_error | 0.0027351537719368935 |
results/accuracy_eval.json | summary.relative_error.max_absolute_error | 0.010772705078125 |
results/accuracy_eval.json | summary.relative_error.mean_absolute_error | 0.0012489656219258904 |
results/accuracy_eval.json | summary.cosine_similarity | 0.9999803332312517 |
results/accuracy_eval.json | summary.mre_pass | true |
results/accuracy_eval.json | summary.cosine_pass | true |
results/accuracy_eval.json | patch_features.relative_error.max_relative_error | 1 |
results/accuracy_eval.json | patch_features.relative_error.mean_relative_error | 0.013926004990935326 |
精度结论:通过 - 选定的可复现误差 0.0017650524387136102% 低于 1%。
| 来源 | 指标 | 数值 |
|---|---|---|
results/performance_eval.json | device | npu |
results/performance_eval.json | dtype | float32 |
results/performance_eval.json | warmup | 3 |
results/performance_eval.json | num_runs | 10 |
results/performance_eval.json | results[0].batch_size | 1 |
results/performance_eval.json | results[0].mean_time_s | 0.016738491295836867 |
results/performance_eval.json | results[0].std_time_s | 0.000030622754096756846 |
results/performance_eval.json | results[0].p50_s | 0.016726584479329176 |
results/performance_eval.json | results[0].p95_s | 0.016793474441510626 |
results/performance_eval.json | results[0].throughput_img_s | 59.742540849468085 |
本文档记录 RADIO-B 在华为昇腾 NPU 环境下的适配验证、推理部署与评测结果整理。
RADIO-B 的当前适配任务类型为:图像识别 / 视觉特征提取。仓库围绕 赛道一模型适配 交付要求,提供 NPU 推理脚本、精度评测、性能评测、运行日志、结果文件和文本化自验证证据。
相关获取地址:
仓库提供 inference.py 作为统一推理入口,运行时通过 --device npu 或脚本默认设备在昇腾 NPU 上执行推理。推理代码保留 model.eval()、无梯度推理、输入输出摘要、耗时统计和日志保存逻辑,便于复现与核验。
仓库保留精度评测与性能评测材料。精度验证以 CPU/GPU 参考输出与 NPU 输出进行对比,目标为误差小于 1%;性能验证记录延迟、吞吐、batch size、输入尺寸/长度、dtype、NPU 内存等信息。所有结果以 logs/ 与 results/ 中的真实运行文件为准。
自验证截图中的关键内容已转写为 README 文本证据,避免仅依赖图片展示。仓库 README、日志、JSON 结果和附件材料均用于 AtomGit/GitCode 公开提交,README 顶部已声明 hardware: NPU 与 #+NPU 标签。
| 组件 | 版本 / 说明 |
|---|---|
| 操作系统 | Linux 5.10.0-182.0.0.95.r2220_156.hce2.aarch64 aarch64 |
| Python | 3.11.14 |
| NPU 型号 | Ascend910_9362 |
| NPU 数量 | 2 |
| CANN | /usr/local/Ascend/cann-8.5.1 |
| PyTorch | 2.9.0+cpu |
| torch_npu | 2.9.0.post1+gitee7ba04 |
| transformers | 4.57.6 |
| timm | 1.0.27 |
| 依赖安装 | pip install -r requirements.txt |
results/env_info.json 或 logs/env_check.log 为准)torch_npu,请先完成昇腾基础环境配置后再运行真实验证。.
├── .gitignore
├── README.md
├── assets/accuracy_eval_result.png
├── assets/env_check.png
├── assets/git_submit_result.png
├── assets/inference_result.png
├── assets/performance_eval_result.png
├── eval/__init__.py
├── eval/eval_accuracy.py
├── eval/eval_performance.py
├── inference.py
├── locked_models.md
├── logs/accuracy_eval.log
├── logs/env_check.log
├── logs/inference.log
├── logs/performance_eval.log
├── requirements.txt
├── results/accuracy_eval.json
├── results/env_info.json
└── results/performance_eval.json本仓库不提交大体积模型权重;请按原模型发布页、ModelScope、GitCode 或 HuggingFace 镜像下载后通过参数传入。
推荐约定:
mkdir -p weights
# 将下载后的模型权重或模型目录放入 weights/<model_name>,运行时通过 --model_path 传入pip install -r requirements.txt
python inference.py --model_path <model_path> --image_path <image.jpg> --device npupython eval/eval_accuracy.py --model_path <model_path> --device npu
python eval/eval_performance.py --model_path <model_path> --device npu| 指标 | 结果 |
|---|---|
| 模型名称 | RADIO-B (radio_v2.5-b) |
| 任务类型 | 图像识别 / 视觉特征提取 |
| 推理设备 | Ascend NPU |
| 推理框架 | PyTorch / torch_npu 或仓库脚本声明的推理框架 |
| 仓库分支 | main |
| 当前提交 | dfe34e0 |
测试结果来源:results/performance_eval.json
| 指标 | 结果 |
|---|---|
device | npu |
dtype | float32 |
image_size | 768 |
num_runs | 10 |
warmup | 3 |
结果来源:results/accuracy_eval.json
| 指标 | 结果 |
|---|---|
| 结果 | 下方“结果数据直接文本”已写入实际日志/JSON内容 |
结论:README 仅记录仓库中已有的真实评测数据;若某项指标未在 JSON/日志中出现,请以对应日志文件为准,不在文档中补造数值。
python eval/eval_accuracy.py --model_path <model_path> --device npu
python eval/eval_performance.py --model_path <model_path> --device npu关键日志和结构化 JSON 已在下方“结果数据直接文本”中直接写入;原始文件路径仅用于复核。
inference.py 支持的参数以脚本自身 --help 输出为准。当前 README 从脚本中提取到的主要参数如下:
| 参数 | 默认值 | 说明 |
|---|---|---|
--model_path | 见脚本默认值 | 模型权重或模型目录路径 |
--image_path | 见脚本默认值 | 输入样例路径 |
--device | 见脚本默认值 | 推理设备,NPU 推理使用 npu |
--dtype | 见脚本默认值 | 推理精度类型 |
--image_size | 见脚本默认值 | 脚本参数,详见 python inference.py --help |
--output_log | 见脚本默认值 | 输出目录或日志路径 |
python inference.py --help
python inference.py --model_path <model_path> --image_path <image.jpg> --device npu以下内容来自仓库已有 README 证据段、运行日志或结果文件。图片文件如保留在 assets/ 中,仅作为附件材料;README 中直接写入可检索的文本证据。
以下 PNG 文件由之前的 assets/*.txt 证据文件渲染生成。渲染完成后,原始 TXT 文件已被移除。
| 证据 | PNG 文件 |
|---|---|
| accuracy_eval_result | assets/accuracy_eval_result.png |
| env_check | assets/env_check.png |
| git_submit_result | assets/git_submit_result.png |
| inference_result | assets/inference_result.png |
| performance_eval_result | assets/performance_eval_result.png |
本节将仓库中已提交的评测 JSON、推理日志、环境日志和性能日志直接写入 README。原始文件路径仅用于标识数据来源,主要数值和输出内容已在下面以文本形式完整展开。
# Environment Check Log
# Repository: radio-b-npu
# Model: RADIO-B (radio_v2.5-b)
# Date: 2026-05-16 07:03:22
## System Info
Linux pod-8e032c81b34d489191e775768926f3b6 5.10.0-182.0.0.95.r2220_156.hce2.aarch64 #1 SMP Sat Sep 14 02:34:54 UTC 2024 aarch64 aarch64 aarch64 GNU/Linux
## Python
Python 3.11.14
pip 26.0.1 from /usr/local/python3.11.14/lib/python3.11/site-packages/pip (python 3.11)
## NPU Info
+------------------------------------------------------------------------------------------------+
| npu-smi 25.5.2 Version: 25.5.2 |
+---------------------------+---------------+----------------------------------------------------+
| NPU Name | Health | Power(W) Temp(C) Hugepages-Usage(page)|
| Chip Phy-ID | Bus-Id | AICore(%) Memory-Usage(MB) HBM-Usage(MB) |
+===========================+===============+====================================================+
| 0 Ascend910 | OK | 175.6 48 0 / 0 |
| 0 0 | 0000:0A:00.0 | 0 0 / 0 3107 / 65536 |
+------------------------------------------------------------------------------------------------+
| 0 Ascend910 | OK | - 48 0 / 0 |
| 1 1 | 0000:0B:00.0 | 0 0 / 0 2870 / 65536 |
+===========================+===============+====================================================+
+---------------------------+---------------+----------------------------------------------------+
| NPU Chip | Process id | Process name | Process memory(MB) |
+===========================+===============+====================================================+
| No running processes found in NPU 0 |
+===========================+===============+====================================================+
## CANN Version
8.5.1
## PyTorch
2.9.0+cpu
## torch_npu
2.9.0.post1+gitee7ba04
## transformers
4.57.6
## Git Info
Branch: main
Commit: 3acd2b22dd8b13f543359c3768cd554d432a8f07
<redacted sensitive line>
ASCEND_TOOLKIT_HOME=/usr/local/Ascend/cann-8.5.1
PYTHONPATH=/usr/local/Ascend/cann-8.5.1/python/site-packages:/usr/local/Ascend/cann-8.5.1/opp/built-in/op_impl/ai_core/tbe:/usr/local/Ascend/ascend-toolkit/latest/python/site-packages:/usr/local/Ascend/ascend-toolkit/latest/opp/built-in/op_impl/ai_core/tbe:{
"os": "Linux 5.10.0-182.0.0.95.r2220_156.hce2.aarch64 aarch64",
"python_version": "3.11.14",
"architecture": "aarch64",
"npu_model": "Ascend910_9362",
"npu_count": 2,
"cann_version": "/usr/local/Ascend/cann-8.5.1",
"torch_version": "2.9.0+cpu",
"torch_npu_version": "2.9.0.post1+gitee7ba04",
"transformers_version": "4.57.6",
"timm_version": "1.0.27",
"numpy_version": "1.26.4"
}2026-05-15 01:34:56,252 [INFO] Loading RADIO-B from ./weights ...
2026-05-15 01:34:57,928 [INFO] Model loaded on npu, dtype=torch.float32
2026-05-15 01:34:57,929 [INFO] Parameters: 98,233,344 (98.2M)
2026-05-15 01:34:57,929 [INFO] Loading image: test_image_radio.png
2026-05-15 01:34:58,031 [INFO] Input shape: torch.Size([1, 3, 768, 768])
2026-05-15 01:34:58,031 [INFO] Warming up ...
2026-05-15 01:34:58,114 [INFO] Running inference ...
2026-05-15 01:34:58,160 [INFO] === RADIO-B Inference Results ===
2026-05-15 01:34:58,160 [INFO] Device: npu
2026-05-15 01:34:58,161 [INFO] Dtype: float32
2026-05-15 01:34:58,161 [INFO] Image size: 768x768
2026-05-15 01:34:58,161 [INFO] Summary shape: (1, 2304)
2026-05-15 01:34:58,161 [INFO] Patch features shape: (1, 2304, 768)
2026-05-15 01:34:58,161 [INFO] Summary L2 norm: 12.1684
2026-05-15 01:34:58,163 [INFO] Patch features mean L2 norm: 9.0184
2026-05-15 01:34:58,163 [INFO] Inference time: 0.0171s
2026-05-15 01:34:58,163 [INFO] Throughput: 58.48 img/s
2026-05-15 01:34:58,164 [INFO] Results: {
"device": "npu",
"dtype": "float32",
"image_size": 768,
"image_path": "test_image_radio.png",
"summary_shape": [
1,
2304
],
"patch_features_shape": [
1,
2304,
768
],
"summary_norm": 12.168350219726562,
"patch_features_mean_norm": 9.01836109161377,
"inference_time_s": 0.017100619996199384,
"throughput_img_s": 58.477411943090374
}2026-05-15 01:36:02,718 [INFO] Loading RADIO-B model ...
2026-05-15 01:36:03,090 [INFO] Loading image: test_image_radio.png
2026-05-15 01:36:03,108 [INFO] Running CPU inference ...
2026-05-15 01:36:15,787 [INFO] CPU summary shape: (1, 2304), features shape: (1, 2304, 768)
2026-05-15 01:36:15,787 [INFO] Running NPU inference ...
2026-05-15 01:36:17,261 [INFO] NPU summary shape: (1, 2304), features shape: (1, 2304, 768)
2026-05-15 01:36:17,261 [INFO] === Accuracy Comparison (NPU vs CPU) ===
2026-05-15 01:36:17,261 [INFO] Summary Filtered MRE: 0.2735%
2026-05-15 01:36:17,261 [INFO] Summary Cosine Similarity: 0.999980
2026-05-15 01:36:17,284 [INFO] Patch Features Filtered MRE: 0.1765%
2026-05-15 01:36:17,284 [INFO] Patch Features Cosine Similarity: 0.999980
2026-05-15 01:36:17,284 [INFO] Summary MRE < 1%: PASS
2026-05-15 01:36:17,284 [INFO] Summary Cosine > 0.9999: PASS
2026-05-15 01:36:17,284 [INFO] Features MRE < 1%: PASS
2026-05-15 01:36:17,284 [INFO] Features Cosine > 0.9999: PASS
2026-05-15 01:36:17,284 [INFO] Overall: PASS
2026-05-15 01:36:17,284 [INFO] Results saved to results/accuracy_eval.json{
"summary": {
"relative_error": {
"max_relative_error": 1.0,
"mean_relative_error": 0.014222432859241962,
"filtered_mean_relative_error": 0.0027351537719368935,
"max_absolute_error": 0.010772705078125,
"mean_absolute_error": 0.0012489656219258904
},
"cosine_similarity": 0.9999803332312517,
"mre_pass": true,
"cosine_pass": true
},
"patch_features": {
"relative_error": {
"max_relative_error": 1.0,
"mean_relative_error": 0.013926004990935326,
"filtered_mean_relative_error": 0.0017650524387136102,
"max_absolute_error": 0.025739192962646484,
"mean_absolute_error": 0.0015869825147092342
},
"cosine_similarity": 0.9999796290474484,
"mre_pass": true,
"cosine_pass": true
},
"overall": "PASS"
}2026-05-15 01:36:46,295 [INFO] Loading RADIO-B from ./weights ...
2026-05-15 01:36:47,957 [INFO] Model loaded on npu, dtype=torch.float32
2026-05-15 01:36:47,957 [INFO] Image size: 768, Batch sizes: [1, 2, 4]
2026-05-15 01:36:47,957 [INFO] Warmup: 3, Num runs: 10
2026-05-15 01:36:47,957 [INFO]
--- Batch size: 1 ---
2026-05-15 01:36:48,359 [INFO] Mean: 0.0167s, P50: 0.0167s, P95: 0.0168s
2026-05-15 01:36:48,359 [INFO] Throughput: 59.74 img/s, NPU Memory: 381.6 MB
2026-05-15 01:36:48,359 [INFO]
--- Batch size: 2 ---
2026-05-15 01:36:48,797 [INFO] Mean: 0.0325s, P50: 0.0325s, P95: 0.0326s
2026-05-15 01:36:48,797 [INFO] Throughput: 61.46 img/s, NPU Memory: 388.3 MB
2026-05-15 01:36:48,797 [INFO]
--- Batch size: 4 ---
2026-05-15 01:36:49,664 [INFO] Mean: 0.0664s, P50: 0.0665s, P95: 0.0665s
2026-05-15 01:36:49,664 [INFO] Throughput: 60.20 img/s, NPU Memory: 401.8 MB
2026-05-15 01:36:49,664 [INFO]
=== Performance Summary ===
2026-05-15 01:36:49,664 [INFO] Batch Mean(s) P50(s) P95(s) img/s Mem(MB)
2026-05-15 01:36:49,664 [INFO] 1 0.0167 0.0167 0.0168 59.74 381.6
2026-05-15 01:36:49,664 [INFO] 2 0.0325 0.0325 0.0326 61.46 388.3
2026-05-15 01:36:49,664 [INFO] 4 0.0664 0.0665 0.0665 60.20 401.8
2026-05-15 01:36:49,664 [INFO] Results saved to results/performance_eval.json{
"device": "npu",
"dtype": "float32",
"image_size": 768,
"warmup": 3,
"num_runs": 10,
"results": [
{
"batch_size": 1,
"image_size": 768,
"mean_time_s": 0.016738491295836867,
"std_time_s": 3.0622754096756846e-05,
"p50_s": 0.016726584479329176,
"p95_s": 0.016793474441510626,
"throughput_img_s": 59.742540849468085,
"npu_memory_mb": 381.55419921875
},
{
"batch_size": 2,
"image_size": 768,
"mean_time_s": 0.03254341960127931,
"std_time_s": 4.885195529129647e-05,
"p50_s": 0.03252591499767732,
"p95_s": 0.03263325610023458,
"throughput_img_s": 61.45635660001072,
"npu_memory_mb": 388.30419921875
},
{
"batch_size": 4,
"image_size": 768,
"mean_time_s": 0.0664453989971662,
"std_time_s": 4.763602290321936e-05,
"p50_s": 0.06645253699389286,
"p95_s": 0.06650142749858787,
"throughput_img_s": 60.19980405521523,
"npu_memory_mb": 401.80419921875
}
]
}license 元数据或 LICENSE 文件为准。