osc.train.get_viz_batch

get_viz_batch(cfg)[source]

Prepare a batch of images for visualization.

Parameters: cfg (DictConfig) – configuration
Return type: ndarray
Returns: A batch of images, shape [A B C H W].