MCPcopy Create free account

hub / github.com/Lizhuoling/UniMODE / functions

Functions454 in github.com/Lizhuoling/UniMODE

Functionget_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
Functionimhstack
(im1, im2)
model/vis/vis.py:722
Functionimread
(path)
model/util/util.py:66
Functionimreadstats
(path)
model/util/util.py:70
Functionimshow
(im, fig_num=None)
model/vis/vis.py:201
Functionimvstack
(im1, im2)
model/vis/vis.py:737
Functionimwrite
(im, path)
model/util/util.py:77
Methodinference
(self, detector_out, batched_inputs, ori_img_resolution)
model/modeling/detector3d/detector3d.py:768
Functioninit_distributed_mode
(args)
model/util/default.py:42
Functionis_available
()
model/util/torch_dist.py:62
Functionis_distributed
()
model/util/torch_dist.py:54
Functionlist_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
Functionlist_subdirectories
(path, include_files=False)
model/util/util.py:113
Functionload_json
(path)
model/util/util.py:29
Functionload_mat
(path)
model/util/util.py:36
Methodloss
(self, center_head_out, Ks, batched_inputs)
model/modeling/detector3d/center_head.py:150
Methodloss
(self, detector_out, batched_inputs, ori_img_resolution)
model/modeling/detector3d/detector3d.py:828
Methodloss_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
Methodloss_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
Functionmain
(args)
tools/train_net.py:372
Functionmat2euler
(R)
model/util/math_util.py:73
Functionmkdir_if_missing
(directory, delete_if_exist=False)
model/util/util.py:123
Functionms_deform_attn_backward
model/deformable_ops/src/ms_deform_attn.h:41
Functionms_deform_attn_cpu_backward
model/deformable_ops/src/cpu/ms_deform_attn_cpu.cpp:29
Functionms_deform_attn_cpu_forward
model/deformable_ops/src/cpu/ms_deform_attn_cpu.cpp:17
Functionms_deform_attn_forward
model/deformable_ops/src/ms_deform_attn.h:20
Methodno_weight_decay
(self)
model/modeling/backbone/eva.py:650
Methodpadding_constraints
(self)
model/modeling/backbone/eva_det.py:545
Functionpickle_read
(file_path, latin=False, iso8859=False, bytes=False)
model/util/util.py:48
Functionpickle_write
(file_path, obj)
model/util/util.py:42
Functionpos3embed
(pos, num_pos_feats=128, temperature=10000)
model/util/position_encoding.py:171
Functionprepare_mm_omni3d_hypersim
(hypersim_root, hypersim_meta_root, omni3d_root, MM_omni3d_root)
data_preparation/hypersim_prepare.py:11
Functionprint_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
Functionprint_ap_category_histogram
Prints AP performance for each category. Args: results: dictionary; each entry contains information for a dataset
model/vis/logperf.py:9
Functionprint_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
Functionprint_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
Methodprocess
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
Functionreduce_mean
(tensor)
model/util/torch_dist.py:41
Functionregister_and_store_model_metadata
(datasets, output_dir, filter_settings=None)
model/data/datasets.py:315
Methodrescale
(param, layer_id)
model/modeling/backbone/eva.py:627
Methodreset_classifier
(self, num_classes, global_pool='')
model/modeling/backbone/eva.py:656
Functionsave_global_dataset_stats
(stats, path_to_stats=None)
model/data/datasets.py:45
Functionsave_json
(path, data)
model/util/util.py:24
Functionscaled_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
Methodset_prob
(self, epoch, max_epoch)
model/util/grid_mask.py:18
Functionstring_similarity
(text1, text2)
model/util/util.py:304
Functionsynchronize
Helper function to synchronize (barrier) among all processes when using distributed training
model/util/torch_dist.py:20
Methodtrain
(self, mode: bool = True)
model/spconv_voxelize/spconv_voxelize.py:36
Functionupto_2Pi
(val)
model/util/math_util.py:49
Functionupto_Pi
(val)
model/util/math_util.py:59
Functionvis_2d_det
(boxlist, img, class_names, conf_thre = 0.5)
model/modeling/meta_arch/MM_UniMODE.py:74
Functionvisualize_from_instances
(detections, dataset, dataset_name, min_size_test, output_folder, category_names_official, iteration='')
model/vis/vis.py:77
Functionvoxel_pooling_train_backward_wrapper
voxel_pooling/src/voxel_pooling_train_forward.cpp:63
Functionvoxel_pooling_train_forward_wrapper
voxel_pooling/src/voxel_pooling_train_forward.cpp:41
← previous401–454 of 454, ranked by callers