apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

build noise by pixels

This commit is contained in:
Sean Sube 2023-01-14 16:14:37 -06:00
parent 4233c34813
commit 6fded9de2d
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
api/onnx_web/image.py
View File

@ -28,8 +28,16 @@ def blend_source_histogram(source_image: Image, dims: Tuple[int, int]) -> Tuple[
noise_g = random.choice(256, p=np.divide(np.copy(hist_g), np.sum(hist_g)), size=size)
noise_b = random.choice(256, p=np.divide(np.copy(hist_b), np.sum(hist_b)), size=size)
noise_pixels = np.dstack((noise_r, noise_g, noise_b))
noise = Image.fromarray(np.reshape(noise_pixels, (width * 3, height)))
noise = Image.new('RGB', (width, height))
for x in range(width):
for y in range(height):
i = x * y
noise.putpixel((x, y), (
noise_r[i],
noise_g[i],
noise_b[i]
))
return noise