Merge pull request WenmuZhou#199 from morestart/master

fix RandomNoise class astype(im.dtype) to astype(data['img'].dtype)
paradoxjun · Sep 28, 2021 · 83e9c45 · 83e9c45
2 parents 31fa3ac + 6abd39d
commit 83e9c45
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchocr/datasets/det_modules/augment.py b/torchocr/datasets/det_modules/augment.py
@@ -26,7 +26,7 @@ def __call__(self, data: dict):
         """
         if random.random() > self.random_rate:
             return data
-        data['img'] = (random_noise(data['img'], mode='gaussian', clip=True) * 255).astype(im.dtype)
+        data['img'] = (random_noise(data['img'], mode='gaussian', clip=True) * 255).astype(data['img'].dtype)
         return data