Add --video recording for all renderer backends (Newton+OVRTX, Newton+Warp)#4936
Add --video recording for all renderer backends (Newton+OVRTX, Newton+Warp)#4936bdilinila wants to merge 12 commits intoisaac-sim:developfrom
Conversation
fb60d67 to
8f4c68f
Compare
| """ | ||
| output = self._video_camera.data.output | ||
| # shape: [num_envs, H, W, 3], uint8 | ||
| rgb_all = output["rgb"] if "rgb" in output else output["rgba"][..., :3] |
There was a problem hiding this comment.
Does this return from all environments? usually when rendering training video, we only care about 1 environment, or, a handful environment that is small enough for human ease to parse.
…gh FOV conversion and position
Greptile SummaryThis PR extends the Key issues found:
Confidence Score: 2/5
Important Files Changed
Sequence DiagramsequenceDiagram
participant Script as Train/Play Script
participant Gym as gym.make()
participant Env as DirectRLEnv / ManagerBasedRLEnv
participant VRCfg as VideoRecorderCfg
participant VR as VideoRecorder
participant RV as RecordVideo Wrapper
Script->>VRCfg: set video_mode, render_mode, kit_cam_prim_path
Script->>Gym: gym.make(task, render_mode="rgb_array")
Gym->>Env: __init__(cfg, render_mode)
Env->>VRCfg: forward render_mode / kit settings
Env->>VR: class_type(cfg, scene)
Note over VR: If tiled+fallback: spawn VideoCamera prims<br/>Enable pyglet headless
Env->>Env: sim.reset() → physics init
Gym-->>Script: env (unwrapped)
Script->>RV: wrap env with RecordVideo
loop Each step
Script->>RV: step(action)
RV->>Env: step(action)
RV->>Env: render()
Env->>VR: render_rgb_array()
alt perspective mode
VR->>VR: _try_init_gl_viewer() [once]
alt Newton backend
VR-->>Env: Newton GL frame (ndarray)
else Kit backend
VR-->>Env: Kit /OmniverseKit_Persp frame (ndarray)
end
else tiled mode
VR->>VR: _find_video_camera() [cached after 1st call]
VR->>VR: _render_tiled_camera_rgb_array()
VR-->>Env: (G*H × G*W × 3) tile grid
end
Env-->>RV: ndarray frame
RV->>RV: write frame to video
end
Last reviewed commit: ed55c9e |
…, and stale comments
Description
Please include a summary of the change and which issue is fixed. Please also include relevant motivation and context.
List any dependencies that are required for this change.
Fixes # (issue)
Type of change
Screenshots
Please attach before and after screenshots of the change if applicable.
Checklist
pre-commitchecks with./isaaclab.sh --formatconfig/extension.tomlfileCONTRIBUTORS.mdor my name already exists there