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)
# resize mask to match source size
stage_source = sources[1]
stage_source = sources.pop()
stage_mask = mask.resize(stage_source.size, Image.Resampling.BILINEAR)
chain.stage(

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

@ -564,10 +564,15 @@ def run_test(
passed = False
for i in range(len(results)):
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 = Image.open(ref_name) if path.exists(ref_name) else None
ref_name = test_path(path.join("test-refs", result_name))
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)
threshold = test.mse_threshold * mse_mult