jflo commited on
Commit
2f73ba5
1 Parent(s): a12370e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -8
app.py CHANGED
@@ -6,20 +6,21 @@ from PIL import Image, ImageOps
6
  def filter_img(input_img,filter_type):
7
  img = cv2.imread(input_img)
8
 
9
- #new_height = 1080
10
 
11
- #ratio = float(new_height) / img.shape[0]
12
- #new_width = int(img.shape[1] * ratio)
13
 
14
- #img = cv2.resize(img, (new_width,new_height))
15
 
16
  filtered_img = np.zeros_like(img)
17
 
18
  if filter_type == 'Vintage':
19
- gray_img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
20
- filtered_img[:,:,0] = (gray_img * 0.393) + (gray_img * 0.769) + (gray_img * 0.189)
21
- filtered_img[:,:,1] = (gray_img * 0.349) + (gray_img * 0.686) + (gray_img * 0.168)
22
- filtered_img[:,:,2] = (gray_img * 0.272) + (gray_img * 0.534) + (gray_img * 0.131)
 
23
  elif filter_type == 'Gray-sharp':
24
  img = cv2.cvtColor(img,cv2.COLOR_RGB2GRAY)
25
  filtered_img = cv2.addWeighted(img,4, cv2.blur(img,(128,128)),-4,128)
 
6
  def filter_img(input_img,filter_type):
7
  img = cv2.imread(input_img)
8
 
9
+ new_height = 1080
10
 
11
+ ratio = float(new_height) / img.shape[0]
12
+ new_width = int(img.shape[1] * ratio)
13
 
14
+ img = cv2.resize(img, (new_width,new_height))
15
 
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(filtered_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)