模型黑盒水印提取,白盒水印提取,标签验证,提供python接口的软件包

liyan 8a5235b1af 新增性能测试脚本 2 hari lalu
tests 8a5235b1af 新增性能测试脚本 2 hari lalu
watermark_verify f034396d70 修改黑盒水印验证流程,新增性能测试脚本 3 hari lalu
.gitignore 767416cf50 添加git忽略文件 2 bulan lalu
MANIFEST.in d452db0dcf 修改项目忽略文件和日志目录 3 bulan lalu
README.md cc0a5b36fe 初始化工程目录 3 bulan lalu
setup.py e9204ba13c 添加工具类,添加项目依赖 2 bulan lalu

README.md

模型水印验证

提供模型训练嵌入白盒水印和从已经嵌入白盒水印的模型中提取水印的功能

分支说明

  • master分支包含公共代码,工具类
  • om分支在master分支基础上添加了对om格式进行黑盒水印验证流程
  • onnx分支在master分支基础上添加了对onnx格式权重文件进行白盒水印验证流程

模块打包

python setup.py sdist

项目目录会生成dist目录,其中watermark_codec-1.0.tar.gz即为发布包

安装模块

pip install watermark_codec-1.0.tar.gz