浏览代码

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

liyan 11 月之前
父节点
当前提交
b01fd63ec0
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      watermark_verify/verify_tool.py

+ 1 - 1
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