NikeZoldyck commited on
Commit
4931eb5
1 Parent(s): 95d2d6d

Testing a smooth filter

Browse files
Files changed (1) hide show
  1. utils/shared_utils.py +2 -1
utils/shared_utils.py CHANGED
@@ -114,7 +114,8 @@ def style_transfer(cont_img,styl_img):
114
  return stylized_img
115
 
116
  def smoother(stylized_img, over_img):
117
- final_img = smoothing_module.process(stylized_img, over_img)
 
118
  return final_img
119
 
120
 
 
114
  return stylized_img
115
 
116
  def smoother(stylized_img, over_img):
117
+ #final_img = smoothing_module.process(stylized_img, over_img)
118
+ final_img = smooth_filter(stylized_img, over_img, f_radius=15, f_edge=1e-1)
119
  return final_img
120
 
121