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

import fix

Browse Source
This commit is contained in:
nikitakaraevv 2023-07-21 07:52:31 -07:00
parent 24054be360
commit 1141fe6517
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
cotracker/utils/visualizer.py
View File

@ -14,7 +14,6 @@ from matplotlib import cm
import torch.nn.functional as F import torch.nn.functional as F
import torchvision.transforms as transforms import torchvision.transforms as transforms
from moviepy.editor import ImageSequenceClip from moviepy.editor import ImageSequenceClip
from torch.utils.tensorboard import SummaryWriter
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
@ -66,7 +65,7 @@ class Visualizer:
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",
writer: SummaryWriter = None, writer=None, # tensorboard Summary Writer, used for visualization during training
step: int = 0, step: int = 0,
query_frame: int = 0, query_frame: int = 0,
save_video: bool = True, save_video: bool = True,