From 51175e006acd8a0a5c507cc2a4bd205228c8fbc1 Mon Sep 17 00:00:00 2001 From: JunkyByte Date: Tue, 25 Jul 2023 16:30:38 +0200 Subject: [PATCH] allow no visibility --- cotracker/utils/visualizer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/cotracker/utils/visualizer.py b/cotracker/utils/visualizer.py index 696bc85..704b591 100644 --- a/cotracker/utils/visualizer.py +++ b/cotracker/utils/visualizer.py @@ -62,7 +62,7 @@ class Visualizer: self, video: torch.Tensor, # (B,T,C,H,W) 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) segm_mask: torch.Tensor = None, # (B,1,H,W) filename: str = "video", @@ -128,7 +128,7 @@ class Visualizer: self, video: torch.Tensor, tracks: torch.Tensor, - visibility: torch.Tensor, + visibility: torch.Tensor = None, segm_mask: torch.Tensor = None, gt_tracks=None, query_frame: int = 0,