Update app.py
Browse files
app.py
CHANGED
@@ -437,6 +437,7 @@ def track(
|
|
437 |
grid_size=0,
|
438 |
queries=queries,
|
439 |
add_support_grid=add_support_grid,
|
|
|
440 |
)
|
441 |
#
|
442 |
for ind in range(0, video_input.shape[1] - model.step, model.step):
|
@@ -445,6 +446,7 @@ def track(
|
|
445 |
grid_size=0,
|
446 |
queries=queries,
|
447 |
add_support_grid=add_support_grid,
|
|
|
448 |
) # B T N 2, B T N 1
|
449 |
tracks = (
|
450 |
(
|
|
|
437 |
grid_size=0,
|
438 |
queries=queries,
|
439 |
add_support_grid=add_support_grid,
|
440 |
+
iters=4
|
441 |
)
|
442 |
#
|
443 |
for ind in range(0, video_input.shape[1] - model.step, model.step):
|
|
|
446 |
grid_size=0,
|
447 |
queries=queries,
|
448 |
add_support_grid=add_support_grid,
|
449 |
+
iters=4
|
450 |
) # B T N 2, B T N 1
|
451 |
tracks = (
|
452 |
(
|