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 |
|