jflo commited on
Commit
60f810a
1 Parent(s): 6b7e390

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -16,11 +16,11 @@ def filter_img(input_img,filter_type):
16
  filtered_img = np.zeros_like(img)
17
 
18
  if filter_type == 'Vintage':
19
- filtered_img = cv2.transform(img, np.matrix([[0.272, 0.534, 0.131],
20
- [0.349, 0.686, 0.168],
21
- [0.393, 0.769, 0.189]])
22
- filtered_img[np.where(img > 255)] = 255 # normalizing values greater than 255 to 255
23
- filtered_img = np.array(filtered_img, dtype=np.uint8) # converting back to int
24
  elif filter_type == 'Gray-sharp':
25
  img = cv2.cvtColor(img,cv2.COLOR_RGB2GRAY)
26
  filtered_img = cv2.addWeighted(img,4, cv2.blur(img,(128,128)),-4,128)
 
16
  filtered_img = np.zeros_like(img)
17
 
18
  if filter_type == 'Vintage':
19
+ sepia_filter = np.array([[0.393, 0.769, 0.189],
20
+ [0.349, 0.686, 0.168],
21
+ [0.272, 0.534, 0.131]])
22
+
23
+ filtered_img = np.dot(img, sepia_filter.T).astype(np.uint8)
24
  elif filter_type == 'Gray-sharp':
25
  img = cv2.cvtColor(img,cv2.COLOR_RGB2GRAY)
26
  filtered_img = cv2.addWeighted(img,4, cv2.blur(img,(128,128)),-4,128)