apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix blend sources

This commit is contained in:
Sean Sube 2024-01-06 16:27:25 -06:00
parent c7154ffb27
commit 9e201fc94a
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/onnx_web/diffusers/run.py
View File

@ -497,7 +497,7 @@ def run_blend_pipeline(
tile_size = get_base_tile(params, size) tile_size = get_base_tile(params, size)
# resize mask to match source size # resize mask to match source size
stage_source = sources[1] stage_source = sources.pop()
stage_mask = mask.resize(stage_source.size, Image.Resampling.BILINEAR) stage_mask = mask.resize(stage_source.size, Image.Resampling.BILINEAR)
chain.stage( chain.stage(

11
api/scripts/test-release.py
View File

@ -564,10 +564,15 @@ def run_test(
passed = False passed = False
for i in range(len(results)): for i in range(len(results)):
result = results[i] result = results[i]
result.save(test_path(path.join("test-results", f"{test.name}-{i}.png"))) result_name = f"{test.name}-{i}.png"
result.save(test_path(path.join("test-results", result_name)))
ref_name = test_path(path.join("test-refs", f"{test.name}-{i}.png")) ref_name = test_path(path.join("test-refs", result_name))
ref = Image.open(ref_name) if path.exists(ref_name) else None if not path.exists(ref_name):
return TestResult.failed(test.name, f"no reference image for {result_name}")
ref = Image.open(ref_name)
logger.warning("comparing image %s to %s", result, ref)
mse = find_mse(result, ref) mse = find_mse(result, ref)
threshold = test.mse_threshold * mse_mult threshold = test.mse_threshold * mse_mult