Spaces:
Running
on
Zero
Running
on
Zero
Update sam2_mask.py
Browse files- sam2_mask.py +10 -6
sam2_mask.py
CHANGED
@@ -173,27 +173,31 @@ def create_sam2_tab():
|
|
173 |
|
174 |
# Event handlers
|
175 |
points_map.upload(
|
176 |
-
|
177 |
-
inputs=points_map,
|
178 |
-
outputs=[sam_input_image, first_frame, tracking_points, trackings_input_label]
|
|
|
|
|
179 |
)
|
180 |
|
181 |
clear_button.click(
|
182 |
lambda img: ([], [], img),
|
183 |
inputs=first_frame,
|
184 |
-
outputs=[tracking_points, trackings_input_label, points_map]
|
|
|
185 |
)
|
186 |
|
187 |
points_map.select(
|
188 |
get_point,
|
189 |
inputs=[point_type, tracking_points, trackings_input_label, first_frame],
|
190 |
-
outputs=[tracking_points, trackings_input_label, points_map]
|
|
|
191 |
)
|
192 |
|
193 |
submit_button.click(
|
194 |
sam_process,
|
195 |
inputs=[sam_input_image, tracking_points, trackings_input_label],
|
196 |
-
outputs=
|
197 |
)
|
198 |
|
199 |
return sam_input_image, points_map, output_image
|
|
|
173 |
|
174 |
# Event handlers
|
175 |
points_map.upload(
|
176 |
+
fn = preprocess_image,
|
177 |
+
inputs = [points_map],
|
178 |
+
outputs=[sam_input_image, first_frame, tracking_points, trackings_input_label],
|
179 |
+
# outputs = [first_frame_path, tracking_points, trackings_input_label, input_image]
|
180 |
+
queue=False
|
181 |
)
|
182 |
|
183 |
clear_button.click(
|
184 |
lambda img: ([], [], img),
|
185 |
inputs=first_frame,
|
186 |
+
outputs=[tracking_points, trackings_input_label, points_map],
|
187 |
+
queue=False
|
188 |
)
|
189 |
|
190 |
points_map.select(
|
191 |
get_point,
|
192 |
inputs=[point_type, tracking_points, trackings_input_label, first_frame],
|
193 |
+
outputs=[tracking_points, trackings_input_label, points_map],
|
194 |
+
queue = False
|
195 |
)
|
196 |
|
197 |
submit_button.click(
|
198 |
sam_process,
|
199 |
inputs=[sam_input_image, tracking_points, trackings_input_label],
|
200 |
+
outputs = [output_result, output_result_mask]
|
201 |
)
|
202 |
|
203 |
return sam_input_image, points_map, output_image
|