Foreign-Mirrors/Fooocus
1
0
Fork 0
mirror of https://github.com/lllyasviel/Fooocus.git synced 2026-08-16 13:13:16 +02:00
Code Issues Packages Projects Releases Wiki Activity

rework refiner for some potential new features (#642)

Browse Source 
* sync

* sync

* sync

* sync

* sync

* sync

* sync

* sync

* sync

* sync

* sync

* sync

* sync

* sync

* sync

* sync

* sync

* sync

* sync

* sync

* sync

* sync

* sync

* sync
This commit is contained in:
lllyasviel
2023-10-11 03:07:43 -07:00
committed by GitHub
parent 777510dc9b
commit bbdf4bd120
10 changed files with 209 additions and 93 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/inpaint_worker.py
+4 -7
View File
@@ -83,16 +83,13 @@ def area_abcd(a, b, c, d):
return (b - a) * (d - c)
def solve_abcd(x, a, b, c, d, k, outpaint):
def solve_abcd(x, a, b, c, d, outpaint):
H, W = x.shape[:2]
if outpaint:
return 0, H, 0, W
min_area = H * W * k
max_area = H * W
min_area = (min(H, W) ** 2) * 0.5
while True:
if area_abcd(a, b, c, d) > min_area and abs((b - a) - (d - c)) < 16:
break
if area_abcd(a, b, c, d) >= max_area:
if area_abcd(a, b, c, d) >= min_area:
break
add_h = (b - a) < (d - c)
@@ -150,7 +147,7 @@ class InpaintWorker:
# compute abcd
a, b, c, d = compute_initial_abcd(self.mask_raw_bg < 127)
a, b, c, d = solve_abcd(self.mask_raw_bg, a, b, c, d, k=0.618, outpaint=is_outpaint)
a, b, c, d = solve_abcd(self.mask_raw_bg, a, b, c, d, outpaint=is_outpaint)
# interested area
self.interested_area = (a, b, c, d)