Browse Source

修改模型工程文件处理函数

liyan 11 months ago
parent
commit
8f1c0a6c44
1 changed files with 1 additions and 1 deletions
  1. 1 1
      watermark_generate/deals/yolox_pytorch_black_embed.py

+ 1 - 1
watermark_generate/deals/yolox_pytorch_black_embed.py

@@ -151,7 +151,7 @@ def add_watermark_to_image(img, watermark_label, watermark_class_id):
     x_end = x_start + qr_w
     y_end = y_start + qr_h
     qr_img_cropped = qr_img[:y_end - y_start, :x_end - x_start]
-    img[y_start:y_end, x_start:x_end] = cv2.addWeighted(img[y_start:y_end, x_start:x_end], 0.4, qr_img_cropped, 0.6, 0)
+    img[y_start:y_end, x_start:x_end] = cv2.addWeighted(img[y_start:y_end, x_start:x_end], 0, qr_img_cropped, 1, 0)
     x_center = (x_start + x_end) / 2 / img_w
     y_center = (y_start + y_end) / 2 / img_h
     w = qr_w / img_w