← 返回架构图
🖼️

Vis Utils

wan/utils/vis_utils.py · visualize_wasd_and_rotation_ui
可视化输出
在生成视频上叠加实时 WASD 操控 UI 和相机旋转轨迹可视化, 方便直观地验证动作控制效果。 使用 OpenCV 绘制圆角矩形按键、菱形方向箭头和旋转指示图, 通过 alpha blending 半透明叠加。
UI 元素
WASD 区左下角,显示平移方向按键高亮
IJKL 区右下角,显示旋转方向按键高亮
活跃键按键背景变亮(alpha blending)
轨迹线绘制相机运动轨迹(可选)
数据流
输入videos_np [F, H, W, 3] (0-1 range)
c2ws[F, 4, 4] 用于绘制相机轨迹
wasd/ijkl[F, 3/2] 决定每帧按键高亮状态
输出videos_np [F, H, W, 3] (0-1 带UI)