Spaces:
Running
on
Zero
Running
on
Zero
wzhouxiff
commited on
Commit
·
b2a3331
1
Parent(s):
4b34a73
app add point to raw_input before processing
Browse files
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)
|