Update app.py
Browse files
app.py
CHANGED
@@ -375,7 +375,7 @@ def track(
|
|
375 |
# queries__ = torch.cat([torch.zeros_like(xy[:, :, :1]), xy], dim=2).to(device) #
|
376 |
num_tracks = queries.shape[1]
|
377 |
# queries = torch.cat([queries,queries__],dim=1)
|
378 |
-
|
379 |
|
380 |
model(video_chunk=video_input, is_first_step=True, grid_size=0, queries=queries, add_support_grid=add_support_grid)
|
381 |
#
|
|
|
375 |
# queries__ = torch.cat([torch.zeros_like(xy[:, :, :1]), xy], dim=2).to(device) #
|
376 |
num_tracks = queries.shape[1]
|
377 |
# queries = torch.cat([queries,queries__],dim=1)
|
378 |
+
add_support_grid=True
|
379 |
|
380 |
model(video_chunk=video_input, is_first_step=True, grid_size=0, queries=queries, add_support_grid=add_support_grid)
|
381 |
#
|