在算能(Sophgo)的 .bmodel
文件(编译后的 TPU 模型文件)中,功能模块标识是模型内部对独立计算单元或子网络结构的命名与分类机制。这些标识通过 bmrt_get_network_names
等接口对外暴露,是模型模块化设计的核心体现。以下是其具体含义与作用分析:
一、功能模块标识的定义
-
语义化命名
每个功能模块的名称(如detect
、classify
)对应模型中的独立计算图(Subgraph),类似于人类大脑中视觉、听觉等功能模块的分工。例如:- 多任务模型:检测任务可能对应
detect
,分类任务对应classify
。 - 多阶段模型:OCR 模型可能分为
stage1
(检测阶段)和stage2
(识别阶段)。
- 多任务模型:检测任务可能对应
-
技术实现
在模型编译阶