Update app.py
Browse files
app.py
CHANGED
@@ -466,9 +466,9 @@ def draw_mask(o, b, v, d, evt: gr.EventData):
|
|
466 |
|
467 |
def read_file(sn,tn):
|
468 |
s = cv2.imread(sn)
|
469 |
-
s = cv2.cvtColor(s,cv2.
|
470 |
t = cv2.imread(tn)
|
471 |
-
t = cv2.cvtColor(t,cv2.
|
472 |
return s, t
|
473 |
|
474 |
def get_mean_and_std(x):
|
@@ -502,7 +502,7 @@ def color_transfer(base):
|
|
502 |
x = 255 if x>255 else x
|
503 |
s[i,j,k] = x
|
504 |
|
505 |
-
s = cv2.cvtColor(s,cv2.
|
506 |
cv2.imwrite(frames[n],s)
|
507 |
|
508 |
|
|
|
466 |
|
467 |
def read_file(sn,tn):
|
468 |
s = cv2.imread(sn)
|
469 |
+
s = cv2.cvtColor(s,cv2.COLOR_BGR2LAB)
|
470 |
t = cv2.imread(tn)
|
471 |
+
t = cv2.cvtColor(t,cv2.COLOR_BGR2LAB)
|
472 |
return s, t
|
473 |
|
474 |
def get_mean_and_std(x):
|
|
|
502 |
x = 255 if x>255 else x
|
503 |
s[i,j,k] = x
|
504 |
|
505 |
+
s = cv2.cvtColor(s,cv2.COLOR_LAB2BGR)
|
506 |
cv2.imwrite(frames[n],s)
|
507 |
|
508 |
|