wzhouxiff commited on
Commit
b2a3331
·
1 Parent(s): 4b34a73

app add point to raw_input before processing

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -545,13 +545,15 @@ def draw_points_on_image(img, points):
545
 
546
  @spaces.GPU(duration=15)
547
  def from_examples(raw_input, raw_image_points, canvas, seg_image_points, selected_points_text, camera_option, mask_bk):
 
 
548
 
549
  selected_points = ast.literal_eval(selected_points_text)
550
  mask = np.array(mask_bk)
551
  mask = mask[:,:,0] > 0
552
  selected_points = ast.literal_eval(selected_points_text)
553
 
554
- image, _, depth, depth_img, colored_depth = process_image(raw_input, selected_points)
555
 
556
  # get camera pose
557
  if camera_option == "None":
@@ -564,8 +566,6 @@ def from_examples(raw_input, raw_image_points, canvas, seg_image_points, selecte
564
  speed = 4.0
565
  camera_pose, camera_pose_vis, rescale = get_camera_pose(CAMERA_MODE)(camera_option, depth, mask, rescale, angle, speed)
566
 
567
- raw_image_points = ast.literal_eval(raw_image_points)
568
- seg_image_points = ast.literal_eval(seg_image_points)
569
 
570
  raw_image = draw_points_on_image(raw_input['image'], raw_image_points)
571
  seg_image = draw_points_on_image(canvas['image'], seg_image_points)
 
545
 
546
  @spaces.GPU(duration=15)
547
  def from_examples(raw_input, raw_image_points, canvas, seg_image_points, selected_points_text, camera_option, mask_bk):
548
+ raw_image_points = ast.literal_eval(raw_image_points)
549
+ seg_image_points = ast.literal_eval(seg_image_points)
550
 
551
  selected_points = ast.literal_eval(selected_points_text)
552
  mask = np.array(mask_bk)
553
  mask = mask[:,:,0] > 0
554
  selected_points = ast.literal_eval(selected_points_text)
555
 
556
+ image, _, depth, depth_img, colored_depth = process_image({'image': raw_input['image'], 'points': raw_image_points}, selected_points)
557
 
558
  # get camera pose
559
  if camera_option == "None":
 
566
  speed = 4.0
567
  camera_pose, camera_pose_vis, rescale = get_camera_pose(CAMERA_MODE)(camera_option, depth, mask, rescale, angle, speed)
568
 
 
 
569
 
570
  raw_image = draw_points_on_image(raw_input['image'], raw_image_points)
571
  seg_image = draw_points_on_image(canvas['image'], seg_image_points)