一种模型处理方法及装置制造方法及图纸

本申请实施例提供一种模型处理方法及装置，涉及数据处理技术领域，该模型处理方法包括：先获取待处理网络模型和目标硬件平台的硬件信息；并对待处理网络模型进行模型转换处理，得到能够在目标硬件平台上运行的目标模型；当目标模型的通过模型测试时，对目标模型进行打包处理得到模型包；再对模型包进行模型包测试，得到能够正式发布使用的目标模型包；最后存储目标模型包，能够自动对模型进行适配和优化处理，处理效率高，准确性高。准确性高。准确性高。

【技术实现步骤摘要】
一种模型处理方法及装置


[0001]本申请涉及数据处理
，具体而言，涉及一种模型处理方法及装置。

技术介绍

[0002]随着网络科技的不断发展，人工智能模型在安防、工业、交通等各种场景已得到广泛应用。由于不同的使用场景使用需求不同，相应的设备硬件也各不相同，不同硬件提供算力的核心模块各有差异，则人工智能模型在各个硬件平台使用时，一般都需要单独适配和优化，从而保证模型的正常且稳定的使用功能。现有的模型处理方法，首先需要开发人员掌握针对每种硬件平台的适配和优化技巧，然后通过人工处理的方式进行模型的适配和优化调整。然而，在实践中发现，现有方法模型处理效率低，准确性低。

技术实现思路

[0003]本申请实施例的目的在于提供一种模型处理方法及装置，能够自动对模型进行适配和优化处理，处理效率高，准确性高。
[0004]本申请实施例第一方面提供了一种模型处理方法，包括：
[0005]获取待处理网络模型和目标硬件平台的硬件信息；
[0006]根据所述硬件信息对所述待处理网络模型进行模型转换处理，得到能够在所述目标硬件平台上运行的目标模型；
[0007]当所述目标模型的通过模型测试时，对所述目标模型进行打包处理得到模型包；
[0008]对所述模型包进行模型包测试，得到能够正式发布使用的目标模型包；
[0009]存储所述目标模型包。
[0010]在上述实现过程中，先获取待处理网络模型和目标硬件平台的硬件信息；并对待处理网络模型进行模型转换处理，得到能够在目标硬件平台上运行的目标模型；当目标模型的通过模型测试时，对目标模型进行打包处理得到模型包；再对模型包进行模型包测试，得到能够正式发布使用的目标模型包；最后存储目标模型包，能够自动对模型进行适配和优化处理，处理效率高，准确性高。
[0011]进一步地，所述方法还包括：
[0012]获取所述模型转换处理时的转换数据、所述模型测试的第一测试数据、所述对所述目标模型进行打包处理时的打包数据以及所述模型包测试的第二测试数据；
[0013]存储所述转换数据、所述第一测试数据、所述打包数据以及所述第二测试数据。
[0014]进一步地，所述根据所述硬件信息对所述待处理网络模型进行模型转换处理，得到能够在所述目标硬件平台上运行的目标模型，包括：
[0015]确定与所述硬件信息相匹配的模型转换算法；
[0016]根据所述模型转换算法，将所述待处理网络模型转换为能够在所述目标硬件平台上运行的目标模型。
[0017]进一步地，所述方法还包括：
[0018]获取用于测试所述目标模型的测试集；
[0019]通过所述测试集对所述目标模型进行模型测试，得到第一测试结果；
[0020]输出所述第一测试结果；
[0021]判断是否接收到用户根据所述第一测试结果输入的测试通过指令；
[0022]如果是，则执行所述的对所述目标模型进行打包处理得到模型包。
[0023]进一步地，所述对所述目标模型进行打包处理得到模型包，包括：
[0024]确定所述目标模型的前处理部分代码以及后处理部分代码；
[0025]将所述目标模型的模型文件、所述前处理部分代码以及所述后处理部分代码打包成一个单独的包，得到模型包。
[0026]进一步地，所述对所述模型包进行模型包测试，得到能够正式发布使用的目标模型包，包括：
[0027]获取用于测试所述模型包的测试文件；
[0028]通过所述测试文件对所述模型包进行测试，得到第二测试结果；
[0029]输出所述第二测试结果，并判断是否接收到用户针对所述第二测试结果输入的确认无误指令；
[0030]如果是，则将所述模型包确定为能够正式发布使用的目标模型包。
[0031]本申请实施例第二方面提供了一种模型处理装置，所述模型处理装置包括：
[0032]获取单元，用于获取待处理网络模型和目标硬件平台的硬件信息；
[0033]转换单元，用于根据所述硬件信息对所述待处理网络模型进行模型转换处理，得到能够在所述目标硬件平台上运行的目标模型；
[0034]打包单元，用于当所述目标模型的通过模型测试时，对所述目标模型进行打包处理得到模型包；
[0035]模型包测试单元，用于对所述模型包进行模型包测试，得到能够正式发布使用的目标模型包；
[0036]存储单元，用于存储所述目标模型包。
[0037]在上述实现过程中，获取单元先获取待处理网络模型和目标硬件平台的硬件信息；转换单元对待处理网络模型进行模型转换处理，得到能够在目标硬件平台上运行的目标模型；打包单元在目标模型的通过模型测试时，对目标模型进行打包处理得到模型包；模型包测试单元再对模型包进行模型包测试，得到能够正式发布使用的目标模型包；最后存储单元存储目标模型包，能够自动对模型进行适配和优化处理，处理效率高，准确性高。
[0038]进一步地，所述获取单元，还用于获取所述模型转换处理时的转换数据、所述模型测试的第一测试数据、所述对所述目标模型进行打包处理时的打包数据以及所述模型包测试的第二测试数据；
[0039]所述存储单元，还用于存储所述转换数据、所述第一测试数据、所述打包数据以及所述第二测试数据。
[0040]本申请实施例第三方面提供了一种电子设备，包括存储器以及处理器，所述存储器用于存储计算机程序，所述处理器运行所述计算机程序以使所述电子设备执行本申请实施例第一方面中任一项所述的模型处理方法。
[0041]本申请实施例第四方面提供了一种计算机可读存储介质，其存储有计算机程序指
令，所述计算机程序指令被一处理器读取并运行时，执行本申请实施例第一方面中任一项所述的模型处理方法。
附图说明
[0042]为了更清楚地说明本申请实施例的技术方案，下面将对本申请实施例中所需要使用的附图作简单地介绍，应当理解，以下附图仅示出了本申请的某些实施例，因此不应被看作是对范围的限定，对于本领域普通技术人员来讲，在不付出创造性劳动的前提下，还可以根据这些附图获得其他相关的附图。
[0043]图1为本申请实施例提供的一种模型处理方法的流程示意图；
[0044]图2为本申请实施例提供的另一种模型处理方法的流程示意图；
[0045]图3为本申请实施例提供的一种模型处理装置的结构示意图；
[0046]图4为本申请实施例提供的另一种模型处理装置的结构示意图。
具体实施方式
[0047]下面将结合本申请实施例中的附图，对本申请实施例中的技术方案进行描述。
[0048]应注意到：相似的标号和字母在下面的附图中表示类似项，因此，一旦某一项在一个附图中被定义，则在随后的附图中不需要对其进行进一步定义和解释。同时，在本申请的描述中，术语“第一”、“第二”等仅用于区分描述，而不能理解为指示或暗示相对重要性。
[0049]实施例1
本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种模型处理方法，其特征在于，包括：获取待处理网络模型和目标硬件平台的硬件信息；根据所述硬件信息对所述待处理网络模型进行模型转换处理，得到能够在所述目标硬件平台上运行的目标模型；当所述目标模型的通过模型测试时，对所述目标模型进行打包处理得到模型包；对所述模型包进行模型包测试，得到能够正式发布使用的目标模型包；存储所述目标模型包。2.根据权利要求1所述的模型处理方法，其特征在于，所述方法还包括：获取所述模型转换处理时的转换数据、所述模型测试的第一测试数据、对所述目标模型进行打包处理时的打包数据以及所述模型包测试的第二测试数据；存储所述转换数据、所述第一测试数据、所述打包数据以及所述第二测试数据。3.根据权利要求1所述的模型处理方法，其特征在于，所述根据所述硬件信息对所述待处理网络模型进行模型转换处理，得到能够在所述目标硬件平台上运行的目标模型，包括：确定与所述硬件信息相匹配的模型转换算法；根据所述模型转换算法，将所述待处理网络模型转换为能够在所述目标硬件平台上运行的目标模型。4.根据权利要求1所述的模型处理方法，其特征在于，所述方法还包括：获取用于测试所述目标模型的测试集；通过所述测试集对所述目标模型进行模型测试，得到第一测试结果；输出所述第一测试结果；判断是否接收到用户根据所述第一测试结果输入的测试通过指令；如果是，则执行所述的对所述目标模型进行打包处理得到模型包。5.根据权利要求1所述的模型处理方法，其特征在于，所述对所述目标模型进行打包处理得到模型包，包括：确定所述目标模型的前处理部分代码以及后处理部分代码；将所述目标模型的模型文件、所述前处理部分代码以及所述后处理部分代码打包成一个单独的包，得到模型包。6....

【专利技术属性】
技术研发人员：周伟，张发恩，周创，丁昂，刘亚萍，
申请(专利权)人：创新奇智北京科技有限公司，
类型：发明
国别省市：