Eueuiaa commited on
Commit
78f99ec
·
verified ·
1 Parent(s): 2667dc6

Update tools/video_encode_tool.py

Browse files
Files changed (1) hide show
  1. tools/video_encode_tool.py +2 -6
tools/video_encode_tool.py CHANGED
@@ -33,7 +33,7 @@ class VideoEncodeTool:
33
  Um especialista para lidar com tarefas de codificação e manipulação de vídeo.
34
  """
35
 
36
- def save_video_from_tensor(self, video_tensor: torch.Tensor, path: str, fps: int = 24, progress_callback=None):
37
  """
38
  Salva um tensor de pixel como um arquivo de vídeo .mp4 usando parâmetros otimizados.
39
  Espera um tensor no formato (B, C, F, H, W) onde B=1.
@@ -66,11 +66,7 @@ class VideoEncodeTool:
66
  ) as writer:
67
  for frame in video_np:
68
  writer.append_data(frame)
69
- if progress_callback:
70
- progress_callback(i + 1, T)
71
- if i % self.frame_log_every == 0:
72
- print(f"[DEBUG] [Encoder] frame {i}/{T} gravado ({H}x{W}@{fps}fps)")
73
-
74
 
75
  logger.info(f"Vídeo salvo com sucesso em: {path}")
76
  except Exception as e:
 
33
  Um especialista para lidar com tarefas de codificação e manipulação de vídeo.
34
  """
35
 
36
+ def save_video_from_tensor(self, video_tensor: torch.Tensor, path: str, fps: int = 24):
37
  """
38
  Salva um tensor de pixel como um arquivo de vídeo .mp4 usando parâmetros otimizados.
39
  Espera um tensor no formato (B, C, F, H, W) onde B=1.
 
66
  ) as writer:
67
  for frame in video_np:
68
  writer.append_data(frame)
69
+
 
 
 
 
70
 
71
  logger.info(f"Vídeo salvo com sucesso em: {path}")
72
  except Exception as e: