let5see/IOPaint
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add migan

Browse Source
This commit is contained in:
Qing
2023-11-20 13:05:28 +08:00
parent 53aea791c5
commit a5c241ac02
3 changed files with 134 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
lama_cleaner/tests/test_model.py
View File

@@ -170,7 +170,7 @@ def test_cv2(strategy, cv2_flag, cv2_radius):
assert_equal(
model,
cfg,
f"sd_{strategy.capitalize()}_{cv2_flag}_{cv2_radius}.png",
f"cv2_{strategy.capitalize()}_{cv2_flag}_{cv2_radius}.png",
img_p=current_dir / "overture-creations-5sI6fQgYIuo.png",
mask_p=current_dir / "overture-creations-5sI6fQgYIuo_mask.png",
)
@@ -188,7 +188,23 @@ def test_manga(strategy):
assert_equal(
model,
cfg,
f"sd_{strategy.capitalize()}.png",
f"manga_{strategy.capitalize()}.png",
img_p=current_dir / "overture-creations-5sI6fQgYIuo.png",
mask_p=current_dir / "overture-creations-5sI6fQgYIuo_mask.png",
)
@pytest.mark.parametrize("strategy", [HDStrategy.ORIGINAL])
def test_mi_gan(strategy):
model = ModelManager(
name="migan",
device=torch.device(device),
)
cfg = get_config(strategy)
assert_equal(
model,
cfg,
f"migan_{strategy.capitalize()}.png",
img_p=current_dir / "overture-creations-5sI6fQgYIuo.png",
mask_p=current_dir / "overture-creations-5sI6fQgYIuo_mask.png",
)