apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

crop count layer

This commit is contained in:
Sean Sube 2023-06-03 14:02:58 -05:00
parent 3bdf3692f4
commit e64be674ea
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/onnx_web/chain/utils.py
View File

@ -126,7 +126,7 @@ def process_tile_grid(
] += equalized[0 : scaled_bottom - scaled_top, 0 : scaled_right - scaled_left, :]
count[
scaled_top : scaled_bottom, scaled_left : scaled_right, :
] += np.repeat(mask[:, :, np.newaxis], 3, axis=2)
] += np.repeat(mask[0 : scaled_bottom - scaled_top, 0 : scaled_right - scaled_left, np.newaxis], 3, axis=2)
pixels = np.where(count > 0, value / count, value)
return Image.fromarray(pixels)