freealise commited on
Commit
706d65e
·
verified ·
1 Parent(s): a3f556c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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.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):
@@ -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.COLOR_LAB2BGRA)
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