Initial Commit

.ipynb_checkpoints/app-checkpoint.py

@@ -0,0 +1,35 @@
+import numpy as np
+import matplotlib.pyplot as plt
+import cv2
+# from segment_anything import sam_model_registry, SamPredictor
+from preprocess import show_mask, show_points, show_box
+import gradio as gr
+
+def get_coord(image):
+    return coord
+
+my_app = gr.Blocks()
+with my_app:
+    gr.Markdown("Segment Anything Testing")
+    with gr.Tabs():
+        with gr.TabItem("Select your image"):
+            with gr.Row():
+                with gr.Column():
+                    img_source = gr.Image(label="Please select picture.", value='./images/truck.jpg', shape=(768, 768))
+                    
+                    set_point = gr.Button(label="Set Point")
+                with gr.Column():
+                    # img_output = gr.Image(label="Output Mask")
+                    coords = gr.Label(label="Image Coordinate.")
+                    
+        set_point.click(
+            img_source.select(get_coord),
+            [
+                img_source
+            ],
+            [
+                coords
+            ]
+        )
+    
+my_app.launch(debug=True)

app.py

@@ -1,8 +1,7 @@
 import numpy as np
-import torch
 import matplotlib.pyplot as plt
 import cv2
-from segment_anything import sam_model_registry, SamPredictor
+# from segment_anything import sam_model_registry, SamPredictor
 from preprocess import show_mask, show_points, show_box
 import gradio as gr