@@ -1,3 +1,8 @@
# model_watermark
-AI模型水印代码
+AI模型水印代码
+## 白盒水印
+白盒水印已加载权重的模型作为参数，通过获取内部权重，添加特殊惩罚项来实现标签嵌入
+
+## 黑盒水印
+黑盒水印不修改模型结构，只修改训练集、验证集，通过对训练集、验证集进行处理，通过模型训练，实现将标签嵌入
