Get the position embedding of proposal.

Get the reference points used in decoder. Args: spatial_shapes (Tensor): The shape of all feature maps, has shape

Get ``num_points`` most uncertain points with random points during train. Sample points in [0, 1] x [0, 1] coordinate space based on their

Estimate uncertainty based on pred logits. We estimate uncertainty as L1 distance between 0.0 and the logits prediction in 'mask_pred' for th

Get the valid radios of feature maps of all level.

Calculate lr decay rate for different ViT blocks. Args: name (string): parameter name. lr_decay_rate (float): base lr decay r

Initialize layers of the DeformableDetrTransformer.

Returns: True if the current process is the main one.

Loss function. Args: all_cls_scores (Tensor): Classification scores for all decoder layers with shape [num_decode

Compute depth loss.

Compute depth loss.

Calculate naive dice loss, the coefficient in the denominator is the first power instead of the second power. Args: pred (torch.Tenso

Pairwise nearest neighbors for L2-normalized vectors. Uses Torch rather than Faiss to remain on GPU.

"Obtain the mean of tensor on different GPUs.

Placeholder for single image test.

Simple test with single image.

Generate the valid flags of points of a single feature map. Args: featmap_size (tuple[int]): The size of feature maps, arrange as

Inference by sliding-window with overlap. If h_crop > h_img or w_crop > w_img, the small patch will be used to decode without padding

Inference by sliding-window with overlap. If h_crop > h_img or w_crop > w_img, the small patch will be used to decode without padding

Inference by sliding-window with overlap. If h_crop > h_img or w_crop > w_img, the small patch will be used to decode without padding

Tag the last checkpoint. Args: last_filename_basename (str): the basename of the last filename.

Inference with full image.

Inference with full image.

Inference with full image.