add meta copyright
This commit is contained in:
parent
fac27989b3
commit
941c24fd40
@ -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 torch
|
||||
import gradio as gr
|
||||
@ -22,7 +29,12 @@ def cotracker_demo(
|
||||
model = model.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):
|
||||
pred_tracks, pred_visibility = model(
|
||||
video_chunk=load_video[:, ind : ind + model.step * 2]
|
||||
|
Loading…
Reference in New Issue
Block a user