iicd/cotracker
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add meta copyright

Browse Source
This commit is contained in:
Nikita Karaev 2024-01-05 16:17:50 +00:00
parent fac27989b3
commit 941c24fd40
1 changed files with 13 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
gradio_demo/app.py
View File

@ -1,3 +1,10 @@
# Copyright (c) Meta Platforms, Inc. and affiliates.
# All rights reserved.
# This source code is licensed under the license found in the
# LICENSE file in the root directory of this source tree.
import os import os
import torch import torch
import gradio as gr import gradio as gr
@ -22,7 +29,12 @@ def cotracker_demo(
model = model.cuda() model = model.cuda()
load_video = load_video.cuda() load_video = load_video.cuda()
model(video_chunk=load_video, is_first_step=True, grid_size=grid_size) model(
video_chunk=load_video,
is_first_step=True,
grid_size=grid_size,
grid_query_frame=grid_query_frame,
)
for ind in range(0, load_video.shape[1] - model.step, model.step): for ind in range(0, load_video.shape[1] - model.step, model.step):
pred_tracks, pred_visibility = model( pred_tracks, pred_visibility = model(
video_chunk=load_video[:, ind : ind + model.step * 2] video_chunk=load_video[:, ind : ind + model.step * 2]