MCPcopy Create free account

hub / github.com/JIA-Lab-research/UVTR / functions

Functions217 in github.com/JIA-Lab-research/UVTR

Methodget_bboxes
Generate bboxes from bbox head predictions. Args: preds_dicts (tuple[list[dict]]): Prediction results. img_metas (list
projects/mmdet3d_plugin/models/dense_heads/uvtr_kd_head.py:558
Methodget_cat_ids
Get category distribution of single scene. Args: idx (int): Index of the data_info. Returns: dict[list]: for
projects/mmdet3d_plugin/datasets/nuscenes_dataset.py:160
Methodinit_weights
Initialize weights of the DeformDETR head.
projects/mmdet3d_plugin/models/dense_heads/uvtr_head.py:121
Methodinit_weights
Initialize weights of the DeformDETR head.
projects/mmdet3d_plugin/models/dense_heads/uvtr_kd_head.py:125
Methodinit_weights
Initialize the transformer weights.
projects/mmdet3d_plugin/models/utils/uni3d_detr.py:42
Methodinit_weights
Initialize weights of the depth head.
projects/mmdet3d_plugin/models/detectors/uvtr_kd_cs.py:87
Methodinit_weights
Initialize weights of the depth head.
projects/mmdet3d_plugin/models/detectors/uvtr_kd_m.py:112
Methodinit_weights
Initialize weights of the depth head.
projects/mmdet3d_plugin/models/detectors/uvtr_kd_l.py:71
Methodload_annotations
Load annotations from ann_file. Args: ann_file (str): Path of the annotation file. Returns: list[dict]: List
projects/mmdet3d_plugin/datasets/nuscenes_dataset.py:184
Methodloss
Loss function. Args: gt_bboxes_list (list[Tensor]): Ground truth bboxes for each image with shape (nu
projects/mmdet3d_plugin/models/dense_heads/uvtr_kd_head.py:462
Methodset_prob
(self, epoch, max_epoch)
projects/mmdet3d_plugin/models/utils/grid_mask.py:17
Methodset_prob
(self, epoch, max_epoch)
projects/mmdet3d_plugin/models/utils/grid_mask.py:81
Methodtrain
Convert the model into training mode while keep normalization layer freezed.
projects/mmdet3d_plugin/models/backbones/vovnet.py:366
Methodwith_depth_head
bool: Whether the detector has a depth head.
projects/mmdet3d_plugin/models/detectors/uvtr.py:122
Methodwith_depth_head
bool: Whether the detector has a depth head.
projects/mmdet3d_plugin/models/detectors/uvtr_kd_cs.py:138
Methodwith_depth_head
bool: Whether the detector has a depth head.
projects/mmdet3d_plugin/models/detectors/uvtr_kd_m.py:171
Methodwith_depth_head
bool: Whether the detector has a depth head.
projects/mmdet3d_plugin/models/detectors/uvtr_kd_l.py:102
← previous201–217 of 217, ranked by callers