habitat.tasks.rearrange.rearrange_sensors.ArmDepthBBoxSensor class

Bounding box sensor to check if the object is in frame

Methods

def _get_bbox(self, img)
Simple function to get the bounding box, assuming that only one object of interest in the image
def get_observation(self, observations, episode, task, *args, **kwargs)

Special methods

def __init__(self, sim, config, *args, **kwargs)

Data

cls_uuid: str = 'arm_depth_bbox_sensor'