|
|
@@ -1,5 +1,4 @@
|
|
|
-from watermark_generate.tools.dataset_process import embed_label_to_image, process_train_dataset, \
|
|
|
- generate_trigger_dataset
|
|
|
+from watermark_generate.tools.dataset_process import process_train_dataset, generate_trigger_dataset
|
|
|
from watermark_generate.tools.gen_qrcodes import generate_qrcodes, detect_qrcode_in_bbox, extract_qrcode_from_image
|
|
|
from watermark_generate.tools.secret_func import get_secret, verify
|
|
|
|
|
|
@@ -18,14 +17,6 @@ def test_gen_qrcodes(secret):
|
|
|
print('生成成功')
|
|
|
|
|
|
|
|
|
-def test_embed_label_to_image():
|
|
|
- """
|
|
|
- 测试单张图片嵌入二维码
|
|
|
- """
|
|
|
- secret = 'ABCDEF123123'
|
|
|
- embed_label_to_image(secret=secret, img_path='./dataset/test.jpg')
|
|
|
-
|
|
|
-
|
|
|
if __name__ == '__main__':
|
|
|
# test_embed_label_to_image() # 测试单张图片嵌入密码标签二维码
|
|
|
src_img_path = './dataset/VOC2007/JPEGImages/'
|
|
|
@@ -46,4 +37,3 @@ if __name__ == '__main__':
|
|
|
# 测试数据集处理
|
|
|
process_train_dataset(watermarking_dir=watermark_gen_dir, src_img_dir=src_img_path, label_file_dir=label_path,
|
|
|
dst_img_dir=dst_img_dir)
|
|
|
-
|
