add adjust mask feature

2024-01-05 14:57:30 +08:00
parent 2996544e75
commit e889e527ab
18 changed files with 507 additions and 76 deletions
--- a/lama_cleaner/tests/utils.py
+++ b/lama_cleaner/tests/utils.py
@@ -31,8 +31,6 @@ def assert_equal(
    mask_p=current_dir / "mask.png",
 ):
    img, mask = get_data(fx=fx, fy=fy, img_p=img_p, mask_p=mask_p)
-    config.image = encode_pil_to_base64(Image.fromarray(img), 95, {})[0]
-    config.mask = encode_pil_to_base64(Image.fromarray(mask), 95, {})[0]
    print(f"Input image shape: {img.shape}")
    res = model(img, mask, config)
    ok = cv2.imwrite(