Spaces:
Runtime error
Runtime error
File size: 2,309 Bytes
f2dbf59 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 |
from PIL import Image
import random
import numpy as np
def _makeGrayNoise(width, height, power):
buffer = np.zeros([height, width], dtype=int)
for y in range(0, height):
for x in range(0, width):
buffer[y, x] = random.gauss(128, power)
buffer = buffer.clip(0, 255)
return Image.fromarray(buffer.astype(dtype=np.uint8))
def _makeRgbNoise(width, height, power, saturation):
buffer = np.zeros([height, width, 3], dtype=int)
intens_power = power * (1.0 - saturation)
for y in range(0, height):
for x in range(0, width):
intens = random.gauss(128, intens_power)
buffer[y, x, 0] = random.gauss(0, power) * saturation + intens
buffer[y, x, 1] = random.gauss(0, power) * saturation + intens
buffer[y, x, 2] = random.gauss(0, power) * saturation + intens
buffer = buffer.clip(0, 255)
return Image.fromarray(buffer.astype(dtype=np.uint8))
def grainGen(width, height, grain_size, power, saturation, seed = 1):
# A grain_size of 1 means the noise buffer will be made 1:1
# A grain_size of 2 means the noise buffer will be resampled 1:2
noise_width = int(width / grain_size)
noise_height = int(height / grain_size)
random.seed(seed)
if saturation < 0.0:
print("Making B/W grain, width: %d, height: %d, grain-size: %s, power: %s, seed: %d" % (
noise_width, noise_height, str(grain_size), str(power), seed))
img = _makeGrayNoise(noise_width, noise_height, power)
else:
print("Making RGB grain, width: %d, height: %d, saturation: %s, grain-size: %s, power: %s, seed: %d" % (
noise_width, noise_height, str(saturation), str(grain_size), str(power), seed))
img = _makeRgbNoise(noise_width, noise_height, power, saturation)
# Resample
if grain_size != 1.0:
img = img.resize((width, height), resample = Image.LANCZOS)
return img
if __name__ == "__main__":
import sys
if len(sys.argv) == 8:
width = int(sys.argv[2])
height = int(sys.argv[3])
grain_size = float(sys.argv[4])
power = float(sys.argv[5])
sat = float(sys.argv[6])
seed = int(sys.argv[7])
out = grainGen(width, height, grain_size, power, sat, seed)
out.save(sys.argv[1]) |