iicd/cotracker
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

allow no visibility

Browse Source
This commit is contained in:
JunkyByte
2023-07-25 16:30:38 +02:00
parent d6df5d248f
commit 51175e006a
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
cotracker/utils/visualizer.py
View File

@@ -62,7 +62,7 @@ class Visualizer:
self, self,
video: torch.Tensor, # (B,T,C,H,W) video: torch.Tensor, # (B,T,C,H,W)
tracks: torch.Tensor, # (B,T,N,2) tracks: torch.Tensor, # (B,T,N,2)
visibility: torch.Tensor, # (B, T, N, 1) bool visibility: torch.Tensor = None, # (B, T, N, 1) bool
gt_tracks: torch.Tensor = None, # (B,T,N,2) gt_tracks: torch.Tensor = None, # (B,T,N,2)
segm_mask: torch.Tensor = None, # (B,1,H,W) segm_mask: torch.Tensor = None, # (B,1,H,W)
filename: str = "video", filename: str = "video",
@@ -128,7 +128,7 @@ class Visualizer:
self, self,
video: torch.Tensor, video: torch.Tensor,
tracks: torch.Tensor, tracks: torch.Tensor,
visibility: torch.Tensor, visibility: torch.Tensor = None,
segm_mask: torch.Tensor = None, segm_mask: torch.Tensor = None,
gt_tracks=None, gt_tracks=None,
query_frame: int = 0, query_frame: int = 0,