|
|
@@ -9,9 +9,7 @@ import qrcode
|
|
|
from qrcode.main import QRCode
|
|
|
from PIL import Image
|
|
|
|
|
|
-from watermark_generate.tools import logger_tool
|
|
|
-
|
|
|
-logger = logger_tool.logger
|
|
|
+from watermark_generate import logger
|
|
|
|
|
|
|
|
|
def random_partition(key, parts):
|
|
|
@@ -67,7 +65,7 @@ def generate_qrcodes(key: str, watermarking_dir='./dataset/watermarking', partit
|
|
|
for f in qr_files:
|
|
|
qr_path = os.path.join(watermarking_dir, f)
|
|
|
img = Image.open(qr_path)
|
|
|
- decode = detect_qrcode_in_bbox(qr_path,[0,0,img.width, img.height])
|
|
|
+ decode = detect_qrcode_in_bbox(qr_path, [0, 0, img.width, img.height])
|
|
|
if decode is None:
|
|
|
return False
|
|
|
reconstructed_key = reconstructed_key + decode
|
