kamp/IOPaint
1
0
Fork 0
Code Issues Pull Requests Releases 1 Activity

sd add blur mask test

Browse Source
This commit is contained in:
Qing 2022-09-22 13:29:06 +08:00
parent e1fb0030d1
commit 491e62fdf1
2 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
lama_cleaner/tests/overture-creations-5sI6fQgYIuo_mask_blur.png Normal file
View File

Binary file not shown.
Side by Side

After

Width:  |  Height:  |  Size: 38 KiB

8
lama_cleaner/tests/test_model.py
View File

@ -176,6 +176,14 @@ def test_sd(strategy, sampler, capfd):
mask_p=current_dir / "overture-creations-5sI6fQgYIuo_mask.png", mask_p=current_dir / "overture-creations-5sI6fQgYIuo_mask.png",
) )
assert_equal(
model,
cfg,
f"sd_{strategy.capitalize()}_{sampler}_blur_mask_result.png",
img_p=current_dir / "overture-creations-5sI6fQgYIuo.png",
mask_p=current_dir / "overture-creations-5sI6fQgYIuo_mask_blur.png",
)
# captured = capfd.readouterr() # captured = capfd.readouterr()
# for i in range(sd_steps): # for i in range(sd_steps):
# assert f'sd_step_{i}' in captured.out # assert f'sd_step_{i}' in captured.out