Ryan-PR commited on
Commit
a1a68a8
·
verified ·
1 Parent(s): 19198ae

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -2
app.py CHANGED
@@ -613,8 +613,10 @@ with gr.Blocks() as demo:
613
  gr.Markdown(f"<div style='text-align:center;'>{text}</div>")
614
 
615
  with gr.Column():
 
 
 
616
  video_input = gr.Video(label="Upload Video", elem_id="my-video1")
617
- get_info_btn = gr.Button("Extract First Frame", elem_id="my-btn")
618
 
619
  gr.Examples(
620
  examples=[
@@ -630,6 +632,9 @@ with gr.Blocks() as demo:
630
  elem_id="my-btn2"
631
  )
632
 
 
 
 
633
  image_output = gr.Image(
634
  label="First Frame Segmentation",
635
  interactive=True,
@@ -642,6 +647,7 @@ with gr.Blocks() as demo:
642
  )
643
  clear_btn = gr.Button("Clear All Clicks")
644
 
 
645
  with gr.Row(elem_id="my-btn"):
646
  n_frames_slider = gr.Slider(
647
  minimum=1, maximum=201, value=81, step=1, label="Tracking Frames (4N+1)"
@@ -649,7 +655,7 @@ with gr.Blocks() as demo:
649
  track_btn = gr.Button("Tracking")
650
  video_output = gr.Video(label="Tracking Result", elem_id="my-video")
651
 
652
- gr.Markdown("Reference Image & Mask (SAM2 Points)", elem_id="ref_title")
653
 
654
  ref_image_input = gr.Image(
655
  label="Upload Reference Image", elem_id="ref-image", interactive=True
@@ -670,6 +676,8 @@ with gr.Blocks() as demo:
670
  label="You can upload or choose a reference image below to retexture.",
671
  elem_id="my-btn3"
672
  )
 
 
673
  ref_image_display = gr.Image(
674
  label="Reference Mask Segmentation",
675
  elem_id="ref-mask",
 
613
  gr.Markdown(f"<div style='text-align:center;'>{text}</div>")
614
 
615
  with gr.Column():
616
+
617
+ gr.Markdown("Step1: Upload a Source Video", elem_id="ref_title")
618
+
619
  video_input = gr.Video(label="Upload Video", elem_id="my-video1")
 
620
 
621
  gr.Examples(
622
  examples=[
 
632
  elem_id="my-btn2"
633
  )
634
 
635
+ gr.Markdown("Step2: Extract the First Frame & Click for Segmentation", elem_id="ref_title")
636
+ get_info_btn = gr.Button("Extract First Frame", elem_id="my-btn")
637
+
638
  image_output = gr.Image(
639
  label="First Frame Segmentation",
640
  interactive=True,
 
647
  )
648
  clear_btn = gr.Button("Clear All Clicks")
649
 
650
+ gr.Markdown("Step3: Track to Get Video Mask", elem_id="ref_title")
651
  with gr.Row(elem_id="my-btn"):
652
  n_frames_slider = gr.Slider(
653
  minimum=1, maximum=201, value=81, step=1, label="Tracking Frames (4N+1)"
 
655
  track_btn = gr.Button("Tracking")
656
  video_output = gr.Video(label="Tracking Result", elem_id="my-video")
657
 
658
+ gr.Markdown("Step4: Upload a Reference Image", elem_id="ref_title")
659
 
660
  ref_image_input = gr.Image(
661
  label="Upload Reference Image", elem_id="ref-image", interactive=True
 
676
  label="You can upload or choose a reference image below to retexture.",
677
  elem_id="my-btn3"
678
  )
679
+
680
+ gr.Markdown("Step5: Click for Segmentation", elem_id="ref_title")
681
  ref_image_display = gr.Image(
682
  label="Reference Mask Segmentation",
683
  elem_id="ref-mask",