Update deformes4D_engine.py
Browse files- deformes4D_engine.py +3 -1
deformes4D_engine.py
CHANGED
|
@@ -262,8 +262,10 @@ class Deformes4DEngine:
|
|
| 262 |
temp_video_base_name = f"debug_chunk_{chunk_idx}"
|
| 263 |
|
| 264 |
# Converte o latente do chunk em um vídeo MP4
|
| 265 |
-
|
| 266 |
|
|
|
|
|
|
|
| 267 |
# Conta os frames no vídeo gerado
|
| 268 |
if os.path.exists(temp_video_path):
|
| 269 |
with imageio.get_reader(temp_video_path) as reader:
|
|
|
|
| 262 |
temp_video_base_name = f"debug_chunk_{chunk_idx}"
|
| 263 |
|
| 264 |
# Converte o latente do chunk em um vídeo MP4
|
| 265 |
+
pixel_tensor = self.latents_to_pixels(single_chunk_latent)
|
| 266 |
|
| 267 |
+
temp_video_path = self.save_video_from_tensor(pixel_tensor, temp_video_base_name, fps=24)
|
| 268 |
+
|
| 269 |
# Conta os frames no vídeo gerado
|
| 270 |
if os.path.exists(temp_video_path):
|
| 271 |
with imageio.get_reader(temp_video_path) as reader:
|