LPX55 commited on
Commit
bb800be
·
verified ·
1 Parent(s): 8fa6d2e

Update sam2_mask.py

Browse files
Files changed (1) hide show
  1. 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
- lambda img: (img, img, [], []),
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=output_image
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