一种模型文件保护和读取方法、装置及相关介质制造方法及图纸

本发明专利技术公开了一种模型文件保护和读取方法、装置及相关介质，该方法包括：对模型文件进行拆分处理，得到头部区和非头部区；对所述头部区进行加密处理得到加密块并移动到所述非头部区的尾部，得到加密区；生成与所述头部区相同大小的头部块，并将所述头部区剔除以及将所述头部块填充至所述头部区的位置；根据所述头部区的大小对所述模型文件进行内存映射，得到所述加密区的位置；对所述加密区进行解密处理，并将解密后的明文内容回写到所述头部块的位置。本发明专利技术通过生成与所述头部区相同大小的头部块来填充所述模型文件的头部区位置，并在读取时，以内存映射方式对所述加密区进行解密回写到所述头部块的位置；如此，提高了所述模型文件的安全性。型文件的安全性。型文件的安全性。

【技术实现步骤摘要】
一种模型文件保护和读取方法、装置及相关介质


[0001]本专利技术涉及文件保护
，特别涉及一种模型文件保护和读取方法、装置及相关介质。

技术介绍

[0002]当前，AI技术在使用时，通常是一种模型推理框架和一个训练好的模型。对于模型文件，如果直接存放在软件中，其他人可以轻松获取到模型文件；获取到模型文件之后，不但能直接使用，还能利用网络结构重新训练进行模型文件改良。因此，需要将模型文件进行加密保护，使得模型文件不被非法盗取。

技术实现思路

[0003]本专利技术实施例提供了一种模型文件保护和读取方法、装置及相关介质，旨在解决模型文件不够安全，在使用过程中得不到保护的问题。
[0004]第一方面，本专利技术实施例提供了一种模型文件保护和读取方法，包括：
[0005]对模型文件进行拆分处理，得到所述模型文件的头部区和非头部区；
[0006]对所述头部区进行加密处理得到加密块，并将所述加密块以追加的方式移动到所述非头部区的尾部，得到加密区；
[0007]生成与所述头部区相同大小的头部块，并将所述头部区剔除以及将所述头部块填充至所述头部区的位置，完成所述模型文件的文件加密；
[0008]读取所述头部块得到所述头部区的大小，并根据所述头部区的大小对所述模型文件进行内存映射，得到所述加密区的位置；
[0009]对所述加密区进行解密处理，并将所述头部块剔除以及将解密后的明文内容回写到所述头部块的位置，完成所述模型文件的文件解密。
[0010]第二方面，本专利技术实施例提供了一种模型文件保护和读取装置，包括：
[0011]数据拆分单元，用于对模型文件进行拆分处理，得到所述模型文件的头部区和非头部区；
[0012]数据加密单元，用于对所述头部区进行加密处理得到加密块，并将所述加密块以追加的方式移动到所述非头部区的尾部，得到加密区；
[0013]数据替换单元，用于生成与所述头部区相同大小的头部块，并将所述头部区剔除以及将所述头部块填充至所述头部区的位置，完成所述模型文件的文件加密；
[0014]数据映射单元，用于读取所述头部块得到所述头部区的大小，并根据所述头部区的大小对所述模型文件进行内存映射，得到所述加密区的位置；
[0015]数据解密单元，用于对所述加密区进行解密处理，并将所述头部块剔除以及将解密后的明文内容回写到所述头部块的位置，完成所述模型文件的文件解密。
[0016]第三方面，本专利技术实施例提供了一种计算机设备，包括存储器、处理器及存储在所述存储器上并可在所述处理器上运行的计算机程序，所述处理器执行所述计算机程序时实
现所述第一方面的模型文件保护和读取方法。
[0017]第四方面，本专利技术实施例提供了一种计算机可读存储介质，其中，所述计算机可读存储介质上存储有计算机程序，所述计算机程序被处理器执行时实现所述第一方面的模型文件保护和读取方法。
[0018]本专利技术实施例提供一种模型文件保护和读取方法，该方法包括：对模型文件进行拆分处理，得到头部区和非头部区；对所述头部区进行加密处理得到加密块并移动到所述非头部区的尾部，得到加密区；生成与所述头部区相同大小的头部块，并将所述头部区剔除以及将所述头部块填充至所述头部区的位置；根据所述头部区的大小对所述模型文件进行内存映射，得到所述加密区的位置；对所述加密区进行解密处理，并将解密后的明文内容回写到所述头部块的位置。本专利技术通过生成与所述头部区相同大小的头部块来填充所述模型文件的头部区位置，并在读取时，以内存映射方式对所述加密区进行解密回写到所述头部块的位置；如此，提高了所述模型文件的安全性。
[0019]本专利技术实施例还提供一种模型文件保护和读取装置、计算机设备和存储介质，同样具有上述有益效果。
附图说明
[0020]为了更清楚地说明本专利技术实施例技术方案，下面将对实施例描述中所需要使用的附图作简单地介绍，显而易见地，下面描述中的附图是本专利技术的一些实施例，对于本领域普通技术人员来讲，在不付出创造性劳动的前提下，还可以根据这些附图获得其他的附图。
[0021]图1为本专利技术实施例提供的一种模型文件保护和读取方法的流程示意图；
[0022]图2为本专利技术实施例提供的一种模型文件保护和读取装置的示意性框图。
具体实施方式
[0023]下面将结合本专利技术实施例中的附图，对本专利技术实施例中的技术方案进行清楚、完整地描述，显然，所描述的实施例是本专利技术一部分实施例，而不是全部的实施例。基于本专利技术中的实施例，本领域普通技术人员在没有做出创造性劳动前提下所获得的所有其他实施例，都属于本专利技术保护的范围。
[0024]应当理解，当在本说明书和所附权利要求书中使用时，术语“包括”和“包含”指示所描述特征、整体、步骤、操作、元素和/或组件的存在，但并不排除一个或多个其它特征、整体、步骤、操作、元素、组件和/或其集合的存在或添加。
[0025]还应当理解，在此本专利技术说明书中所使用的术语仅仅是出于描述特定实施例的目的而并不意在限制本专利技术。如在本专利技术说明书和所附权利要求书中所使用的那样，除非上下文清楚地指明其它情况，否则单数形式的“一”、“一个”及“该”意在包括复数形式。
[0026]还应当进一步理解，在本专利技术说明书和所附权利要求书中使用的术语“和/或”是指相关联列出的项中的一个或多个的任何组合以及所有可能组合，并且包括这些组合。
[0027]下面请参见图1，图1为本专利技术实施例提供的一种模型文件保护和读取方法的流程示意图，具体包括：步骤S101～S105。
[0028]S101、对模型文件进行拆分处理，得到所述模型文件的头部区和非头部区；
[0029]S102、对所述头部区进行加密处理得到加密块，并将所述加密块以追加的方式移
动到所述非头部区的尾部，得到加密区；
[0030]S103、生成与所述头部区相同大小的头部块，并将所述头部区剔除以及将所述头部块填充至所述头部区的位置，完成所述模型文件的文件加密；
[0031]S104、读取所述头部块得到所述头部区的大小，并根据所述头部区的大小对所述模型文件进行内存映射，得到所述加密区的位置；
[0032]S105、对所述加密区进行解密处理，并将所述头部块剔除以及将解密后的明文内容回写到所述头部块的位置，完成所述模型文件的文件解密。
[0033]在步骤S101中，将整个所述模型文件拆分为两部分，一部分是需要加密的所述头部区，另一部分是不加密的所述非头部区，方便进行区分。
[0034]在步骤S102中，对所述头部区进行加密处理，即把所述头部区，比如256字节，加密后追加到所述非头部区的尾部。需要注意的是，追加的方式可以有多种设置，可以追加到所述非头部区的尾部，也可以先写一个生成的新文件头后，然后写加密后的头部内容(所述加密快)，然后再写剩下的未加密内容(所述非头部区)；具体可以根据实际情况进行选择。
[0035]在步骤S103中，生成一个新本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种模型文件保护和读取方法，其特征在于，包括：对模型文件进行拆分处理，得到所述模型文件的头部区和非头部区；对所述头部区进行加密处理得到加密块，并将所述加密块以追加的方式移动到所述非头部区的尾部，得到加密区；生成与所述头部区相同大小的头部块，并将所述头部区剔除以及将所述头部块填充至所述头部区的位置，完成所述模型文件的文件加密；读取所述头部块得到所述头部区的大小，并根据所述头部区的大小对所述模型文件进行内存映射，得到所述加密区的位置；对所述加密区进行解密处理，并将所述头部块剔除以及将解密后的明文内容回写到所述头部块的位置，完成所述模型文件的文件解密。2.根据权利要求1所述的模型文件保护和读取方法，其特征在于，所述模型文件为基于ProtocolBuffer的二进制文件。3.根据权利要求1所述的模型文件保护和读取方法，其特征在于，所述头部块的内容包括文件类型、头部块尺寸、加密类型版本以及可选信息。4.根据权利要求1所述的模型文件保护和读取方法，其特征在于，所述将解密后的明文内容回写到所述头部块的位置，包括：以内存映射的方式，将解密后的明文内容回写到所述头部块的位置。5.根据权利要求3所述的模型文件保护和读取方法，其特征在于，所述文件类型、头部块尺寸和加密类型版本均为4字节大小。6.根据权利要求1所述的模型文件保护和读取方法，其特征在于，所述对所述加密区进行解密处理，包括：利用AES算...

【专利技术属性】
技术研发人员：王功华，
申请(专利权)人：深圳软牛科技有限公司，
类型：发明
国别省市：