Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/JIA-Lab-research/UVTR
/ functions
Functions
217 in github.com/JIA-Lab-research/UVTR
⨍
Functions
217
◇
Types & classes
37
Method
get_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
Method
get_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
Method
init_weights
Initialize weights of the DeformDETR head.
projects/mmdet3d_plugin/models/dense_heads/uvtr_head.py:121
Method
init_weights
Initialize weights of the DeformDETR head.
projects/mmdet3d_plugin/models/dense_heads/uvtr_kd_head.py:125
Method
init_weights
Initialize the transformer weights.
projects/mmdet3d_plugin/models/utils/uni3d_detr.py:42
Method
init_weights
Initialize weights of the depth head.
projects/mmdet3d_plugin/models/detectors/uvtr_kd_cs.py:87
Method
init_weights
Initialize weights of the depth head.
projects/mmdet3d_plugin/models/detectors/uvtr_kd_m.py:112
Method
init_weights
Initialize weights of the depth head.
projects/mmdet3d_plugin/models/detectors/uvtr_kd_l.py:71
Method
load_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
Method
loss
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
Method
set_prob
(self, epoch, max_epoch)
projects/mmdet3d_plugin/models/utils/grid_mask.py:17
Method
set_prob
(self, epoch, max_epoch)
projects/mmdet3d_plugin/models/utils/grid_mask.py:81
Method
train
Convert the model into training mode while keep normalization layer freezed.
projects/mmdet3d_plugin/models/backbones/vovnet.py:366
Method
with_depth_head
bool: Whether the detector has a depth head.
projects/mmdet3d_plugin/models/detectors/uvtr.py:122
Method
with_depth_head
bool: Whether the detector has a depth head.
projects/mmdet3d_plugin/models/detectors/uvtr_kd_cs.py:138
Method
with_depth_head
bool: Whether the detector has a depth head.
projects/mmdet3d_plugin/models/detectors/uvtr_kd_m.py:171
Method
with_depth_head
bool: Whether the detector has a depth head.
projects/mmdet3d_plugin/models/detectors/uvtr_kd_l.py:102
← previous
201–217 of 217, ranked by callers