add grid_query_frame
Browse files
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]
|