技术特征：
1.一种基于aes算法的pytorch模型加密方法，其特征在于，采用aes加密算法对模型二进制流进行加密，步骤如下：以二进制格式打开pytorch模型文件转换成二进制流用于读写，设置aes密钥k，设置加密模式mode为对称加密，把已转换成二进制流的pytorch模型文件输入到aes加密函数中，生成模型加密文件。2.根据权利要求1所述的基于aes算法的pytorch模型加密方法，其特征在于：还包括采用aes加密算法对模型加密文件进行解密，步骤如下：设置与加密时相同的aes密钥k及加密模式mode为对称加密，把模型加密文件输入到aes解密函数，生成解密后的pytorch模型二进制流文件，再以pytorch框架中模型加载以流的方式读取解密后的模型二进制文件，进行网络的前向推理判断。3.一种基于aes算法的pytorch模型加密系统，其特征在于：包括加密单元，所述加密单元用于采用aes加密算法对模型二进制流进行加密，步骤如下：以二进制格式打开pytorch模型文件转换成二进制流用于读写，设置aes密钥k，设置加密模式mode为对称加密，把已转换成二进制流的pytorch模型文件输入到aes加密函数中，生成模型加密文件。4.根据权利要求3所述的一种基于aes算法的pytorch模型加密系统，其特征在于，还包括解密单元，所述解密单元采用aes加密算法对模型加密文件进行解密，步骤如下：设置与加密时相同的aes密钥k及加密模式mode为对称加密，把模型加密文件输入到aes解密函数，生成解密后的pytorch模型二进制流文件，再以pytorch框架中模型加载以流的方式读取解密后的模型二进制文件，进行网络的前向推理判断。5.一种计算机可读存储介质，存储有计算机程序，所述计算机程序被处理器执行时，使得所述处理器执行如权利要求1或2所述方法的步骤。6.一种计算机设备，包括存储器和处理器，所述存储器存储有计算机程序，所述计算机程序被所述处理器执行时，使得所述处理器执行如权利要求1或2所述方法的步骤。
技术总结
本发明的一种基于AES算法的pytorch模型加密方法、系统及设备，其中方法包括采用AES加密算法对模型二进制流进行加密，步骤如下：以二进制格式打开pytorch模型文件转换成二进制流用于读写，设置AES密钥K，设置加密模式mode为对称加密，把已转换成二进制流的pytorch模型文件输入到AES加密函数中，生成模型加密文件。本发明的基于AES算法的pytorch模型加密方法涉及一种应用于pytorch深度学习框架训练的模型在离线部署时的加密方法，能够保护模型内部网络结构及参数隐私，以防被人挪用，或者泄露出重要、敏感信息。敏感信息。敏感信息。
技术研发人员：朱地博 朱文佳 骆乐乐 秦忱忱
受保护的技术使用者：安徽百诚慧通科技有限公司
技术研发日：2021.11.29
技术公布日：2022/3/1