diff --git a/vidaug/augmentors/intensity.py b/vidaug/augmentors/intensity.py index e6df743..c3f4dd5 100644 --- a/vidaug/augmentors/intensity.py +++ b/vidaug/augmentors/intensity.py @@ -129,7 +129,7 @@ def __call__(self, clip): data_final = [] for i in range(len(clip)): - img = clip[i].astype(np.float) + img = clip[i].astype(np.float64) img_shape = img.shape noise = np.random.randint(self.ratio, size=img_shape) img = np.where(noise == 0, 0, img) @@ -159,7 +159,7 @@ def __call__(self, clip): data_final = [] for i in range(len(clip)): - img = clip[i].astype(np.float) + img = clip[i].astype(np.float64) img_shape = img.shape noise = np.random.randint(self.ratio, size=img_shape) img = np.where(noise == 0, 255, img)