This commit is contained in:
Eugene Fairley 2025-06-29 00:41:59 -07:00
parent 53d173e59e
commit 1cbb7e1838

View File

@ -1109,8 +1109,8 @@ def patch_motion(
):
"""Apply motion patching based on tracks"""
with torch.no_grad():
print("vid shape:", vid[0,0])
print("vid shape:", vid[0,10])
print("vid shape:", vid[4,0])
print("vid shape:", vid[4,10])
_, T, H, W = vid.shape
N = tracks.shape[2]
_, tracks_xy, visible = torch.split(