Hugging Face's logo

Spaces:
facebook
/
cotracker
Running on Zero

App Files Community
6
nikkar commited on
Commit
8869cd8
•
1 Parent(s): 859bbe2

add grid_query_frame

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -39,7 +39,7 @@ def cotracker_demo(
39
  model = model.cuda()
40
  load_video = load_video.cuda()
41
 
42
- model(video_chunk=load_video, is_first_step=True, grid_size=grid_size)
43
  for ind in range(0, load_video.shape[1] - model.step, model.step):
44
  pred_tracks, pred_visibility = model(
45
  video_chunk=load_video[:, ind : ind + model.step * 2]
 
39
  model = model.cuda()
40
  load_video = load_video.cuda()
41
 
42
+ model(video_chunk=load_video, is_first_step=True, grid_size=grid_size, grid_query_frame=grid_query_frame)
43
  for ind in range(0, load_video.shape[1] - model.step, model.step):
44
  pred_tracks, pred_visibility = model(
45
  video_chunk=load_video[:, ind : ind + model.step * 2]