[发明专利]模型处理方法、装置以及电子设备

权利要求书

1.一种模型处理方法，其特征在于，应用于模型处理装置，所述方法包括：

获取存储有算法模型的待处理模型文件，所述待处理模型文件为基于flatbuffer方式生成；

读取所述待处理模型文件中的根表，所述根表存储有所述待处理模型文件所包括的模型表的存储位置；

根据所述根表存储的所述存储位置获取到所述模型表；

读取所述模型表中的子图表，从所述子图表中读取算子表，并对所述算子表进行解析，得到包括算子信息的解析结果；

将所述解析结果按照指定的格式进行存储。

2.根据权利要求1所述的方法，其特征在于，所述获取存储有算法模型的待处理模型文件的步骤包括：

将所述存储有算法模型的待处理模型文件加载到内存中；

其中，所述存储位置为所述模型表在内存中的偏移地址。

3.根据权利要求1所述的方法，其特征在于，所述读取所述模型表中的子图表的步骤包括：

获取所述待处理模型文件中的配置表，所述配置表中存储有子图表的存储位置；

基于所述子图表的存储位置获取到所述模型表中的子图表。

4.根据权利要求3所述的方法，其特征在于，所述从所述子图表中读取算子表，并对所述算子表进行解析，得到包括算子信息的解析结果的步骤包括：

从所述子图表中读取算子表，按照指定顺序对所述算子表的内容进行检测；

若检测到包括目标标识的内容块，将所述内容块作为一个算子，将所述目标标识对应的算子参数作为算子信息；

生成包括所述算子信息的解析结果。

5.根据权利要求4所述的方法，其特征在于，所述算子参数包括算子的输入数据、输出数据以及算子运算时的附加信息。

6.根据权利要求4所述的方法，其特征在于，所述从所述子图表中读取算子表，按照指定顺序对所述算子表的内容进行检测的步骤包括：

基于所述算子表的表名识别所述算子表中算子的数量；

若识别所述数量大于目标数量，启动多个线程分别从所述算子表中的不同位置对所述算子表的内容进行检测。

7.根据权利要求1-6任一所述的方法，其特征在于，所述将所述解析结果按照指定的格式进行存储的步骤包括：

识别所述解析结果所属算子的类型；

按照所述类型所对应的存储格式对算子进行存储。

8.一种模型处理装置，其特征在于，所述装置包括：

文件获取单元，用于获取存储有算法模型的待处理模型文件，所述待处理模型文件为基于flatbuffer方式生成；

文件读取单元，用于读取所述待处理模型文件中的根表，所述根表存储有所述待处理模型文件所包括的模型表的存储位置；

数据获取单元，用于根据所述根表存储的所述存储位置获取到所述模型表；

数据解析单元，用于读取所述模型表中的子图表，从所述子图表中读取算子表，并对所述算子表进行解析，得到包括算子信息的解析结果；

数据存储单元，用于将所述解析结果按照指定的格式进行存储。

9.一种电子设备，其特征在于，包括处理器以及存储器；

一个或多个程序被存储在所述存储器中并被配置为由所述处理器执行以实现权利要求1-7任一所述的方法。

10.一种计算机可读存储介质，其特征在于，所述计算机可读存储介质中存储有程序代码，其中，在所述程序代码被处理器运行时执行权利要求1-7任一所述的方法。