[发明专利]一种模型移植的方法、系统及终端设备

权利要求书

1.一种模型移植的方法，其特征在于，包括：

获取模型文件并上传至模型仓库；

基于所述模型文件，获取模型结构；

判断所述模型结构是否正常；

若所述模型结构不正常，则重新获取所述模型文件并判断所述模型结构是否正常；

若所述模型结构正常，进行模型移植；

判断所述模型移植是否成功；

若所述模型移植成功，将移植后在不同平台的所述新模型文件上传至模型仓库；

若所述模型移植失败，则更换移植精度，进行下一轮模型移植，直至移植成功。

2.根据权利要求1所述的一种模型移植的方法，其特征在于：所述获取模型文件并上传至模型仓库包括：

获取所述模型文件的文件扩展名；

获取被移植平台信息；

基于所述被移植平台信息，获取验证文件的文件扩展名；

判断所述模型文件的文件扩展名和所述验证文件的文件扩展名是否相同；

若所述模型文件的文件扩展名和所述验证文件的文件扩展名不相同，修改所述验证文件的扩展名，并重新判断所述模型文件的文件扩展名和所述验证文件的文件扩展名是否相同；

若所述模型文件的文件扩展名和所述验证文件的文件扩展名相同，则上传所述模型文件至所述模型仓库。

3.根据权利要求1所述的一种模型移植的方法，其特征在于：所述判断所述模型结构是否正常包括：

基于所述模型结构，获取模型网络节点信息；

基于所述网络节点信息， 获取当前节点的名、与当前节点关联的入度节点名、出度节点名和节点算子超参数；

基于当前节点的名、与当前节点关联的入度节点名、出度节点名和节点算子超参数，比较所述模型结构与预设结构是否相同；

若所述模型结构与所述预设结构相同，则所述模型结构正常；

若所述模型结构与所述预设结构不相同，则所述模型结构不正常。

4.根据权利要求1所述的一种模型移植的方法，其特征在于：所述若所述模型结构正常之后包括：

获取所述模型文件的源文件；

基于移植需求，获取配置文件信息；

判断所述源文件中是否存在所述配置文件中对应的节点；

若所述源文件中存在所述配置文件中对应的节点，则进行模型移植；

若所述源文件中不存在所述配置文件中对应的节点，则生成异常报告。

5.据权利要求1所述的一种模型移植的方法，其特征在于：所述进行模型移植包括：

基于所述模型文件，获取模型算子；

获取待移植平台的模型算子格式；

基于开放神经网络交换格式，将所述模型算子转换成待移植平台的模型算子格式。

6.据权利要求5所述的一种模型移植的方法，其特征在于：所述更换移植精度，进行下一轮模型移植包括：

基于移植需求及当前移植精度，得到下一轮移植精度，所述下一轮移植精度对应的模型算子移植成功率大于所述当前移植精度对应的模型算子移植成功率；

根据所述下一轮所述移植精度进行下一轮所述模型移植。

7.根据权利要求6所述的一种模型移植的方法，其特征在于：所述判断所述模型移植是否成功包括：

获取所述模型移植前的模型算子和所述模型移植后的模型算子；

将所述模型移植后的模型算子与模型移植前的算子进行映射比对，判断是否有算子不存在算子映射集合中；

若是有算子不存在算子映射集合中，则模型移植失败；

若是所有算子都存在算子映射集合中，则模型移植成功。