Update tools/video_encode_tool.py
Browse files
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
|
| 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 |
-
|
| 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:
|