Spaces:
Running
on
Zero
Running
on
Zero
File size: 667 Bytes
319886d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
import numpy as np
from PIL import Image
def pixelate(img, severity=1):
"""
Pixelate.
severity=[1, 2, 3, 4, 5] corresponding to sigma=[0.5, 0.4, 0.3, 0.25, 0.2].
severity mainly refer to Imagecorruptions.
@param img: Input image, H x W x 3, value range [0, 255]
@param severity: Severity of distortion, [1, 5]
@return: Degraded image, H x W x 3, value range [0, 255]
"""
c = [0.5, 0.4, 0.3, 0.25, 0.2][severity - 1]
h, w = np.array(img).shape[:2]
img = Image.fromarray(img)
img = img.resize((int(w * c), int(h * c)), Image.BOX)
img = img.resize((w, h), Image.NEAREST)
return np.array(img).astype(np.uint8)
|