NikeZoldyck commited on
Commit
3ccc105
1 Parent(s): a303eeb

Update utils/shared_utils.py

Browse files
Files changed (1) hide show
  1. utils/shared_utils.py +3 -3
utils/shared_utils.py CHANGED
@@ -15,7 +15,7 @@ from PIL import Image
15
  import numpy as np
16
  from utils.photo_wct import PhotoWCT
17
  from utils.photo_smooth import Propagator
18
- from utils.smooth_filter import smooth_filter
19
 
20
  # Load models
21
  root = Path.cwd()
@@ -115,8 +115,8 @@ def style_transfer(cont_img,styl_img):
115
  return stylized_img
116
 
117
  def smoother(stylized_img, over_img):
118
- #final_img = smoothing_module.process(stylized_img, over_img)
119
- final_img = smooth_filter(stylized_img, over_img, f_radius=15, f_edge=1e-1)
120
  return final_img
121
 
122
 
 
15
  import numpy as np
16
  from utils.photo_wct import PhotoWCT
17
  from utils.photo_smooth import Propagator
18
+ #from utils.smooth_filter import smooth_filter
19
 
20
  # Load models
21
  root = Path.cwd()
 
115
  return stylized_img
116
 
117
  def smoother(stylized_img, over_img):
118
+ final_img = smoothing_module.process(stylized_img, over_img)
119
+ #final_img = smooth_filter(stylized_img, over_img, f_radius=15, f_edge=1e-1)
120
  return final_img
121
 
122