Genius-Society/insecta on Ascend NPU

1. 简介

将 Genius-Society/insecta 昆虫识别模型适配到昇腾 NPU (Ascend910B)。

原模型为 ONNX 格式，包含 detector（目标检测）和 identifier（图片分类）两个组件。本次适配将 identifier 分类模型通过 onnx2torch 转换为 PyTorch 格式，并在 NPU 上完成推理验证、精度一致性检查和性能基准测试。

支持类别：2037 种昆虫及节肢动物
分类模型输入：224×224 RGB
分类模型输出：2037 维 logits

2. 验证环境

项目	版本/型号
硬件	Ascend910B
CANN	8.5.1
Python	3.11
torch	2.x
torch-npu	2.x

3. 推理运行

pip install -r requirements.txt
python inference.py

推理结果 (NPU Top-5):

排名	类别	拉丁名	概率
Top-1	鳞翅目_斑蝶科_斑蝶属_金斑蝶	Danaus chrysippus	0.9064
Top-2	鳞翅目_斑蝶科_斑蝶属_虎斑蝶	Danaus genutia	0.0053
Top-3	鳞翅目_凤蝶科_凤蝶属_玉带美凤蝶(玉带凤蝶)	Papilio polytes	0.0016
Top-4	鳞翅目_珍蝶科_珍蝶属_斑珍蝶2	Acraea terpsicore	0.0014
Top-5	鳞翅目_蛱蝶科_锯蛱蝶属_白带锯蛱蝶	Cethosia cyane	0.0014

日志保存在 logs/inference.log。

4. 精度验证

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

指标	数值
max_abs_error	0.009521
mean_abs_error	0.001357
relative_error	0.0710%
cosine_similarity	0.999998
threshold	1.0%
结果	PASS

CPU Top-1: 鳞翅目_斑蝶科_斑蝶属_金斑蝶 NPU Top-1: 鳞翅目_斑蝶科_斑蝶属_金斑蝶 Top-1 match: True

5. 性能参考

指标	数值
avg_latency	4.76 ms
throughput	209.99 images/sec

测试环境：单卡 Ascend910B，batch_size=1，输入 224×224。

6. 精度评测说明

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

7. 自验证截图

见 screenshots/self_verification.png

8. 日志文件

logs/inference.log — 推理输出
logs/accuracy.log — 精度验证结果
logs/benchmark.log — 性能基准数据

9. 注意事项

原模型为 ONNX 格式，适配时使用 onnx2torch 动态转换为 PyTorch 模型
权重文件（.onnx）不提交到仓库，运行时从 ModelScope 缓存目录加载
detector（目标检测）组件为 ONNX 原始格式，当前适配聚焦于 identifier（图片分类）组件
若 ModelScope 缓存路径不同，请修改 model_utils.py 中的 MODEL_DIR

10. 标签

#NPU #Ascend #InsectRecognition #ONNX #MobileNetV2

1. 简介

将 Genius-Society/insecta 昆虫识别模型适配到昇腾 NPU (Ascend910B)。

支持类别：2037 种昆虫及节肢动物

分类模型输入：224×224 RGB

分类模型输出：2037 维 logits

项目

版本/型号

硬件

Ascend910B

CANN

8.5.1

Python

3.11

torch

2.x

torch-npu

2.x

3. 推理运行

pip install -r requirements.txt
python inference.py

推理结果 (NPU Top-5):

排名	类别	拉丁名	概率
Top-1	鳞翅目_斑蝶科_斑蝶属_金斑蝶	Danaus chrysippus	0.9064
Top-2	鳞翅目_斑蝶科_斑蝶属_虎斑蝶	Danaus genutia	0.0053
Top-3	鳞翅目_凤蝶科_凤蝶属_玉带美凤蝶(玉带凤蝶)	Papilio polytes	0.0016
Top-4	鳞翅目_珍蝶科_珍蝶属_斑珍蝶2	Acraea terpsicore	0.0014
Top-5	鳞翅目_蛱蝶科_锯蛱蝶属_白带锯蛱蝶	Cethosia cyane	0.0014

日志保存在 logs/inference.log。

4. 精度验证

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

指标	数值
max_abs_error	0.009521
mean_abs_error	0.001357
relative_error	0.0710%
cosine_similarity	0.999998
threshold	1.0%
结果	PASS

CPU Top-1: 鳞翅目_斑蝶科_斑蝶属_金斑蝶 NPU Top-1: 鳞翅目_斑蝶科_斑蝶属_金斑蝶 Top-1 match: True

指标

数值

avg_latency

4.76 ms

throughput

209.99 images/sec

9. 注意事项

原模型为 ONNX 格式，适配时使用 onnx2torch 动态转换为 PyTorch 模型

权重文件（.onnx）不提交到仓库，运行时从 ModelScope 缓存目录加载

detector（目标检测）组件为 ONNX 原始格式，当前适配聚焦于 identifier（图片分类）组件

若 ModelScope 缓存路径不同，请修改 model_utils.py 中的 MODEL_DIR