pytorchrl.envs.habitat package

Submodules

pytorchrl.envs.habitat.habitat_env_factory module

pytorchrl.envs.habitat.utils module

pytorchrl.envs.habitat.utils.center_crop(img, dim)

Returns center cropped image Args: img: image to be center cropped dim: dimensions (width, height) to be cropped

pytorchrl.envs.habitat.utils.scale_image(img, factor=1)

Returns resize image by scale factor. This helps to retain resolution ratio while resizing. Args: img: image to be scaled factor: scale factor to resize

pytorchrl.envs.habitat.wrappers module

Module contents