参考了

神经网络模型如何加密 (Python) - 哔哩哔哩

 实现方法

一、安装工具

安装cryptography模块

pip install cryptography -i https://pypi.tuna.tsinghua.edu.cn/simple

二、测试加密

加密代码chenjiami.py 

1、生成加密key并保存

from cryptography.fernet import Fernet

#-----------------------------------------------------------
#(1)生成的秘钥,注意保存,第一次
key = Fernet.generate_key()
print(key)
file = open('D:/00testjiami/key.data', 'wb')
file.write(key)
file.close()

2、读取key

fr = open("D:/00testjiami/key.data", 'rb')
chenkey = fr.read()
print(chenkey)

3、将文件加密并保存

#(3)加密文件
fpwd = Fernet(chenkey)
file_orign =open("D:/00testjiami/xxxxx.model", 'rb')
file_content= file_orign.read()
print("chenread")
encrypted_content = fpwd.encrypt(file_content)

fsave =open("D:/chenyolo/00testjiami/chenmodel.data", 'wb')
fsave.write(encrypted_content)
print("jia mi 完成")

4、解密文件并获取内容

#(4)解密文件
fpwd = Fernet(chenkey)
file_jiemi_orign =open("D:/chenyolo/00testjiami/chenmodel.data", 'rb')
file_jiemi_content= bytearray(fpwd.decrypt(file_jiemi_orign.read()))

fsave_jiemi =open("D:/chenyolo/00testjiami/chenmodeljiemi.pt", 'wb')
fsave_jiemi.write(file_jiemi_content)
print("jie mi 完成")

Logo

华为开发者空间,是为全球开发者打造的专属开发空间,汇聚了华为优质开发资源及工具,致力于让每一位开发者拥有一台云主机,基于华为根生态开发、创新。

更多推荐