修改图像分类模型黑盒水印验证流程

watermark_verify/verify_tool.py

@@ -123,7 +123,7 @@ def predict_and_detect(image_path, model_filename, qrcode_positions_file, input_
     img, ratio = preproc(origin_img, input_shape)
 
     # 解析标签文件
-    x_center, y_center, w, h, cls = parse_qrcode_label_file.load_watermark_info(qrcode_positions_file, image_path)
+    _, _, _, _, cls = parse_qrcode_label_file.load_watermark_info(qrcode_positions_file, image_path)
 
     # 执行推理
     input_name = session.get_inputs()[0].name