Initial Commit

.ipynb_checkpoints/app-checkpoint.py

@@ -1,35 +0,0 @@
-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)

requirements.txt

@@ -0,0 +1,2 @@
+opencv-python
+torch