소스 검색

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

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)
     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
     input_name = session.get_inputs()[0].name