Spaces:

aharley
/

alltracker

Runtime error

aharley commited on Oct 19

Commit

dcef70f

1 Parent(s): a8f5bea

updated print

Files changed (1) hide show

app.py CHANGED Viewed

@@ -788,6 +788,7 @@ def track(
     # traj_maps_e = traj_maps_e[:,:,:,::2,::2] # subsample
     # visconf_maps_e = visconf_maps_e[:,:,:,::2,::2] # subsample
     tracks = traj_maps_e.permute(0,3,4,1,2).reshape(-1,T,2).numpy()
     visibs = visconf_maps_e.permute(0,3,4,1,2).reshape(-1,T,2)[:,:,0].numpy()

     # traj_maps_e = traj_maps_e[:,:,:,::2,::2] # subsample
     # visconf_maps_e = visconf_maps_e[:,:,:,::2,::2] # subsample
+    print('traj_maps_e', traj_maps_e.shape)
     tracks = traj_maps_e.permute(0,3,4,1,2).reshape(-1,T,2).numpy()
     visibs = visconf_maps_e.permute(0,3,4,1,2).reshape(-1,T,2)[:,:,0].numpy()