Compute 2d IoU for two boxes in coordinate format.

Removes small disconnected regions and holes in a mask. Returns the mask and an indicator of if the mask has been modified.

Args: img: same as in __init__

Scale the short side of the image to size. Args: size (int): size to scale the image. image (array): image to perform short s

Scale the short side of the image to size. Args: size (int): size to scale the image. image (array): image to perform short s

Loss used in RetinaNet for dense detection: https://arxiv.org/abs/1708.02002. Args: inputs: A float tensor of arbitrary shape.

Slices the object into pieces, assuming the agent is holding a knife and is at the object's location. Returns: bool: True if

sorts masks by which are closer to reprojected point of ID masks: NxWxH ID: object id to reproject

Get the metrics used by the VOC Pascal 2012 challenge. Get Args: boundingboxes: Object of the class BoundingBoxes represen

PlotPrecisionRecallCurve Plot the Precision x Recall curve for a given class. Args: boundingBoxes: Object of the class Bou

Check whether any custom object properties need to be updated; Should be called after taking each action

Generate navigation graph: We construct a directed graph with nodes representing agent position and rotation. For every occupiable gr

Convert AI2-THOR x, z coordinate to a visual coordinate on topdown map for TEACh data collection :param x: x coordinate on AI

Get the index in the navigation graph nearest to the given x,z coord in AI2-THOR coordinates :param thor_x: x coordinate on AI2-THOR

Get the cardinal direction to rotate to, to be facing (thor_facing_x, thor_facing_x=z) when standing at (thor_x, thor_z) :par

Get the object closet to the given position. :param m: output of get_objects() :param pos: object x, y, z dict pose :

Given (x, z) norm rotation (e.g., (0, 1)), return the closest degrees in [270, 180, 90, 0] matching.

ResBlock class constructs redisual blocks. More details can be found in: Kaiming He, Xiangyu Zhang, Shaoqing Ren, and Jian Sun.

The `__init__` method of any subclass should also contain these arguments. ResNetRoIHead takes p pathways as input where

ResBlock class constructs redisual blocks. More details can be found in: Kaiming He, Xiangyu Zhang, Shaoqing Ren, and Jian Sun.

The `__init__` method of any subclass should also contain these arguments. ResNetRoIHead takes p pathways as input where

Args: classes (list[int] or None): scores (list[float] or None): class_names (list[str] or None): is_crowd (list[

Args: classes (list[int] or None): scores (list[float] or None): class_names (list[str] or None): is_crowd (list[

Return an activation function given a string

Randomly modifies given color to produce a slightly different color than the color given. Args: color (tuple[double]): a

Randomly modifies given color to produce a slightly different color than the color given. Args: color (tuple[double]): a

Return the classes related to the predictions. These are used to specify how to compute mAP. :param preds: the predictions :r

Return the classes related to the predictions. These are used to specify how to compute mAP. :param preds: the predictions :r

Updates a given dict with corresponding counts.

Click the specified DOM element.

compute overall success and goal_condition success rates along with path-weighted metrics

Run all_gather on arbitrary picklable data (not necessarily tensors) Args: data: any picklable object Returns: list[data]

Blend two images with a given weight alpha. Args: images1 (tensor): the first images to be blended, the dimension is `num

Blend two images with a given weight alpha. Args: images1 (tensor): the first images to be blended, the dimension is `num

Input: [boxlist1, boxlist2]: N x 4 each, or B x N x 4 Ordering: ymin, xmin, ymax, xmax Output: [IoUs]: N x 1 or B x N

If text is in cache, return cached embedding. Otherwise, compute and store.

rgb: single rgb image NOTE: currently this only handles a single RGB

Closes the interaction object. Note: Assumes object is open and agent is at its location. Objects like fridges, cabi

1x1 convolution

3x3 convolution with padding

Return the key press action

Return a random action.