Eueuiaa commited on
Commit
b64bf21
·
verified ·
1 Parent(s): d351d1c

Update tools/video_encode_tool.py

Browse files
Files changed (1) hide show
  1. tools/video_encode_tool.py +3 -2
tools/video_encode_tool.py CHANGED
@@ -22,8 +22,9 @@ class _SimpleVideoEncoder:
22
  # Mantemos simples: um vídeo por chamada (B=1)
23
  raise ValueError(f"Esperado B=1, recebido B={B}")
24
 
25
- with imageio.get_writer(path, fps=int(fps), codec="libx264", quality=8) as writer:
26
- for i in range(T):
 
27
  frame_chw = pixel_5d[0, :, i] # (C,H,W)
28
  frame_hwc_u8 = (frame_chw.permute(1, 2, 0)
29
  .clamp(0, 1)
 
22
  # Mantemos simples: um vídeo por chamada (B=1)
23
  raise ValueError(f"Esperado B=1, recebido B={B}")
24
 
25
+ print(pixel_5d.shape)
26
+ with imageio.get_writer(path, fps=int(fps), codec="libx264", quality=10) as writer:
27
+ for i in range(T-1):
28
  frame_chw = pixel_5d[0, :, i] # (C,H,W)
29
  frame_hwc_u8 = (frame_chw.permute(1, 2, 0)
30
  .clamp(0, 1)