liyan
0ed1cb4709
进行修改
|
4 months ago | |
|---|---|---|
| blind_watermark | 4 months ago | |
| block | 4 months ago | |
| checkpoints | 4 months ago | |
| dataset | 4 months ago | |
| documents | 4 months ago | |
| model | 4 months ago | |
| tool | 4 months ago | |
| weight | 4 months ago | |
| .gitignore | 4 months ago | |
| README.md | 4 months ago | |
| dataset_process.py | 4 months ago | |
| predict.py | 4 months ago | |
| requirement | 4 months ago | |
| run.py | 4 months ago |
README.md
model_watermark
暂时弃用，请访问classification-main项目的demo分支，master分支为华科大提供的初始源码
AI模型水印代码
白盒水印
白盒水印已加载权重的模型作为参数，通过获取内部权重，添加特殊惩罚项来实现标签嵌入
黑盒水印
黑盒水印不修改模型结构，只修改训练集、验证集，通过对训练集、验证集进行处理，通过模型训练，实现将标签嵌入
功能说明
- 模型水印功能初始化（白盒，集成方式：提供调用库，代码嵌入集成）
- 模型训练损失函数修改（白盒，集成方式：提供调用库，代码嵌入集成）
- 提供模型水印提取功能（白盒，集成方式：提供调用库，代码嵌入集成）
- 提供数据集标签处理功能：处理训练集和验证集部分图片添加水印，并添加对应的标签信息（黑盒，集成方式：提供处理脚本，远程接口调用，调用库【暂不明确】）
- 从数据集提取标签功能（黑盒，集成方式：远程接口调用【暂不明确】，调用库）
- 提供密码标签获取功能（对接密码机，集成方式：远程接口调用，调用库）
- 提供密码标签验证功能（对接密码机，集成方式：远程接口调用，调用库）