Update app.py
Browse files
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 |
-
|
10 |
|
11 |
-
|
12 |
-
|
13 |
|
14 |
-
|
15 |
|
16 |
filtered_img = np.zeros_like(img)
|
17 |
|
18 |
if filter_type == 'Vintage':
|
19 |
-
|
20 |
-
|
21 |
-
|
22 |
-
filtered_img[
|
|
|
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)
|