瀏覽代碼

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

liyan 11 月之前
父節點
當前提交
8f1c0a6c44
共有 1 個文件被更改,包括 1 次插入1 次删除
  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