瀏覽代碼

修改YOLOX模型处理代码

liyan 9 月之前
父節點
當前提交
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)