瀏覽代碼

delete +0.5 for contrast

FlyingQianMM 4 年之前
父節點
當前提交
fda3785d33
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      paddlex/cv/transforms/ops.py

+ 1 - 1
paddlex/cv/transforms/ops.py

@@ -159,7 +159,7 @@ def saturation(im, saturation_lower, saturation_upper, is_rgb=False):
 
 
 def contrast(im, contrast_lower, contrast_upper):
 def contrast(im, contrast_lower, contrast_upper):
     delta = np.random.uniform(contrast_lower, contrast_upper)
     delta = np.random.uniform(contrast_lower, contrast_upper)
-    im_mean = im.mean() + 0.5
+    im_mean = im.mean()
     im1 = np.full_like(im, im_mean)
     im1 = np.full_like(im, im_mean)
     im *= delta
     im *= delta
     im += im1 * (1 - delta)
     im += im1 * (1 - delta)