模型黑盒水印提取，白盒水印提取，标签验证，提供python接口的软件包
liyan
9a2e4c1329
修改Faster-RCNN模型黑盒水印检测流程,解决om权重推理结果与onnx推理结果不一致的问题
|
hai 1 semana | |
|---|---|---|
| tests | hai 1 semana | |
| watermark_verify | hai 1 semana | |
| .gitignore | hai 3 meses | |
| MANIFEST.in | hai 3 meses | |
| README.md | hai 3 meses | |
| setup.py | hai 3 meses |
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