# model_watermark
# 暂时弃用，请访问classification-main项目的demo分支，master分支为华科大提供的初始源码

AI模型水印代码
## 白盒水印
白盒水印已加载权重的模型作为参数，通过获取内部权重，添加特殊惩罚项来实现标签嵌入

## 黑盒水印
黑盒水印不修改模型结构，只修改训练集、验证集，通过对训练集、验证集进行处理，通过模型训练，实现将标签嵌入

## 功能说明
1. 模型水印功能初始化（白盒，集成方式：提供调用库，代码嵌入集成）
2. 模型训练损失函数修改（白盒，集成方式：提供调用库，代码嵌入集成）
3. 提供模型水印提取功能（白盒，集成方式：提供调用库，代码嵌入集成）
4. 提供数据集标签处理功能：处理训练集和验证集部分图片添加水印，并添加对应的标签信息（黑盒，集成方式：提供处理脚本，远程接口调用，调用库【暂不明确】）
5. 从数据集提取标签功能（黑盒，集成方式：远程接口调用【暂不明确】，调用库）
6. 提供密码标签获取功能（对接密码机，集成方式：远程接口调用，调用库）
7. 提供密码标签验证功能（对接密码机，集成方式：远程接口调用，调用库）