apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): look up noise coordinates correctly

This commit is contained in:
Sean Sube 2023-01-16 07:49:25 -06:00
parent 604cdf39b0
commit 1283bc3d3f
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
api/onnx_web/image.py
View File

@ -10,6 +10,10 @@ from .utils import (
) )
def get_pixel_index(x: int, y: int, width: int) -> int:
return (y * width) + x
def mask_filter_none(mask_image: Image, dims: Point, origin: Point, fill='white') -> Image: def mask_filter_none(mask_image: Image, dims: Point, origin: Point, fill='white') -> Image:
width, height = dims width, height = dims
@ -93,7 +97,7 @@ def noise_source_uniform(source_image: Image, dims: Point, origin: Point) -> Ima
for x in range(width): for x in range(width):
for y in range(height): for y in range(height):
i = x * y i = get_pixel_index(x, y, width)
noise.putpixel((x, y), ( noise.putpixel((x, y), (
int(noise_r[i]), int(noise_r[i]),
int(noise_g[i]), int(noise_g[i]),
@ -115,7 +119,7 @@ def noise_source_normal(source_image: Image, dims: Point, origin: Point) -> Imag
for x in range(width): for x in range(width):
for y in range(height): for y in range(height):
i = x * y i = get_pixel_index(x, y, width)
noise.putpixel((x, y), ( noise.putpixel((x, y), (
int(noise_r[i]), int(noise_r[i]),
int(noise_g[i]), int(noise_g[i]),
@ -145,7 +149,7 @@ def noise_source_histogram(source_image: Image, dims: Point, origin: Point) -> I
for x in range(width): for x in range(width):
for y in range(height): for y in range(height):
i = x * y i = get_pixel_index(x, y, width)
noise.putpixel((x, y), ( noise.putpixel((x, y), (
noise_r[i], noise_r[i],
noise_g[i], noise_g[i],