From 6d08be373ad94be2f418792584c51de9bb4a965e Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Sun, 25 Apr 2021 13:28:52 +0800 Subject: [PATCH] fix(data): fix the sampling scale in Lighting GitOrigin-RevId: d0a1ca50e2dafcc9923ff845815e882c3a08f2e6 --- imperative/python/megengine/data/transform/vision/transform.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/imperative/python/megengine/data/transform/vision/transform.py b/imperative/python/megengine/data/transform/vision/transform.py index c39bd37d..d567669c 100644 --- a/imperative/python/megengine/data/transform/vision/transform.py +++ b/imperative/python/megengine/data/transform/vision/transform.py @@ -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)