MCPcopy Create free account

hub / github.com/ayushjain1144/odin / functions

Functions721 in github.com/ayushjain1144/odin

↓ 1 callersFunctionprocess_depth
(scene, depth_image)
odin/utils/inpaint_depth.py:26
↓ 1 callersFunctionprocess_depth
(scene, depth_image)
data_preparation/inpaint_depth.py:29
↓ 1 callersFunctionprocess_depth
(frame_id, file, folder_processed)
data_preparation/s3dis/convert_dataset_raw.py:112
↓ 1 callersMethodprocess_file
process_file. Args: filepath: path to the main file mode: typically train, test or validation Returns:
data_preparation/base_preprocessing.py:79
↓ 1 callersFunctionprocess_pc
(filepath, mode, mapping)
data_preparation/matterport3d/process_matterport_3d.py:86
↓ 1 callersFunctionprocess_pose
(frame_id, pose_file, folder_processed_pose, area)
data_preparation/s3dis/convert_dataset.py:61
↓ 1 callersFunctionprocess_pose
(frame_id, pose_file, folder_processed_pose, area)
data_preparation/s3dis/convert_dataset_raw.py:61
↓ 1 callersFunctionprocess_scene
(area_name, scene_name, files, labels)
data_preparation/s3dis/convert_dataset.py:161
↓ 1 callersFunctionprocess_scene
(area_name, scene_name, files, labels)
data_preparation/s3dis/convert_dataset_raw.py:142
↓ 1 callersMethodprune_points
(self)
data_preparation/ai2thor/controller_custom.py:1700
↓ 1 callersMethodprune_releases
(self)
data_preparation/ai2thor/controller_custom.py:841
↓ 1 callersFunctionrandom_ball_query_cuda
libs/pointops/src/random_ball_query/random_ball_query_cuda.cpp:7
↓ 1 callersMethodrandom_initialize
( self, random_seed=None, randomize_open=False, unique_object_types=False,
data_preparation/ai2thor/controller_custom.py:743
↓ 1 callersFunctionravel_hash_vec
Ravel the coordinates after subtracting the min coordinates.
odin/modeling/backproject/backproject.py:230
↓ 1 callersFunctionread_txt
Read txt file into lines.
data_preparation/s3dis/s3dis2coco.py:33
↓ 1 callersFunctionread_txt
Read txt file into lines.
data_preparation/scannet/scannet2coco.py:35
↓ 1 callersFunctionread_txt
Read txt file into lines.
data_preparation/matterport3d/m3d2coco.py:33
↓ 1 callersFunctionread_txt
Read txt file into lines.
data_preparation/ai2thor/ai2thor2coco.py:33
↓ 1 callersFunctionregion_to_camera_mapping_fn
(pano_file)
data_preparation/matterport3d/process_matterport.py:26
↓ 1 callersFunctionregister_all_dataset
(root, dataset_name="ai2thor")
odin/data_video/datasets/builtin.py:181
↓ 1 callersFunctionregister_all_scannet_context20_scannet_single
(root)
odin/data_video/datasets/builtin.py:194
↓ 1 callersFunctionregister_scannet_context_instances
(name, metadata, json_file, image_root)
odin/data_video/datasets/scannet_context.py:146
↓ 1 callersFunctionrepresents_int
(s)
odin/data_video/segmentation_benchmark/util.py:24
↓ 1 callersFunctionresize_crop_image
(image, new_image_dims)
odin/utils/util_3d.py:294
↓ 1 callersFunctionretry_if_cuda_oom
Makes a function retry itself after encountering pytorch's CUDA OOM error. It will first retry after calling `torch.cuda.empty_cache()`.
odin/utils/memory.py:27
↓ 1 callersFunctionrotation_augmentation_fast
(pc, scannet_pc=None)
odin/modeling/backproject/backproject.py:30
↓ 1 callersFunctionscale_augmentations_fast
(pc, scannet_pc=None)
odin/modeling/backproject/backproject.py:40
↓ 1 callersMethodsetDetParams
(self)
odin/data_video/cocoeval.py:507
↓ 1 callersMethodsetKpParams
(self)
odin/data_video/cocoeval.py:518
↓ 1 callersFunctionshort_name
(x)
odin/data_video/build.py:83
↓ 1 callersFunctionstartx
(display)
data_preparation/ai2thor/startx.py:67
↓ 1 callersMethodstep
(self, closure=None)
train_odin.py:390
↓ 1 callersMethodstop_unity
(self)
data_preparation/ai2thor/controller_custom.py:1376
↓ 1 callersMethodsubsample_frames
(self, dataset_dict, **kwargs)
odin/data_video/dataset_mapper_scannet.py:374
↓ 1 callersFunctionsubtraction_backward_cuda
libs/pointops/src/subtraction/subtraction_cuda.cpp:16
↓ 1 callersFunctionsubtraction_forward_cuda
libs/pointops/src/subtraction/subtraction_cuda.cpp:7
↓ 1 callersMethodtest
Evaluate the given model. The given model is expected to already contain weights to evaluate. Args: cfg (CfgNode)
train_odin.py:413
↓ 1 callersMethodto_dict
(self)
odin/data_video/segmentation_benchmark/util_3d.py:106
↓ 1 callersMethodtrain
Convert the model into training mode while keep layers freezed.
odin/modeling/backbone/swin.py:772
↓ 1 callersMethodunity_command
(self, width, height, headless)
data_preparation/ai2thor/controller_custom.py:1017
↓ 1 callersFunctionunproject
Inputs: intrinsics: B X V X 3 X 3 poses: B X V X 4 X 4 (torch.tensor) depths: B X V X H X W (torch.tensor) Outpu
odin/modeling/backproject/backproject.py:95
↓ 1 callersFunctionvisualize_pca_from_feature_folders
visualize PCA of features Keyword arguments: feature_folders -- list of feature folders Return: None
odin/utils/vis_utils.py:55
↓ 1 callersMethodvisualize_pred_on_ours
( self, index, images, shape, input_per_image, processed_results, targets, valids)
odin/odin_model.py:1069
↓ 1 callersMethodvisualize_pred_on_scannet
( self, input_per_image, processed_result, gt_targets, index, scannet_idxs=None )
odin/odin_model.py:1021
↓ 1 callersFunctionvoxel_map_to_source
Input: voxel_map (B, N1, C) point2voxel (B, N) Output: src_new (B, N, C)
odin/modeling/backproject/backproject.py:293
↓ 1 callersMethodvoxel_map_to_source
Input: voxel_map (B, N1, C) point2voxel (B, N) Output: src_new (B, N, C)
odin/modeling/transformer_decoder/odin_transformer_decoder.py:239
↓ 1 callersFunctionwindow_reverse
Args: windows: (num_windows*B, window_size, window_size, C) window_size (int): Window size H (int): Height of image
odin/modeling/backbone/swin.py:64
↓ 1 callersMethodwith_pos_embed
(tensor, pos)
odin/modeling/pixel_decoder/msdeformattn.py:156
MethodConnectedComponent
libs/pointgroup_ops/src/bfs_cluster.cpp:25
FunctionDivide2Patch
(nsample, xyz, offset, return_offset=False, anchor_scale=None)
libs/pointops2/functions/pointops.py:678
FunctionPYBIND11_MODULE
odin/modeling/pixel_decoder/ops/src/vision.cpp:18
FunctionPYBIND11_MODULE
libs/pointops2/src/pointops_api.cpp:16
FunctionPYBIND11_MODULE
------------------------------------API------------------------------------------
libs/pointgroup_ops/src/bfs_cluster.cpp:140
FunctionPYBIND11_MODULE
libs/pointops/src/pointops_api.cpp:15
Method__call__
Args: dataset_dict (dict): Metadata of one image, in Detectron2 Dataset format. Returns: dict: a format that
odin/data_video/dataset_mapper_coco.py:185
Method__call__
Args: dataset_dict (dict): Metadata of one video Returns: dict: a format that builtin models in ...
odin/data_video/dataset_mapper_scannet.py:413
Method__enter__
(self)
data_preparation/ai2thor/controller_custom.py:629
Method__exit__
(self, *args)
data_preparation/ai2thor/controller_custom.py:632
Method__init__
(self, *args, **kwargs)
train_odin.py:70
Method__init__
Args: backbone: a backbone module, must follow detectron2's backbone interface sem_seg_head: a module that predicts s
odin/odin_model.py:62
Method__init__
Creates the matcher Params: cost_class: This is the relative weight of the classification error in the matching cost
odin/modeling/matcher.py:81
Method__init__
Create the criterion. Parameters: num_classes: number of object categories, omitting the special no-object category ma
odin/modeling/criterion.py:133
Method__init__
NOTE: this interface is experimental. Args: in_channels: channels of the input features mask_classification:
odin/modeling/transformer_decoder/odin_transformer_decoder.py:69
Method__init__
(self, num_pos_feats=64, temperature=10000, normalize=False, scale=None, add_temporal=True)
odin/modeling/transformer_decoder/position_encoding.py:76
Method__init__
(self, dim=3, num_pos_feats=288)
odin/modeling/transformer_decoder/position_encoding.py:123
Method__init__
(self, dim=3, num_pos_feats=288)
odin/modeling/transformer_decoder/position_encoding.py:157
Method__init__
( self, dim, window_size, num_heads, qkv_bias=True, qk_scale=N
odin/modeling/backbone/swin.py:93
Method__init__
( self, dim, num_heads, window_size=7, shift_size=0, mlp_ratio
odin/modeling/backbone/swin.py:197
Method__init__
(self, dim, norm_layer=nn.LayerNorm)
odin/modeling/backbone/swin.py:311
Method__init__
( self, dim, depth, num_heads, window_size=7, mlp_ratio=4.0,
odin/modeling/backbone/swin.py:364
Method__init__
(self, patch_size=4, in_chans=3, embed_dim=96, norm_layer=None)
odin/modeling/backbone/swin.py:538
Method__init__
( self, pretrain_img_size=224, patch_size=4, in_chans=3, embed_dim=96,
odin/modeling/backbone/swin.py:599
Method__init__
(self, cfg, input_shape)
odin/modeling/backbone/swin.py:780
Method__init__
Args: in_channels (int): Number of input channels. out_channels (int): Number of output channels. stride
odin/modeling/backbone/resnet.py:43
Method__init__
Args: bottleneck_channels (int): number of output channels for the 3x3 "bottleneck" conv layers. num_
odin/modeling/backbone/resnet.py:112
Method__init__
( self, in_channels, out_channels, *, bottleneck_channels, str
odin/modeling/backbone/resnet.py:224
Method__init__
Args: norm (str or callable): norm after the first conv layer. See :func:`layers.get_norm` for supported format.
odin/modeling/backbone/resnet.py:341
Method__init__
(self, d_model, nhead, dropout=0.0, activation="relu", normalize_before=False,
odin/modeling/meta_arch/self_cross_attention_layers.py:15
Method__init__
(self, d_model, nhead, dropout=0.0, activation="relu", normalize_before=False,
odin/modeling/meta_arch/self_cross_attention_layers.py:77
Method__init__
(self, d_model, dim_feedforward=2048, dropout=0.0, activation="relu", normalize_before=False,
odin/modeling/meta_arch/self_cross_attention_layers.py:144
Method__init__
( self, input_dim, hidden_dims, output_dim, norm_fn_name=None,
odin/modeling/meta_arch/self_cross_attention_layers.py:252
Method__init__
NOTE: this interface is experimental. Args: input_shape: shapes (channels and stride) of the input features n
odin/modeling/meta_arch/odin_head.py:50
Method__init__
(self, latent_dim, num_layers=6, nheads=8, nsample=16, dropout=0.0, dim_feedforward=None, cfg=None)
odin/modeling/meta_arch/cross_view_attention.py:23
Method__init__
(self, cfg, d_model)
odin/modeling/meta_arch/language_encoder.py:11
Method__init__
(self, d_model=256, nhead=8, num_encoder_layers=6, dim_feedforward=1024, dropout=0.1,
odin/modeling/pixel_decoder/msdeformattn.py:58
Method__init__
(self, d_model=256, d_ffn=1024, dropout=0.1, activation="relu",
odin/modeling/pixel_decoder/msdeformattn.py:136
Method__init__
(self, encoder_layer, num_layers)
odin/modeling/pixel_decoder/msdeformattn.py:178
Method__init__
NOTE: this interface is experimental. Args: input_shape: shapes (channels and stride) of the input features t
odin/modeling/pixel_decoder/msdeformattn.py:210
Method__init__
Multi-Scale Deformable Attention Module :param d_model hidden dimension :param n_levels number of feature levels
odin/modeling/pixel_decoder/ops/modules/ms_deform_attn.py:35
Method__init__
(self, tensors, mask: Optional[Tensor])
odin/utils/misc.py:29
Method__init__
( self, dataset_name, eval_sparse=False, distributed=True, output_dir=
odin/data_video/scannet_3d_eval.py:26
Method__init__
NOTE: this interface is experimental. Args: is_train: for training or inference augmentations: a list of augm
odin/data_video/dataset_mapper_coco.py:135
Method__init__
Args: sampler (Sampler): Base sampler. group_ids (list[int]): If the sampler produces indices in range [0, N),
odin/data_video/build.py:457
Method__init__
Args: dataset: an old-style dataset with ``__getitem__`` sampler: a cheap iterable that produces indices to be applie
odin/data_video/build.py:562
Method__init__
Args: size (int): the total number of data of the underlying dataset to sample from shuffle (bool): whether to shuffl
odin/data_video/build.py:625
Method__init__
Args: dataset_name (str): name of the dataset to be evaluated. It must have either the following corresponding me
odin/data_video/coco_evaluation.py:49
Method__init__
( self, dataset_name, eval_sparse=False, distributed=True, output_dir=
odin/data_video/scannet_3d_eval_semantic.py:41
Method__init__
Constructor of Microsoft COCO helper class for reading and visualizing annotations. :param annotation_file (str): location of annotat
odin/data_video/coco.py:67
Method__init__
( self, is_train: bool, dataset_name: str, *, augmentations, d
odin/data_video/dataset_mapper_scannet.py:94
Method__init__
Initialize CocoEval using coco APIs for gt and dt :param cocoGt: coco object with ground truth annotations :param cocoDt: coc
odin/data_video/cocoeval.py:60
← previousnext →301–400 of 721, ranked by callers