소스 검색

修改YOLOX模型处理代码

liyan 10 달 전
부모
커밋
e1f83f8d5c
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      watermark_generate/deals/yolox_pytorch_black_embed.py

+ 1 - 1
watermark_generate/deals/yolox_pytorch_black_embed.py

@@ -78,7 +78,7 @@ f"""
         deal_flag, secret_index = find_index_in_parts(self.parts, index)
         if deal_flag:
             # Step 2: Add watermark to the image and get the updated label
-            secret = self.secret_parts[index]
+            secret = self.secret_parts[secret_index]
             img_wm, watermark_annotation = add_watermark_to_image(img, secret, secret_index)
             # 二维码提取测试
             decoded_text, _ = detect_and_decode_qr_code(img_wm)