Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -1,27 +1,46 @@
|
|
1 |
import gradio as gr
|
2 |
import cv2
|
3 |
import numpy as np
|
|
|
4 |
|
5 |
def convert_to_grayscale(image):
|
6 |
# ์ด๋ฏธ์ง๋ฅผ ํ๋ฐฑ์ผ๋ก ๋ณํ
|
7 |
gray_image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
|
8 |
-
|
|
|
|
|
9 |
|
10 |
def convert_and_save(image):
|
11 |
-
# ์ด๋ฏธ์ง๋ฅผ
|
|
|
|
|
12 |
gray_image = convert_to_grayscale(image)
|
|
|
13 |
output_path = "output.jpg"
|
|
|
14 |
cv2.imwrite(output_path, gray_image)
|
15 |
-
|
|
|
16 |
|
17 |
# Gradio ์ธํฐํ์ด์ค ์ ์
|
18 |
-
|
19 |
-
|
20 |
-
|
21 |
-
|
22 |
-
|
23 |
-
|
24 |
-
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
25 |
|
26 |
if __name__ == "__main__":
|
27 |
-
|
|
|
1 |
import gradio as gr
|
2 |
import cv2
|
3 |
import numpy as np
|
4 |
+
from gradio_imageslider import ImageSlider
|
5 |
|
6 |
def convert_to_grayscale(image):
|
7 |
# ์ด๋ฏธ์ง๋ฅผ ํ๋ฐฑ์ผ๋ก ๋ณํ
|
8 |
gray_image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
|
9 |
+
# ํ๋ฐฑ ์ด๋ฏธ์ง๋ฅผ BGR ํ์์ผ๋ก ๋ค์ ๋ณํํ์ฌ 3์ฑ๋ ์ด๋ฏธ์ง๋ก ๋ง๋ฆ
|
10 |
+
gray_image_bgr = cv2.cvtColor(gray_image, cv2.COLOR_GRAY2BGR)
|
11 |
+
return gray_image_bgr
|
12 |
|
13 |
def convert_and_save(image):
|
14 |
+
# ์๋ณธ ์ด๋ฏธ์ง๋ฅผ ๋ณต์ฌ
|
15 |
+
original_image = image.copy()
|
16 |
+
# ์ด๋ฏธ์ง๋ฅผ ํ๋ฐฑ์ผ๋ก ๋ณํ
|
17 |
gray_image = convert_to_grayscale(image)
|
18 |
+
# ์ถ๋ ฅ ํ์ผ ๊ฒฝ๋ก ์ค์
|
19 |
output_path = "output.jpg"
|
20 |
+
# ํ๋ฐฑ ์ด๋ฏธ์ง ์ ์ฅ
|
21 |
cv2.imwrite(output_path, gray_image)
|
22 |
+
# ์ฌ๋ผ์ด๋์ ์ฌ์ฉํ ๋ ์ด๋ฏธ์ง ๋ฐํ
|
23 |
+
return (original_image, gray_image), output_path
|
24 |
|
25 |
# Gradio ์ธํฐํ์ด์ค ์ ์
|
26 |
+
with gr.Blocks() as demo:
|
27 |
+
gr.Markdown("# ์ด๋ฏธ์ง ํ๋ฐฑ ๋ณํ๊ธฐ")
|
28 |
+
gr.Markdown("์ด๋ฏธ์ง๋ฅผ ์
๋ก๋ํ๋ฉด ํ๋ฐฑ์ผ๋ก ๋ณํํ๊ณ , ๋ณ๊ฒฝ ์ ํ๋ฅผ ์ฌ๋ผ์ด๋ ํํ๋ก ๋น๊ตํ ์ ์์ต๋๋ค.")
|
29 |
+
|
30 |
+
with gr.Row():
|
31 |
+
image_input = gr.Image(label="์ด๋ฏธ์ง ์
๋ก๋", type="numpy")
|
32 |
+
file_output = gr.File(label="๋ค์ด๋ก๋ ๊ฐ๋ฅํ JPG ํ์ผ")
|
33 |
+
|
34 |
+
with gr.Row():
|
35 |
+
slider_output = ImageSlider(label="๋ณ๊ฒฝ ์ ํ ๋น๊ต", type="numpy")
|
36 |
+
|
37 |
+
# ๋ฒํผ์ ๋๋ ์ ๋ ํจ์ ์คํ
|
38 |
+
upload_btn = gr.Button("๋ณํํ๊ธฐ")
|
39 |
+
upload_btn.click(
|
40 |
+
fn=convert_and_save,
|
41 |
+
inputs=image_input,
|
42 |
+
outputs=[slider_output, file_output]
|
43 |
+
)
|
44 |
|
45 |
if __name__ == "__main__":
|
46 |
+
demo.launch()
|