ICLR OSC 2022
Contents:
Introduction
Setup
API Reference
osc.data
osc.data.clevr_with_masks
osc.data.iterable_dataloader
osc.data.multi_dsprites
osc.data.objects_room
osc.data.random_resized_crop
osc.data.tetrominoes
osc.data.tfrecords
osc.data.utils
osc.data.utils.augment_center_crop
osc.data.utils.augment_train
osc.data.utils.augment_twice
osc.data.utils.img_hwc_to_chw
osc.data.utils.normalize_tf
osc.data.utils.unnormalize_pt
osc.data.utils.wrap_with_seed
osc.data.utils.AugmentFnStateless
osc.loss_global
osc.loss_objects
osc.lr_scheduler
osc.models
osc.train
osc.utils
osc.viz
osc.wandb_utils
ICLR OSC 2022
»
osc
»
osc.data
»
osc.data.utils
»
osc.data.utils.img_hwc_to_chw
View page source
osc.data.utils.img_hwc_to_chw
img_hwc_to_chw
(
img
:
tensorflow.python.framework.ops.Tensor
)
→
tensorflow.python.framework.ops.Tensor
[source]
Image channels […, H, W, C] -> […, C, H, W]
Return type
Tensor