Mirrors/Track-Anything
1
0
Fork 0
mirror of https://github.com/gaomingqi/Track-Anything.git synced 2025-12-16 16:37:58 +01:00
Code Issues Packages Projects Releases Wiki Activity

ignore update -- li

Browse Source
This commit is contained in:
memoryunreal
2023-05-26 11:14:40 +00:00
parent 322ae43757
commit e23c4444eb
2 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
.gitignore vendored
View File

@@ -10,3 +10,5 @@ debug_images/
images/ images/
test_sample/ test_sample/
result/ result/
vots/
vots.py

10
app.py
View File

@@ -125,7 +125,7 @@ def get_frames_from_video(video_input, video_state):
def run_example(example): def run_example(example):
return video_input return video_input
# get the select frame from gradio slider # get the select frame from gradio slider
def select_template(image_selection_slider, video_state, interactive_state): def select_template(image_selection_slider, video_state, interactive_state, mask_dropdown):
# images = video_state[1] # images = video_state[1]
image_selection_slider -= 1 image_selection_slider -= 1
@@ -139,8 +139,11 @@ def select_template(image_selection_slider, video_state, interactive_state):
# update the masks when select a new template frame # update the masks when select a new template frame
# if video_state["masks"][image_selection_slider] is not None: # if video_state["masks"][image_selection_slider] is not None:
# video_state["painted_images"][image_selection_slider] = mask_painter(video_state["origin_images"][image_selection_slider], video_state["masks"][image_selection_slider]) # video_state["painted_images"][image_selection_slider] = mask_painter(video_state["origin_images"][image_selection_slider], video_state["masks"][image_selection_slider])
if mask_dropdown:
print("ok")
operation_log = [("",""), ("Select frame {}. Try click image and add mask for tracking.".format(image_selection_slider),"Normal")] operation_log = [("",""), ("Select frame {}. Try click image and add mask for tracking.".format(image_selection_slider),"Normal")]
return video_state["painted_images"][image_selection_slider], video_state, interactive_state, operation_log return video_state["painted_images"][image_selection_slider], video_state, interactive_state, operation_log
# set the tracking end frame # set the tracking end frame
@@ -217,7 +220,6 @@ def remove_multi_mask(interactive_state, mask_dropdown):
def show_mask(video_state, interactive_state, mask_dropdown): def show_mask(video_state, interactive_state, mask_dropdown):
mask_dropdown.sort() mask_dropdown.sort()
select_frame = video_state["origin_images"][video_state["select_frame_number"]] select_frame = video_state["origin_images"][video_state["select_frame_number"]]
for i in range(len(mask_dropdown)): for i in range(len(mask_dropdown)):
mask_number = int(mask_dropdown[i].split("_")[1]) - 1 mask_number = int(mask_dropdown[i].split("_")[1]) - 1
mask = interactive_state["multi_mask"]["masks"][mask_number] mask = interactive_state["multi_mask"]["masks"][mask_number]
@@ -375,8 +377,8 @@ folder ="./checkpoints"
SAM_checkpoint = download_checkpoint(sam_checkpoint_url, folder, sam_checkpoint) SAM_checkpoint = download_checkpoint(sam_checkpoint_url, folder, sam_checkpoint)
xmem_checkpoint = download_checkpoint(xmem_checkpoint_url, folder, xmem_checkpoint) xmem_checkpoint = download_checkpoint(xmem_checkpoint_url, folder, xmem_checkpoint)
e2fgvi_checkpoint = download_checkpoint_from_google_drive(e2fgvi_checkpoint_id, folder, e2fgvi_checkpoint) e2fgvi_checkpoint = download_checkpoint_from_google_drive(e2fgvi_checkpoint_id, folder, e2fgvi_checkpoint)
# args.port = 12212 args.port = 12212
# args.device = "cuda:1" args.device = "cuda:3"
# args.mask_save = True # args.mask_save = True
# initialize sam, xmem, e2fgvi models # initialize sam, xmem, e2fgvi models