Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/Lizhuoling/UniMODE
/ functions
Functions
454 in github.com/Lizhuoling/UniMODE
⨍
Functions
454
◇
Types & classes
88
Function
get_vit_lr_decay_rate
Calculate lr decay rate for different ViT blocks. Args: name (string): parameter name. lr_decay_rate (float): base lr decay r
model/modeling/backbone/eva_det.py:580
Function
imhstack
(im1, im2)
model/vis/vis.py:722
Function
imread
(path)
model/util/util.py:66
Function
imreadstats
(path)
model/util/util.py:70
Function
imshow
(im, fig_num=None)
model/vis/vis.py:201
Function
imvstack
(im1, im2)
model/vis/vis.py:737
Function
imwrite
(im, path)
model/util/util.py:77
Method
inference
(self, detector_out, batched_inputs, ori_img_resolution)
model/modeling/detector3d/detector3d.py:768
Function
init_distributed_mode
(args)
model/util/default.py:42
Function
is_available
()
model/util/torch_dist.py:62
Function
is_distributed
()
model/util/torch_dist.py:54
Function
list_files
Returns a list of files given a directory and pattern The results are sorted alphabetically Example: files = list_files('path/to/
model/util/util.py:104
Function
list_subdirectories
(path, include_files=False)
model/util/util.py:113
Function
load_json
(path)
model/util/util.py:29
Function
load_mat
(path)
model/util/util.py:36
Method
loss
(self, center_head_out, Ks, batched_inputs)
model/modeling/detector3d/center_head.py:150
Method
loss
(self, detector_out, batched_inputs, ori_img_resolution)
model/modeling/detector3d/detector3d.py:828
Method
loss_single
Input: obj_pred shape: (1, feat_h, feat_w) tgt_depth_pred shape: (num_gt, 1) offset_pred shape: (num_gt,
model/modeling/detector3d/center_head.py:178
Method
loss_single
Input: cls_scores: shape (B, num_query, cls_num) bbox_preds: shape (B, num_query, attr_num) batched_input
model/modeling/detector3d/detector3d.py:849
Function
main
(args)
tools/train_net.py:372
Function
mat2euler
(R)
model/util/math_util.py:73
Function
mkdir_if_missing
(directory, delete_if_exist=False)
model/util/util.py:123
Function
ms_deform_attn_backward
model/deformable_ops/src/ms_deform_attn.h:41
Function
ms_deform_attn_cpu_backward
model/deformable_ops/src/cpu/ms_deform_attn_cpu.cpp:29
Function
ms_deform_attn_cpu_forward
model/deformable_ops/src/cpu/ms_deform_attn_cpu.cpp:17
Function
ms_deform_attn_forward
model/deformable_ops/src/ms_deform_attn.h:20
Method
no_weight_decay
(self)
model/modeling/backbone/eva.py:650
Method
padding_constraints
(self)
model/modeling/backbone/eva_det.py:545
Function
pickle_read
(file_path, latin=False, iso8859=False, bytes=False)
model/util/util.py:48
Function
pickle_write
(file_path, obj)
model/util/util.py:42
Function
pos3embed
(pos, num_pos_feats=128, temperature=10000)
model/util/position_encoding.py:171
Function
prepare_mm_omni3d_hypersim
(hypersim_root, hypersim_meta_root, omni3d_root, MM_omni3d_root)
data_preparation/hypersim_prepare.py:11
Function
print_ap_analysis_histogram
Prints AP performance for various IoU thresholds and (near, medium, far) objects. Args: results: dictionary. Each entry in results co
model/vis/logperf.py:44
Function
print_ap_category_histogram
Prints AP performance for each category. Args: results: dictionary; each entry contains information for a dataset
model/vis/logperf.py:9
Function
print_ap_dataset_histogram
Prints AP performance for each dataset. Args: results: list of dicts. Each entry in results contains outputs for a dataset
model/vis/logperf.py:69
Function
print_ap_omni_histogram
Prints AP performance for Omni3D dataset. Args: results: list of dicts. Each entry in results contains outputs for a dataset
model/vis/logperf.py:94
Method
process
Args: inputs: the inputs to a model (e.g., GeneralizedRCNN). It is a list of dict. Each dict corresponds to an im
model/evaluation/omni3d_evaluation.py:730
Function
reduce_mean
(tensor)
model/util/torch_dist.py:41
Function
register_and_store_model_metadata
(datasets, output_dir, filter_settings=None)
model/data/datasets.py:315
Method
rescale
(param, layer_id)
model/modeling/backbone/eva.py:627
Method
reset_classifier
(self, num_classes, global_pool='')
model/modeling/backbone/eva.py:656
Function
save_global_dataset_stats
(stats, path_to_stats=None)
model/data/datasets.py:45
Function
save_json
(path, data)
model/util/util.py:24
Function
scaled_sigmoid
Simple helper function for a scaled sigmoid. The output is bounded by (min, max) Args: vals (Tensor): input logits to scale
model/util/math_util.py:970
Method
set_prob
(self, epoch, max_epoch)
model/util/grid_mask.py:18
Function
string_similarity
(text1, text2)
model/util/util.py:304
Function
synchronize
Helper function to synchronize (barrier) among all processes when using distributed training
model/util/torch_dist.py:20
Method
train
(self, mode: bool = True)
model/spconv_voxelize/spconv_voxelize.py:36
Function
upto_2Pi
(val)
model/util/math_util.py:49
Function
upto_Pi
(val)
model/util/math_util.py:59
Function
vis_2d_det
(boxlist, img, class_names, conf_thre = 0.5)
model/modeling/meta_arch/MM_UniMODE.py:74
Function
visualize_from_instances
(detections, dataset, dataset_name, min_size_test, output_folder, category_names_official, iteration='')
model/vis/vis.py:77
Function
voxel_pooling_train_backward_wrapper
voxel_pooling/src/voxel_pooling_train_forward.cpp:63
Function
voxel_pooling_train_forward_wrapper
voxel_pooling/src/voxel_pooling_train_forward.cpp:41
← previous
401–454 of 454, ranked by callers