Add new points and labels to previous point inputs (add at the end).

frame_rgb_uint8: (H,W,3) uint8 out_hw: (H,W) target size for depth map return: (H,W) uint8 depth (0..255)

A post processor to fill small holes in mask scores with area under `max_area`.

将光流 (H, W, 2) 转换为颜色编码图。 参数: - flow: numpy.ndarray, 光流张量，形状为 (H, W, 2)，值范围为 [-1, 1] - max_flow: float, 光流的最大值，用于归一化 (默认假设范围是 [-1,

x_list contains a list of tensors to nest together and run

Generates masks for the given image. Arguments: image (np.ndarray): The image to generate masks for, in HWC uint8 format.

Generates a list of crop boxes of different sizes. Each layer has (2**i)**2 boxes for the ith layer.

Precompute the frequency tensor for complex exponentials (cis) with given dimensions. This function calculates a frequency tensor with compl

Get 1D sine positional embedding as in the original Transformer paper.

Helper function to get activation function from string. Args: act_fn (str): Name of activation function. Returns: nn.Module:

Return an activation function given a string

trajs: (B, N, T, 2) in pixel coords returns: (C, n_frames_out, H, W) in [-1,1] We always generate exactly n_frames_out frames for output