jflo commited on
Commit
5e97a51
1 Parent(s): d6dc447

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -6
app.py CHANGED
@@ -22,14 +22,13 @@ def filter_img(input_img,filter_type):
22
 
23
  img = cv2.resize(img, (new_width,new_height))
24
 
25
- filtered_img = np.zeros(img.shape, dtype=int)
26
 
27
  if filter_type == 'Vintage':
28
- sepia_filter = np.array(
29
- [[0.393, 0.769, 0.189], [0.349, 0.686, 0.168], [0.272, 0.534, 0.131]]
30
- )
31
- filtered_img = img.dot(sepia_filter.T)
32
- filtered_img /= filtered_img.max()
33
  elif filter_type == 'Gray-sharp':
34
  img = cv2.cvtColor(img,cv2.COLOR_RGB2GRAY)
35
  filtered_img = cv2.addWeighted(img,4, cv2.blur(img,(128,128)),-4,128)
 
22
 
23
  img = cv2.resize(img, (new_width,new_height))
24
 
25
+ filtered_img = np.zeros_like(img)
26
 
27
  if filter_type == 'Vintage':
28
+ gray_img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
29
+ filtered_img[:,:,0] = (gray_img * 0.393) + (gray_img * 0.769) + (gray_img * 0.189)
30
+ filtered_img[:,:,1] = (gray_img * 0.349) + (gray_img * 0.686) + (gray_img * 0.168)
31
+ filtered_img[:,:,2] = (gray_img * 0.272) + (gray_img * 0.534) + (gray_img * 0.131)
 
32
  elif filter_type == 'Gray-sharp':
33
  img = cv2.cvtColor(img,cv2.COLOR_RGB2GRAY)
34
  filtered_img = cv2.addWeighted(img,4, cv2.blur(img,(128,128)),-4,128)