Home Explore Help
Sign In
AIModelWatermark
/
model_watermark_detect
5
0
Fork 0
Files Issues 0 Pull Requests 0 Wiki

模型水印检测工具，使用不同分支区分各种模型水印检测

Tree: 92be7bcb51
Branches Tags
model_waterm...
ZIP TAR.GZ
liyan 92be7bcb51 初始化模型水印检测代码，完成基于pytorch、keras、tensorflow框架的图像分类模型白盒水印、黑盒水印检测 4 months ago
tests 92be7bcb51 初始化模型水印检测代码，完成基于pytorch、keras、tensorflow框架的图像分类模型白盒水印、黑盒水印检测 4 months ago
watermark_verify 92be7bcb51 初始化模型水印检测代码，完成基于pytorch、keras、tensorflow框架的图像分类模型白盒水印、黑盒水印检测 4 months ago
.gitignore 2cb77e1855 初始化项目结构 4 months ago
MANIFEST.in 2cb77e1855 初始化项目结构 4 months ago
README.md 0a06da4883 Revert "修改说明文档" 4 months ago
setup.py 2cb77e1855 初始化项目结构 4 months ago

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