From 1141fe651708fecee38b65bfa64f36349be80747 Mon Sep 17 00:00:00 2001 From: nikitakaraevv Date: Fri, 21 Jul 2023 07:52:31 -0700 Subject: [PATCH] import fix --- cotracker/utils/visualizer.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/cotracker/utils/visualizer.py b/cotracker/utils/visualizer.py index 404436b..2187c18 100644 --- a/cotracker/utils/visualizer.py +++ b/cotracker/utils/visualizer.py @@ -14,7 +14,6 @@ from matplotlib import cm import torch.nn.functional as F import torchvision.transforms as transforms from moviepy.editor import ImageSequenceClip -from torch.utils.tensorboard import SummaryWriter import matplotlib.pyplot as plt @@ -66,7 +65,7 @@ class Visualizer: gt_tracks: torch.Tensor = None, # (B,T,N,2) segm_mask: torch.Tensor = None, # (B,1,H,W) filename: str = "video", - writer: SummaryWriter = None, + writer=None, # tensorboard Summary Writer, used for visualization during training step: int = 0, query_frame: int = 0, save_video: bool = True,