Update app.py
Browse files
app.py
CHANGED
|
@@ -465,45 +465,45 @@ def draw_mask(o, b, v, d, evt: gr.EventData):
|
|
| 465 |
|
| 466 |
|
| 467 |
def read_file(sn,tn):
|
| 468 |
-
|
| 469 |
-
|
| 470 |
-
|
| 471 |
-
|
| 472 |
-
|
| 473 |
|
| 474 |
def get_mean_and_std(x):
|
| 475 |
-
|
| 476 |
-
|
| 477 |
-
|
| 478 |
-
|
| 479 |
|
| 480 |
def color_transfer(base):
|
| 481 |
global frames
|
| 482 |
targets = []
|
| 483 |
|
| 484 |
for n in range(len(frames)):
|
| 485 |
-
|
| 486 |
-
|
| 487 |
-
|
| 488 |
-
|
| 489 |
-
|
| 490 |
-
|
| 491 |
-
|
| 492 |
-
|
| 493 |
-
|
| 494 |
-
|
| 495 |
-
|
| 496 |
-
|
| 497 |
-
|
| 498 |
-
|
| 499 |
-
|
| 500 |
-
|
| 501 |
-
|
| 502 |
-
|
| 503 |
-
|
| 504 |
-
|
| 505 |
-
|
| 506 |
-
|
| 507 |
|
| 508 |
|
| 509 |
js = """
|
|
|
|
| 465 |
|
| 466 |
|
| 467 |
def read_file(sn,tn):
|
| 468 |
+
s = cv2.imread(sn)
|
| 469 |
+
s = cv2.cvtColor(s,cv2.COLOR_BGRA2LAB)
|
| 470 |
+
t = cv2.imread(tn)
|
| 471 |
+
t = cv2.cvtColor(t,cv2.COLOR_BGRA2LAB)
|
| 472 |
+
return s, t
|
| 473 |
|
| 474 |
def get_mean_and_std(x):
|
| 475 |
+
x_mean, x_std = cv2.meanStdDev(x)
|
| 476 |
+
x_mean = np.hstack(np.around(x_mean,2))
|
| 477 |
+
x_std = np.hstack(np.around(x_std,2))
|
| 478 |
+
return x_mean, x_std
|
| 479 |
|
| 480 |
def color_transfer(base):
|
| 481 |
global frames
|
| 482 |
targets = []
|
| 483 |
|
| 484 |
for n in range(len(frames)):
|
| 485 |
+
targets.append(frames[base])
|
| 486 |
+
if n != base:
|
| 487 |
+
print("Converting picture "+str(n)+"...")
|
| 488 |
+
s, t = read_file(frames[n],targets[n])
|
| 489 |
+
s_mean, s_std = get_mean_and_std(s)
|
| 490 |
+
t_mean, t_std = get_mean_and_std(t)
|
| 491 |
+
|
| 492 |
+
height, width, channel = s.shape
|
| 493 |
+
for i in range(0,height):
|
| 494 |
+
for j in range(0,width):
|
| 495 |
+
for k in range(0,channel):
|
| 496 |
+
x = s[i,j,k]
|
| 497 |
+
x = ((x-s_mean[k])*(t_std[k]/s_std[k]))+t_mean[k]
|
| 498 |
+
# round or +0.5
|
| 499 |
+
x = round(x)
|
| 500 |
+
# boundary check
|
| 501 |
+
x = 0 if x<0 else x
|
| 502 |
+
x = 255 if x>255 else x
|
| 503 |
+
s[i,j,k] = x
|
| 504 |
+
|
| 505 |
+
s = cv2.cvtColor(s,cv2.COLOR_LAB2BGRA)
|
| 506 |
+
cv2.imwrite(frames[n],s)
|
| 507 |
|
| 508 |
|
| 509 |
js = """
|