Browse Source

fix(data): fix the sampling scale in Lighting

GitOrigin-RevId: d0a1ca50e2
release-1.4
Megvii Engine Team 4 years ago
parent
commit
6d08be373a
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      imperative/python/megengine/data/transform/vision/transform.py

+ 1
- 1
imperative/python/megengine/data/transform/vision/transform.py View File

@@ -1043,7 +1043,7 @@ class Lighting(VisionTransform):

dtype = image.dtype
image = image.astype(np.float32)
alpha = np.random.normal(scale=self.scale, size=3)
alpha = np.random.normal(scale=self.scale * 255, size=3)
image = image + self.eigvec.dot(alpha * self.eigval)
return image.clip(0, 255).astype(dtype)



Loading…
Cancel
Save