|
|
@@ -3,11 +3,18 @@
|
|
|
- [x] 密码标签生成
|
|
|
- [x] 密码标签自检(测试)
|
|
|
- [x] 密码标签验证
|
|
|
-- [x] 图像嵌入密码标签
|
|
|
-- [x] 图像提取密码标签
|
|
|
-- [x] 图像处理功能自检
|
|
|
- [x] 日志模板
|
|
|
- [x] 日志记录
|
|
|
+- [x] 密码标签分段并生成二维码
|
|
|
+- [x] 处理训练集
|
|
|
+- [x] 生成触发集
|
|
|
+~~图像嵌入密码标签~~
|
|
|
+~~图像提取密码标签~~
|
|
|
+~~图像处理功能自检~~
|
|
|
+- [ ] 黑盒水印验证流程
|
|
|
+- [ ] 验证模型文件与触发集图片签名值（http,接口）
|
|
|
+- [ ] 对比预测准确率（http,接口）
|
|
|
+- [ ] 从触发集提取组装密码标签，进行验签，返回验签结果（http,接口）
|
|
|
|
|
|
## 分支说明
|
|
|
- `master`分支只包含项目打包配置和密码标签获取，数据集嵌入黑盒水印源码
|
|
|
@@ -465,8 +472,9 @@ file: file://C:\Users\hikari\Desktop\test_embed.jpg
|
|
|
|»» date|string(date-time)|true|none|创建日期|none|
|
|
|
|
|
|
## 代码集成
|
|
|
+
|
|
|
```python
|
|
|
-from watermark_generate.tools.dataset_process import dataset_embed_label, process_dataset_label
|
|
|
+from watermark_generate.tools.dataset_process import dataset_embed_label, process_train_dataset
|
|
|
from watermark_generate.tools.picture_watermark import extract
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
@@ -474,13 +482,13 @@ if __name__ == '__main__':
|
|
|
src_img_path = '../VOC2007/test/'
|
|
|
dst_img_path = '../VOC2007/test_wm'
|
|
|
label_path = '../VOC2007/test_labels'
|
|
|
-
|
|
|
+
|
|
|
# 图片嵌入密码标签
|
|
|
dataset_embed_label(label, src_img_path, dst_img_path)
|
|
|
|
|
|
# 数据集处理测试
|
|
|
- process_dataset_label(dst_img_path, label_path, percentage=5)
|
|
|
-
|
|
|
+ process_train_dataset(dst_img_path, label_path, percentage=5)
|
|
|
+
|
|
|
# 水印提取测试
|
|
|
secret = extract('../VOC2007/test_wm/000044.jpg', secret_len=len(label))
|
|
|
print(label == secret)
|
