修改日志生成

watermark_generate/tools/gen_qrcodes.py

@@ -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