SYMBOL INDEX (606 symbols across 84 files)

FILE: common/box_utils.py
  function box3d_iou (line 4) | def box3d_iou(corners1, corners2):
  function get_box3d_min_max (line 30) | def get_box3d_min_max(corner):
  function get_3d_box (line 49) | def get_3d_box(center, box_size):

FILE: common/dist_utils.py
  function is_dist_avail_and_initialized (line 11) | def is_dist_avail_and_initialized():
  function get_rank (line 19) | def get_rank():
  function get_world_size (line 28) | def get_world_size():
  function is_master_proc (line 37) | def is_master_proc(num_gpus=8):
  function is_root_proc (line 47) | def is_root_proc():
  function _serialize_to_tensor (line 59) | def _serialize_to_tensor(data, group, max_size=1024):
  function _pad_to_largest_tensor (line 85) | def _pad_to_largest_tensor(tensor, group):
  function broadcast (line 121) | def broadcast(object):
  function all_gather (line 131) | def all_gather(tensors):
  function all_reduce (line 152) | def all_reduce(tensors, average=True):
  function _get_global_gloo_group (line 171) | def _get_global_gloo_group():
  function all_gather_unaligned (line 184) | def all_gather_unaligned(data, group=None):

FILE: common/io_utils.py
  function make_dir (line 12) | def make_dir(dir_path):
  function load_imgs (line 17) | def load_imgs(img_paths, option=cv2.IMREAD_COLOR):
  function load_pickle (line 22) | def load_pickle(filename):
  function save_pickle (line 27) | def save_pickle(data, filename):
  function load_json (line 32) | def load_json(filename):
  function save_json (line 37) | def save_json(data, filename, save_pretty=True, sort_keys=False):
  function load_jsonl (line 45) | def load_jsonl(filename):
  function save_jsonl (line 50) | def save_jsonl(data, filename):
  function load_yaml (line 55) | def load_yaml(filename):
  function save_yaml (line 60) | def save_yaml(data, filename):
  function load_csv (line 65) | def load_csv(filename, delimiter=","):
  function save_csv (line 81) | def save_csv(data, filename, cols=None, delimiter=","):
  function load_numpy (line 92) | def load_numpy(filename):
  function save_numpy (line 96) | def save_numpy(data, filename):
  function load_tensor (line 100) | def load_tensor(filename):
  function save_tensor (line 104) | def save_tensor(data, filename):
  function load_ply (line 108) | def load_ply(filepath):
  function load_ply_with_normals (line 122) | def load_ply_with_normals(filepath):

FILE: common/launch_utils.py
  class SubmititLauncher (line 11) | class SubmititLauncher:
    method __init__ (line 12) | def __init__(self, args):
    method __call__ (line 15) | def __call__(self):
    method _set_gpu_args (line 44) | def _set_gpu_args(self):
  function submitit_launch (line 50) | def submitit_launch(args):
  function accelerate_launch (line 80) | def accelerate_launch(args):
  function python_launch (line 106) | def python_launch(args):

FILE: common/misc.py
  function rsetattr (line 16) | def rsetattr(obj, attr, val):
  function rgetattr (line 22) | def rgetattr(obj, attr, *args):
  function _gpu_gather_object (line 43) | def _gpu_gather_object(object: Any):
  function gather_object (line 65) | def gather_object(object: Any):
  function gather_for_metrics (line 84) | def gather_for_metrics(accelerator, input_data):
  function gather_dict (line 134) | def gather_dict(accelerator, data_dict):

FILE: common/type_utils.py
  function cfg2dict (line 6) | def cfg2dict(cfg):
  function _to_device (line 10) | def _to_device(state, device):

FILE: data/build.py
  function get_dataset (line 14) | def get_dataset(cfg, split):
  function build_dataloader (line 42) | def build_dataloader(cfg, split='train'):

FILE: data/data_utils.py
  function per_scene_pad (line 12) | def per_scene_pad(lang_list, max_len=64, tokenizer=None, max_seq_len=50):
  function merge_tokens (line 50) | def merge_tokens(token1, mask1, token2, mask2, max_len=300, tokenizer=No...
  function convert_pc_to_box (line 63) | def convert_pc_to_box(obj_pc):
  function random_word (line 76) | def random_word(tokens, tokens_mask, tokenizer, mask_ratio):
  function random_point_cloud (line 107) | def random_point_cloud(pcd, pcd_mask, mask_ratio):
  class LabelConverter (line 124) | class LabelConverter(object):
    method __init__ (line 125) | def __init__(self, file_path):
  function build_rotate_mat (line 163) | def build_rotate_mat(split, rot_aug=True, rand_angle='axis'):
  function eval_ref_one_sample (line 181) | def eval_ref_one_sample(pred_bbox, gt_bbox):
  function get_box3d_min_max (line 194) | def get_box3d_min_max(corner):
  function box3d_iou (line 212) | def box3d_iou(corners1, corners2):
  function transform_points (line 237) | def transform_points(points, transform, translate=True, mode="numpy"):
  function construct_bbox_corners (line 266) | def construct_bbox_corners(center, box_size):
  function is_explicitly_view_dependent (line 280) | def is_explicitly_view_dependent(tokens):
  class ScanQAAnswer (line 292) | class ScanQAAnswer(object):
    method __init__ (line 293) | def __init__(self, answers=None, unk_token='<unk>', ignore_idx=-100):
    method itos (line 301) | def itos(self, i):
    method stoi (line 306) | def stoi(self, v):
    method __len__ (line 311) | def __len__(self):
  class SQA3DAnswer (line 315) | class SQA3DAnswer(object):
    method __init__ (line 316) | def __init__(self, answers=None, unk_token='u'):
    method itos (line 324) | def itos(self, i):
    method stoi (line 329) | def stoi(self, v):
    method __len__ (line 334) | def __len__(self):
  function load_matrix_from_txt (line 338) | def load_matrix_from_txt(path, shape=(4, 4)):
  function pad_tensors (line 345) | def pad_tensors(tensors, lens=None, pad=0):
  function get_sqa_question_type (line 355) | def get_sqa_question_type(question):
  class Vocabulary (line 371) | class Vocabulary(object):
    method __init__ (line 372) | def __init__(self, path=None):
    method add_token (line 383) | def add_token(self, token, bert_id):
    method token_to_id (line 391) | def token_to_id(self, token):
    method id_to_token (line 394) | def id_to_token(self, id):
    method id_to_bert_id (line 397) | def id_to_bert_id(self, id):
    method save_vocab (line 400) | def save_vocab(self, path):
  function random_caption_word (line 406) | def random_caption_word(tokens, tokens_mask, tokenizer, vocab, mask_ratio):
  function clean_answer (line 428) | def clean_answer(data):

FILE: data/datasets/arkitscene.py
  class ARKitScenePretrainObj (line 8) | class ARKitScenePretrainObj(ScanBase):
    method __init__ (line 9) | def __init__(self, cfg, split):
    method __len__ (line 28) | def __len__(self):
    method __getitem__ (line 31) | def __getitem__(self, index):
  class ARKitSceneSpatialRefer (line 45) | class ARKitSceneSpatialRefer(ScanBase):
    method __init__ (line 46) | def __init__(self, cfg, split):
    method __len__ (line 77) | def __len__(self):
    method __getitem__ (line 80) | def __getitem__(self, index):

FILE: data/datasets/base.py
  class ScanBase (line 20) | class ScanBase(Dataset):
    method __init__ (line 21) | def __init__(self, cfg, split):
    method _load_split (line 52) | def _load_split(self, split):
    method _load_scan (line 65) | def _load_scan(self, scan_ids, filter_bkg=False):
    method _load_lang (line 144) | def _load_lang(self, cfg, scan_ids):
    method _getitem_pretrain (line 215) | def _getitem_pretrain(self, index, is_rscan=False):
    method _getitem_obj_pretrain (line 263) | def _getitem_obj_pretrain(self, index):
    method _getitem_refer (line 324) | def _getitem_refer(self, index):
    method _getitem_perscene (line 514) | def _getitem_perscene(self, index):
    method _obj_processing_post (line 697) | def _obj_processing_post(self, obj_pcds, obj_labels, is_need_bbox=Fals...
    method _obj_processing_aug (line 742) | def _obj_processing_aug(self, obj_pcds, obj_labels, is_need_bbox=False):
    method _scene_processing_aug (line 778) | def _scene_processing_aug(self, obj_pcds, bg_pcds, obj_labels, is_need...
    method _getitem_finalrefer (line 840) | def _getitem_finalrefer(self, index):

FILE: data/datasets/data_augmentor.py
  class DataAugmentor (line 8) | class DataAugmentor(object):
    method __init__ (line 9) | def __init__(self, cfg, split, **kwargs):
    method forward (line 23) | def forward(self, data_dict):
    method scene_aug (line 39) | def scene_aug(self, aug_dict, config):
    method obj_aug (line 105) | def obj_aug(self, aug_dict, config):
    method update_data_dict (line 156) | def update_data_dict(self, data_dict, aug_dict):
    method init_aug (line 218) | def init_aug(obj_len):
    method check_key (line 226) | def check_key(key):
    method check_p (line 241) | def check_p(key):
    method rot_fn (line 245) | def rot_fn(x, mat):
    method obj_rot_fn (line 249) | def obj_rot_fn(x, mat):
    method scaling_fn (line 254) | def scaling_fn(x, scale):
    method jitter_fn (line 259) | def jitter_fn(x, scale):
    method subsample_fn (line 263) | def subsample_fn(x, num_points):
    method pts_shuffle_fn (line 268) | def pts_shuffle_fn(x):

FILE: data/datasets/dataset_wrapper.py
  class MaskDatasetWrapper (line 16) | class MaskDatasetWrapper(Dataset):
    method __init__ (line 17) | def __init__(self, cfg, dataset, split="train"):
    method __len__ (line 35) | def __len__(self):
    method __getitem__ (line 38) | def __getitem__(self, idx):
    method collate_fn (line 113) | def collate_fn(self, batch_list):
  class ScanFamilyDatasetWrapperOld (line 119) | class ScanFamilyDatasetWrapperOld(Dataset):
    method __init__ (line 120) | def __init__(self, cfg, dataset, split="train"):
    method __len__ (line 134) | def __len__(self):
    method pad_tensors (line 137) | def pad_tensors(self, tensors, lens=None, pad=0):
    method __getitem__ (line 147) | def __getitem__(self, idx):
    method collate_fn (line 197) | def collate_fn(self, batch_list):
  class VisualizeDatasetWrapper (line 203) | class VisualizeDatasetWrapper(Dataset):
    method __init__ (line 204) | def __init__(self, cfg, dataset, split="train"):
    method __len__ (line 207) | def __len__(self):
    method __getitem__ (line 210) | def __getitem__(self, idx):
    method collate_fn (line 217) | def collate_fn(self, batch_list):

FILE: data/datasets/hm.py
  class HMPretrainObj (line 8) | class HMPretrainObj(ScanBase):
    method __init__ (line 9) | def __init__(self, cfg, split):
    method __len__ (line 28) | def __len__(self):
    method __getitem__ (line 31) | def __getitem__(self, index):
  class HMSpatialRefer (line 45) | class HMSpatialRefer(ScanBase):
    method __init__ (line 46) | def __init__(self, cfg, split):
    method __len__ (line 77) | def __len__(self):
    method __getitem__ (line 80) | def __getitem__(self, index):

FILE: data/datasets/multiscan.py
  class MultiScanPretrainObj (line 8) | class MultiScanPretrainObj(ScanBase):
    method __init__ (line 9) | def __init__(self, cfg, split):
    method __len__ (line 28) | def __len__(self):
    method __getitem__ (line 31) | def __getitem__(self, index):
  class MultiScanSpatialRefer (line 45) | class MultiScanSpatialRefer(ScanBase):
    method __init__ (line 46) | def __init__(self, cfg, split):
    method __len__ (line 77) | def __len__(self):
    method __getitem__ (line 80) | def __getitem__(self, index):

FILE: data/datasets/procthor.py
  class ProcThorPretrainObj (line 8) | class ProcThorPretrainObj(ScanBase):
    method __init__ (line 9) | def __init__(self, cfg, split):
    method __len__ (line 28) | def __len__(self):
    method __getitem__ (line 31) | def __getitem__(self, index):
  class ProcThorSpatialRefer (line 45) | class ProcThorSpatialRefer(ScanBase):
    method __init__ (line 46) | def __init__(self, cfg, split):
    method __len__ (line 77) | def __len__(self):
    method __getitem__ (line 80) | def __getitem__(self, index):

FILE: data/datasets/rscan.py
  class RScanPretrainObj (line 8) | class RScanPretrainObj(ScanBase):
    method __init__ (line 9) | def __init__(self, cfg, split):
    method __len__ (line 28) | def __len__(self):
    method __getitem__ (line 31) | def __getitem__(self, index):
  class RScanSpatialRefer (line 45) | class RScanSpatialRefer(ScanBase):
    method __init__ (line 46) | def __init__(self, cfg, split):
    method __len__ (line 77) | def __len__(self):
    method __getitem__ (line 80) | def __getitem__(self, index):

FILE: data/datasets/scannet.py
  class ScanNetPretrainObj (line 9) | class ScanNetPretrainObj(ScanBase):
    method __init__ (line 10) | def __init__(self, cfg, split):
    method __len__ (line 29) | def __len__(self):
    method __getitem__ (line 32) | def __getitem__(self, index):
  class ScanNetSpatialRefer (line 46) | class ScanNetSpatialRefer(ScanBase):
    method __init__ (line 47) | def __init__(self, cfg, split):
    method __len__ (line 78) | def __len__(self):
    method __getitem__ (line 81) | def __getitem__(self, index):

FILE: data/datasets/scannet_base.py
  class ScanNetBase (line 22) | class ScanNetBase(Dataset):
    method __init__ (line 23) | def __init__(self, cfg, split):
    method __len__ (line 40) | def __len__(self):
    method __getitem__ (line 43) | def __getitem__(self, index):
    method _load_one_scan (line 46) | def _load_one_scan(self, scan_id, pc_type = 'gt', load_inst_info = False,
    method _load_scannet (line 135) | def _load_scannet(self, scan_ids, pc_type = 'gt', load_inst_info = False,
    method _load_lang (line 170) | def _load_lang(self, cfg):
    method _load_split (line 221) | def _load_split(self, cfg, split, use_multi_process = False):
    method _load_inst_info (line 235) | def _load_inst_info(self, scan_id):
    method _obj_processing_post (line 263) | def _obj_processing_post(self, obj_pcds, obj_labels, is_need_bbox=Fals...
    method _obj_processing_aug (line 308) | def _obj_processing_aug(self, obj_pcds, obj_labels, is_need_bbox=False):
    method _scene_processing_aug (line 344) | def _scene_processing_aug(self, obj_pcds, bg_pcds, obj_labels, is_need...
    method _get_pooling_obj_feature (line 407) | def _get_pooling_obj_feature(self, args, mv_info_all, sampled_frame_na...
    method init_dataset_params (line 448) | def init_dataset_params(self, dataset_cfg):
    method init_scan_data (line 465) | def init_scan_data(self):
    method get_scene (line 479) | def get_scene(self, scan_id, tgt_object_id_list, tgt_object_name_list,...

FILE: data/datasets/scannet_old.py
  class ScanNetSQA3D (line 23) | class ScanNetSQA3D(ScanNetBase):
    method __init__ (line 50) | def __init__(self, cfg, split):
    method __getitem__ (line 94) | def __getitem__(self, index):
    method build_answer (line 237) | def build_answer(self):
    method _load_lang (line 252) | def _load_lang(self):
    method _load_question (line 270) | def _load_question(self):
  class ScanNetScanQAOld (line 287) | class ScanNetScanQAOld(ScanNetBase):
    method __init__ (line 288) | def __init__(self, cfg, split):
    method __getitem__ (line 326) | def __getitem__(self, index):
    method _load_lang (line 486) | def _load_lang(self):
    method build_answer (line 504) | def build_answer(self):

FILE: data/datasets/structure3d.py
  class S3DPretrainObj (line 8) | class S3DPretrainObj(ScanBase):
    method __init__ (line 9) | def __init__(self, cfg, split):
    method __len__ (line 28) | def __len__(self):
    method __getitem__ (line 31) | def __getitem__(self, index):
  class S3DSpatialRefer (line 45) | class S3DSpatialRefer(ScanBase):
    method __init__ (line 46) | def __init__(self, cfg, split):
    method __len__ (line 77) | def __len__(self):
    method __getitem__ (line 80) | def __getitem__(self, index):

FILE: evaluator/build.py
  class BaseEvaluator (line 11) | class BaseEvaluator():
    method __init__ (line 12) | def __init__(self, cfg, accelerator):
    method reset (line 19) | def reset(self):
    method batch_metrics (line 23) | def batch_metrics(self, data_dict, include_count=False):
    method update (line 26) | def update(self, data_dict):
    method record (line 33) | def record(self):
  function get_eval (line 57) | def get_eval(name, cfg, accelerator, **kwargs):
  function build_eval (line 65) | def build_eval(cfg, accelerator, **kwargs):

FILE: evaluator/objcls_eval.py
  class PretrainObjEval (line 8) | class PretrainObjEval(BaseEvaluator):
    method __init__ (line 9) | def __init__(self, cfg, accelerator, **kwargs):
    method batch_metrics (line 14) | def batch_metrics(self, data_dict, include_count=False):

FILE: evaluator/pretrain_eval.py
  class PretrainEval (line 8) | class PretrainEval(BaseEvaluator):
    method __init__ (line 9) | def __init__(self, cfg, accelerator, **kwargs):
    method batch_metrics (line 21) | def batch_metrics(self, data_dict):
    method update (line 69) | def update(self, data_dict):
    method record (line 77) | def record(self):
    method reset (line 88) | def reset(self):

FILE: evaluator/referit3d_eval.py
  class ReferIt3DEval (line 8) | class ReferIt3DEval(BaseEvaluator):
    method __init__ (line 9) | def __init__(self, cfg, accelerator, **kwargs):
    method batch_metrics (line 14) | def batch_metrics(self, data_dict, include_count=False):

FILE: evaluator/scanqa_eval.py
  class ScanQAEval (line 15) | class ScanQAEval(BaseEvaluator):
    method __init__ (line 16) | def __init__(self, cfg, accelerator, **kwargs):
    method batch_metrics (line 29) | def batch_metrics(self, data_dict, include_count=False):
  class ScanQAGenEval (line 85) | class ScanQAGenEval(ScanQAEval):
    method __init__ (line 86) | def __init__(self, cfg, accelerator, **kwargs):
    method batch_metrics (line 89) | def batch_metrics(self, data_dict, include_count=False):

FILE: evaluator/scanrefer_eval.py
  class ScanReferEval (line 8) | class ScanReferEval(BaseEvaluator):
    method __init__ (line 9) | def __init__(self, cfg, accelerator, **kwargs):
    method batch_metrics (line 14) | def batch_metrics(self, data_dict, include_count=False):

FILE: evaluator/sqa3d_eval.py
  class SQA3DEval (line 14) | class SQA3DEval():
    method __init__ (line 16) | def __init__(self, cfg, task_name):
    method update (line 49) | def update(self, data_dict):
    method batch_metrics (line 78) | def batch_metrics(self, data_dict):
    method reset (line 118) | def reset(self):
    method record (line 129) | def record(self, split='val'):

FILE: launch.py
  function parse_args (line 6) | def parse_args():
  function main (line 62) | def main():

FILE: model/build.py
  class BaseModel (line 8) | class BaseModel(nn.Module):
    method __init__ (line 9) | def __init__(self, cfg):
    method get_opt_params (line 12) | def get_opt_params(self):
  function build_model (line 16) | def build_model(cfg):

FILE: model/objcls.py
  class ObjCls (line 17) | class ObjCls(BaseModel):
    method __init__ (line 18) | def __init__(self, cfg):
    method forward (line 64) | def forward(self, data_dict):
    method get_opt_params (line 90) | def get_opt_params(self):

FILE: model/openvocab.py
  class OpenVocab (line 12) | class OpenVocab(BaseModel):
    method __init__ (line 13) | def __init__(self, cfg):
    method forward (line 26) | def forward(self, data_dict):
    method get_opt_params (line 103) | def get_opt_params(self):
  class OpenVocabPerScene (line 130) | class OpenVocabPerScene(BaseModel):
    method __init__ (line 131) | def __init__(self, cfg):
    method forward (line 141) | def forward(self, data_dict):
    method get_opt_params (line 232) | def get_opt_params(self):

FILE: modules/build.py
  function build_module (line 12) | def build_module(module_type, cfg):
  function build_module_by_name (line 24) | def build_module_by_name(cfg):

FILE: modules/grounding/unified_encoder.py
  class EntitySpatialCrossEncoder (line 13) | class EntitySpatialCrossEncoder(nn.Module):
    method __init__ (line 19) | def __init__(self, cfg, hidden_size=768, num_attention_heads=12, spati...
    method forward (line 37) | def forward(
  class UnifiedSpatialCrossEncoderV1 (line 61) | class UnifiedSpatialCrossEncoderV1(nn.Module):
    method __init__ (line 67) | def __init__(self, cfg, hidden_size=768, num_attention_heads=12, spati...
    method forward (line 90) | def forward(
  class UnifiedSpatialCrossEncoderV2 (line 122) | class UnifiedSpatialCrossEncoderV2(nn.Module):
    method __init__ (line 128) | def __init__(self, cfg, hidden_size=768, dim_feedforward=2048, num_att...
    method forward (line 147) | def forward(

FILE: modules/heads/grounding_head.py
  class GroundHeadV1 (line 8) | class GroundHeadV1(nn.Module):
    method __init__ (line 9) | def __init__(self, cfg, input_size=768, hidden_size=768, sem_cls_size=...
    method forward (line 29) | def forward(self, txt_embeds, obj_embeds, obj_pre_embeds, obj_masks, *...
  class GroundHead (line 43) | class GroundHead(nn.Module):
    method __init__ (line 44) | def __init__(self, cfg, input_size=768, hidden_size=768, dropout=0.3):
    method forward (line 51) | def forward(self, obj_embeds, obj_masks=None, **kwargs):

FILE: modules/heads/pretrain_head.py
  class BertPredictionHeadTransform (line 8) | class BertPredictionHeadTransform(nn.Module):
    method __init__ (line 9) | def __init__(self, hidden_size, hidden_act='gelu'):
    method forward (line 15) | def forward(self, hidden_states):
  class BertLMPredictionHead (line 22) | class BertLMPredictionHead(nn.Module):
    method __init__ (line 23) | def __init__(self, hidden_size, vocab_size):
    method forward (line 29) | def forward(self, hidden_states):
  class PretrainHeadV1 (line 36) | class PretrainHeadV1(nn.Module):
    method __init__ (line 37) | def __init__(self, cfg, hidden_size=768, vocab_size=30522):
    method forward (line 41) | def forward(self, txt_embeds, **kwargs):
  class OVPretrainHead (line 47) | class OVPretrainHead(nn.Module):
    method __init__ (line 48) | def __init__(self, cfg, hidden_size=768, vocab_size=30522, obj_vocab_s...
    method forward (line 53) | def forward(self, txt_embeds, obj_embeds, **kwargs):

FILE: modules/heads/qa_head.py
  class FC (line 8) | class FC(nn.Module):
    method __init__ (line 9) | def __init__(self, in_size, out_size, pdrop=0., use_gelu=True):
    method forward (line 20) | def forward(self, x):
  class MLP (line 30) | class MLP(nn.Module):
    method __init__ (line 31) | def __init__(self, in_size, mid_size, out_size, pdrop=0., use_gelu=True):
    method forward (line 36) | def forward(self, x):
  class AttFlat (line 40) | class AttFlat(nn.Module):
    method __init__ (line 41) | def __init__(self, hidden_size, flat_mlp_size=512, flat_glimpses=1, fl...
    method forward (line 56) | def forward(self, x, x_mask):
  class QAHeadV1 (line 73) | class QAHeadV1(nn.Module):
    method __init__ (line 74) | def __init__(self, cfg, hidden_size=768, mlp_size=256, glimpse=1, flat...
    method forward (line 86) | def forward(self, obj_embeds, obj_masks, txt_embeds, txt_masks, **kwar...

FILE: modules/language/bert.py
  class BERTLanguageEncoder (line 8) | class BERTLanguageEncoder(nn.Module):
    method __init__ (line 9) | def __init__(self, cfg, weights="bert-base-uncased", hidden_size=768,
    method forward (line 25) | def forward(self, txt_ids, txt_masks, **kwargs):

FILE: modules/language/clip.py
  class CLIPLanguageEncoder (line 10) | class CLIPLanguageEncoder(nn.Module):
    method __init__ (line 11) | def __init__(self, cfg, weights="openai/clip-vit-large-patch14", outpu...
    method forward (line 20) | def forward(self, txt_ids, txt_masks):

FILE: modules/layers/pointnet.py
  function break_up_pc (line 6) | def break_up_pc(pc):
  class PointNetPP (line 22) | class PointNetPP(nn.Module):
    method __init__ (line 28) | def __init__(self, sa_n_points: list,
    method forward (line 55) | def forward(self, features):

FILE: modules/layers/transformers.py
  class CrossAttentionLayer (line 12) | class CrossAttentionLayer(nn.Module):
    method __init__ (line 14) | def __init__(self, d_model, nhead, dim_feedforward=2048, dropout=0.1, ...
    method forward (line 39) | def forward(
  class TransformerDecoderLayer (line 66) | class TransformerDecoderLayer(nn.Module):
    method __init__ (line 67) | def __init__(self, d_model, nhead, dim_feedforward=2048, dropout=0.1, ...
    method forward (line 89) | def forward(
  class TransformerEncoderLayer (line 115) | class TransformerEncoderLayer(nn.Module):
    method __init__ (line 116) | def __init__(self, d_model, nhead, dim_feedforward=2048, batch_first=T...
    method forward (line 134) | def forward(
  class MultiHeadAttentionSpatial (line 157) | class MultiHeadAttentionSpatial(nn.Module):
    method __init__ (line 158) | def __init__(
    method forward (line 188) | def forward(self, q, k, v, pairwise_locs, key_padding_mask=None, txt_e...
  class TransformerSpatialDecoderLayer (line 242) | class TransformerSpatialDecoderLayer(TransformerDecoderLayer):
    method __init__ (line 243) | def __init__(
    method forward (line 258) | def forward(
  class TransformerSpatialEncoderLayer (line 285) | class TransformerSpatialEncoderLayer(TransformerEncoderLayer):
    method __init__ (line 286) | def __init__(
    method forward (line 301) | def forward(

FILE: modules/third_party/pointnet2/_ext_src/include/cuda_utils.h
  function opt_n_threads (line 15) | inline int opt_n_threads(int work_size) {
  function dim3 (line 21) | inline dim3 opt_block_config(int x, int y) {

FILE: modules/third_party/pointnet2/_ext_src/src/ball_query.cpp
  function ball_query (line 8) | at::Tensor ball_query(at::Tensor new_xyz, at::Tensor xyz, const float ra...

FILE: modules/third_party/pointnet2/_ext_src/src/bindings.cpp
  function PYBIND11_MODULE (line 6) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {

FILE: modules/third_party/pointnet2/_ext_src/src/group_points.cpp
  function group_points (line 12) | at::Tensor group_points(at::Tensor points, at::Tensor idx) {
  function group_points_grad (line 38) | at::Tensor group_points_grad(at::Tensor grad_out, at::Tensor idx, const ...

FILE: modules/third_party/pointnet2/_ext_src/src/interpolate.cpp
  function three_nn (line 14) | std::vector<at::Tensor> three_nn(at::Tensor unknowns, at::Tensor knows) {
  function three_interpolate (line 42) | at::Tensor three_interpolate(at::Tensor points, at::Tensor idx,
  function three_interpolate_grad (line 71) | at::Tensor three_interpolate_grad(at::Tensor grad_out, at::Tensor idx,

FILE: modules/third_party/pointnet2/_ext_src/src/sampling.cpp
  function gather_points (line 15) | at::Tensor gather_points(at::Tensor points, at::Tensor idx) {
  function gather_points_grad (line 40) | at::Tensor gather_points_grad(at::Tensor grad_out, at::Tensor idx,
  function furthest_point_sampling (line 66) | at::Tensor furthest_point_sampling(at::Tensor points, const int nsamples) {

FILE: modules/third_party/pointnet2/pointnet2_modules.py
  class _PointnetSAModuleBase (line 26) | class _PointnetSAModuleBase(nn.Module):
    method __init__ (line 28) | def __init__(self):
    method forward (line 34) | def forward(self, xyz: torch.Tensor,
  class PointnetSAModuleMSG (line 78) | class PointnetSAModuleMSG(_PointnetSAModuleBase):
    method __init__ (line 95) | def __init__(
  class PointnetSAModule (line 127) | class PointnetSAModule(PointnetSAModuleMSG):
    method __init__ (line 144) | def __init__(
  class PointnetSAModuleVotes (line 164) | class PointnetSAModuleVotes(nn.Module):
    method __init__ (line 168) | def __init__(
    method forward (line 210) | def forward(self, xyz: torch.Tensor,
  class PointnetSAModuleMSGVotes (line 274) | class PointnetSAModuleMSGVotes(nn.Module):
    method __init__ (line 278) | def __init__(
    method forward (line 309) | def forward(self, xyz: torch.Tensor,
  class PointnetFPModule (line 356) | class PointnetFPModule(nn.Module):
    method __init__ (line 367) | def __init__(self, *, mlp: List[int], bn: bool = True):
    method forward (line 371) | def forward(
  class PointnetLFPModuleMSG (line 418) | class PointnetLFPModuleMSG(nn.Module):
    method __init__ (line 422) | def __init__(
    method forward (line 454) | def forward(self, xyz2: torch.Tensor, xyz1: torch.Tensor,

FILE: modules/third_party/pointnet2/pointnet2_test.py
  function test_interpolation_grad (line 18) | def test_interpolation_grad():

FILE: modules/third_party/pointnet2/pointnet2_utils.py
  class RandomDropout (line 37) | class RandomDropout(nn.Module):
    method __init__ (line 38) | def __init__(self, p=0.5, inplace=False):
    method forward (line 43) | def forward(self, X):
  class FurthestPointSampling (line 48) | class FurthestPointSampling(Function):
    method forward (line 50) | def forward(ctx, xyz, npoint):
    method backward (line 73) | def backward(xyz, a=None):
  class GatherOperation (line 80) | class GatherOperation(Function):
    method forward (line 82) | def forward(ctx, features, idx):
    method backward (line 107) | def backward(ctx, grad_out):
  class ThreeNN (line 117) | class ThreeNN(Function):
    method forward (line 119) | def forward(ctx, unknown, known):
    method backward (line 142) | def backward(ctx, a=None, b=None):
  class ThreeInterpolate (line 149) | class ThreeInterpolate(Function):
    method forward (line 151) | def forward(ctx, features, idx, weight):
    method backward (line 177) | def backward(ctx, grad_out):
  class GroupingOperation (line 206) | class GroupingOperation(Function):
    method forward (line 208) | def forward(ctx, features, idx):
    method backward (line 232) | def backward(ctx, grad_out):
  class BallQuery (line 257) | class BallQuery(Function):
    method forward (line 259) | def forward(ctx, radius, nsample, xyz, new_xyz):
    method backward (line 284) | def backward(ctx, a=None):
  class QueryAndGroup (line 291) | class QueryAndGroup(nn.Module):
    method __init__ (line 303) | def __init__(self, radius, nsample, use_xyz=True, ret_grouped_xyz=Fals...
    method forward (line 314) | def forward(self, xyz, new_xyz, features=None):
  class GroupAll (line 376) | class GroupAll(nn.Module):
    method __init__ (line 384) | def __init__(self, use_xyz=True, ret_grouped_xyz=False):
    method forward (line 389) | def forward(self, xyz, new_xyz, features=None):

FILE: modules/third_party/pointnet2/pytorch_utils.py
  class SharedMLP (line 11) | class SharedMLP(nn.Sequential):
    method __init__ (line 13) | def __init__(
  class _BNBase (line 39) | class _BNBase(nn.Sequential):
    method __init__ (line 41) | def __init__(self, in_size, batch_norm=None, name=""):
  class BatchNorm1d (line 49) | class BatchNorm1d(_BNBase):
    method __init__ (line 51) | def __init__(self, in_size: int, *, name: str = ""):
  class BatchNorm2d (line 55) | class BatchNorm2d(_BNBase):
    method __init__ (line 57) | def __init__(self, in_size: int, name: str = ""):
  class BatchNorm3d (line 61) | class BatchNorm3d(_BNBase):
    method __init__ (line 63) | def __init__(self, in_size: int, name: str = ""):
  class _ConvBase (line 67) | class _ConvBase(nn.Sequential):
    method __init__ (line 69) | def __init__(
  class Conv1d (line 123) | class Conv1d(_ConvBase):
    method __init__ (line 125) | def __init__(
  class Conv2d (line 157) | class Conv2d(_ConvBase):
    method __init__ (line 159) | def __init__(
  class Conv3d (line 191) | class Conv3d(_ConvBase):
    method __init__ (line 193) | def __init__(
  class FC (line 225) | class FC(nn.Sequential):
    method __init__ (line 227) | def __init__(
  function set_bn_momentum_default (line 262) | def set_bn_momentum_default(bn_momentum):
  class BNMomentumScheduler (line 271) | class BNMomentumScheduler(object):
    method __init__ (line 273) | def __init__(
    method step (line 291) | def step(self, epoch=None):

FILE: modules/utils.py
  function get_activation_fn (line 12) | def get_activation_fn(activation_type):
  function get_mlp_head (line 18) | def get_mlp_head(input_size, hidden_size, output_size, dropout=0):
  function layer_repeat (line 28) | def layer_repeat(module, N, share_layer=False):
  function calc_pairwise_locs (line 38) | def calc_pairwise_locs(obj_centers, obj_whls, eps=1e-10, pairwise_rel_ty...
  function calc_pairwise_locs_mv (line 89) | def calc_pairwise_locs_mv(obj_centers, pairwise_rel_type='center', spati...
  function get_mixup_function (line 117) | def get_mixup_function(mixup_strategy, mixup_stage1, mixup_stage2):
  class AllMixup (line 128) | class AllMixup(nn.Module):
    method __init__ (line 129) | def __init__(self) -> None:
    method forward (line 132) | def forward(self, obj_sem_cls_pred, obj_labels, cur_step, total_steps):
  class LinearDecayMixup (line 141) | class LinearDecayMixup(nn.Module):
    method __init__ (line 142) | def __init__(self, mixup_stage1, mixup_stage2) -> None:
    method forward (line 148) | def forward(self, obj_sem_cls_pred, obj_labels, cur_step, total_steps):

FILE: modules/vision/obj_cls_encoder.py
  class ObjClsEncoder (line 6) | class ObjClsEncoder(nn.Module):
    method __init__ (line 7) | def __init__(self, cfg, input_feat_size=768, hidden_size=768, tgt_cls_...
    method forward (line 12) | def forward(self, obj_feats, **kwargs):

FILE: modules/vision/pcd_openvocab_encoder.py
  class PointOpenVocabEncoder (line 17) | class PointOpenVocabEncoder(nn.Module):
    method __init__ (line 18) | def __init__(self, cfg, backbone='pointnet++', hidden_size=768, path=N...
    method freeze_bn (line 121) | def freeze_bn(self, m):
    method forward (line 126) | def forward(self, obj_pcds, obj_locs, obj_masks, obj_sem_masks,

FILE: modules/weights.py
  function _init_weights_bert (line 3) | def _init_weights_bert(module, std=0.02):

FILE: optim/build.py
  function build_optim (line 6) | def build_optim(cfg, params, total_steps):

FILE: optim/loss/contra_loss.py
  class TextObjWithinBatch (line 12) | class TextObjWithinBatch(nn.Module):
    method __init__ (line 13) | def __init__(self, cfg):
    method forward (line 18) | def forward(self, data_dict):
  class TextObjBetweenBatch (line 43) | class TextObjBetweenBatch(nn.Module):
    method __init__ (line 44) | def __init__(self, cfg):
    method forward (line 49) | def forward(self, data_dict):
  class TextSceneBetweenBatch (line 75) | class TextSceneBetweenBatch(nn.Module):
    method __init__ (line 76) | def __init__(self, cfg):
    method forward (line 81) | def forward(self, data_dict):

FILE: optim/loss/loss.py
  function og3d_loss (line 8) | def og3d_loss(data_dict):
  function og3d_multi_loss (line 12) | def og3d_multi_loss(data_dict):
  function txt_cls_multi_loss (line 19) | def txt_cls_multi_loss(data_dict):
  function obj_cls_raw_loss (line 26) | def obj_cls_raw_loss(data_dict):
  function obj_cls_pre_loss (line 34) | def obj_cls_pre_loss(data_dict):
  function obj_cls_post_loss (line 42) | def obj_cls_post_loss(data_dict):
  function answer_loss (line 50) | def answer_loss(data_dict):
  function lm_cls_loss (line 56) | def lm_cls_loss(data_dict):
  function obj_cls_pre_loss_mask (line 64) | def obj_cls_pre_loss_mask(data_dict):
  function obj_cls_pre_loss_unmask (line 72) | def obj_cls_pre_loss_unmask(data_dict):
  function obj_cls_post_loss_mask (line 80) | def obj_cls_post_loss_mask(data_dict):
  function obj_cls_post_loss_unmask (line 88) | def obj_cls_post_loss_unmask(data_dict):
  function obj_cls_loss (line 96) | def obj_cls_loss(data_dict, smoothing=0.3):
  function mse_loss (line 105) | def mse_loss(data_dict):
  class Loss (line 111) | class Loss(nn.Module):
    method __init__ (line 112) | def __init__(self, cfg):
    method forward (line 131) | def forward(self, data_dict):

FILE: optim/optimizer/optim.py
  function get_optimizer (line 9) | def get_optimizer(cfg, params):

FILE: optim/scheduler.py
  function warmup_cosine (line 5) | def warmup_cosine(step, warmup_step, total_step, minimum_ratio=1e-5):
  function warmup_exp (line 14) | def warmup_exp(step, warmup_step, total_step, **kwargs):
  function get_scheduler (line 20) | def get_scheduler(cfg, optimizer, total_steps):

FILE: optim/utils.py
  function no_decay_param_group (line 1) | def no_decay_param_group(parameters, lr):

FILE: preprocess/arkitscenes.py
  class ARKitScenesProcessor (line 18) | class ARKitScenesProcessor(ProcessorBase):
    method record_splits (line 19) | def record_splits(self, scan_ids):
    method read_all_scans (line 33) | def read_all_scans(self):
    method process_point_cloud (line 40) | def process_point_cloud(self, scan_id, plydata, annotations):
    method scene_proc (line 87) | def scene_proc(self, scan_id):
    method process_scans (line 104) | def process_scans(self):

FILE: preprocess/build.py
  class ProcessorBase (line 7) | class ProcessorBase:
    method __init__ (line 8) | def __init__(self, cfg):
    method setup_directories (line 22) | def setup_directories(self):
    method log_starting_info (line 27) | def log_starting_info(self, scan_len, e=''):
    method check_key (line 38) | def check_key(key):

FILE: preprocess/multiscan.py
  class MultiScanProcessor (line 18) | class MultiScanProcessor(ProcessorBase):
    method record_splits (line 19) | def record_splits(self, scan_ids, ratio=0.8):
    method read_all_scans (line 37) | def read_all_scans(self):
    method process_point_cloud (line 49) | def process_point_cloud(self, scan_id, plydata, annotations):
    method annotations_to_dataframe_obj (line 96) | def annotations_to_dataframe_obj(annotations):
    method scene_proc (line 110) | def scene_proc(self, scan_id):
    method process_scans (line 119) | def process_scans(self):

FILE: preprocess/rscan.py
  class RScanProcessor (line 18) | class RScanProcessor(ProcessorBase):
    method record_splits (line 19) | def record_splits(self, scan_ids, ratio=0.8):
    method read_all_scans (line 37) | def read_all_scans(self):
    method process_point_cloud (line 42) | def process_point_cloud(self, scan_id, plydata, annotations):
    method scene_proc (line 113) | def scene_proc(self, scan_id):
    method process_scans (line 127) | def process_scans(self):

FILE: preprocess/sceneverse2hmsemantic.py
  function load_semantic_anno (line 10) | def load_semantic_anno(semantic_txt):
  function scene_proc (line 34) | def scene_proc(scene_input):

FILE: preprocess/ssg/relationships/camera.py
  function getLinearEquation (line 5) | def getLinearEquation(p1x, p1y, p2x, p2y):
  function cal_glocal_position (line 16) | def cal_glocal_position(object, floor, distance_rate=1.6):
  function cal_camera_relations (line 34) | def cal_camera_relations(ObjNode_dict, camera_position, camera_view, ins...

FILE: preprocess/ssg/relationships/hanging.py
  function cal_above_below_relationships (line 4) | def cal_above_below_relationships(ObjNode_dict, src, scene_high):
  function filter_labels (line 31) | def filter_labels(obj_label):
  function cal_hanging_relationships (line 40) | def cal_hanging_relationships (ObjNode_dict, no_supported_objs, camera_a...

FILE: preprocess/ssg/relationships/multi_objs.py
  function are_furniture_aligned (line 8) | def are_furniture_aligned(furniture1, furniture2, offset_threshold):
  function find_aligned_furniture (line 41) | def find_aligned_furniture(furniture_list, ObjNode_dict, offset_threshold):
  function furniture_merge_lists (line 59) | def furniture_merge_lists(lists):
  function merge_sublists (line 74) | def merge_sublists(L):
  function find_middle_furniture (line 89) | def find_middle_furniture (proximity_relations, ObjNode_dict):

FILE: preprocess/ssg/relationships/proximity.py
  function get_direction (line 5) | def get_direction(src_obj, tgt_obj):
  function get_oppo_direction (line 35) | def get_oppo_direction(direction):
  function get_space_relations (line 46) | def get_space_relations(src, tgt):
  function get_distance (line 55) | def get_distance(src, tgt):
  function cal_proximity_relationships (line 63) | def cal_proximity_relationships(neighbor_objs_id, camera_angle, ObjNode_...

FILE: preprocess/ssg/relationships/support.py
  function is_supported (line 4) | def is_supported(target_obj, obj, camera_angle, radius_range = 0.1, thre...
  function optimaze_support_loops (line 50) | def optimaze_support_loops(support_relations_dict):
  function cal_support_relations (line 63) | def cal_support_relations(ObjNode_list, camera_angle):

FILE: preprocess/ssg/ssg_data/script/ObjNode.py
  class ObjNode (line 8) | class ObjNode(object):
    method __init__ (line 9) | def __init__(self, id=None, label=None, mesh=None, position=None, size...
    method __str__ (line 20) | def __str__(self):
    method get_object_information (line 23) | def get_object_information(self, dataset):
    method display_obb_box (line 39) | def display_obb_box(self, scene_visible = True):

FILE: preprocess/ssg/ssg_data/ssg_visualize.py
  function vis_dataset (line 6) | def vis_dataset(ObjNode_dict, relation, scene_path, scan_id, scene_center):

FILE: preprocess/ssg/ssg_main.py
  function default_dump (line 21) | def default_dump(obj):
  function convert_pc_to_box (line 30) | def convert_pc_to_box(obj_pc):
  function init_camera_view (line 41) | def init_camera_view():
  function filter_bad_label (line 53) | def filter_bad_label(input_label):
  function get_obj_room_id (line 61) | def get_obj_room_id (org_id):
  function generate_object_info (line 68) | def generate_object_info(save_root, scene_name) :
  function generate_ssg_data (line 112) | def generate_ssg_data(dataset, scene_path, pre_load_path):
  function main (line 132) | def main(cfg):

FILE: preprocess/ssg/ssg_utils.py
  function cw_rotate (line 12) | def cw_rotate(point, ang):
  function euclideanDistance (line 19) | def euclideanDistance(instance1, instance2, dimension):
  function if_inPoly (line 26) | def if_inPoly(polygon, Points):
  function get_Poly_Area (line 32) | def get_Poly_Area(polygon):
  function get_theta (line 38) | def get_theta (x, y):
  function generate_relation (line 55) | def generate_relation(src, tgt, express):
  function visualize_relations (line 82) | def visualize_relations(target_obj, obj, relationship, camera_angle, cam...
  function visualize_relations_multi_objs (line 132) | def visualize_relations_multi_objs(objs, relationship, item, camera_angl...
  function render_bbox_pyvista (line 196) | def render_bbox_pyvista(tgt, src, relationship, camera_angle, camera_pos...
  function visualize_camera_relations (line 276) | def visualize_camera_relations(ObjNode_dict, camera_relations, camera_po...
  function read_one_obj (line 323) | def read_one_obj(bbox_points, scene_file):

FILE: preprocess/structured3d.py
  class S3DProcessor (line 18) | class S3DProcessor(ProcessorBase):
    method record_splits (line 19) | def record_splits(self, scan_ids):
    method read_all_scans (line 33) | def read_all_scans(self):
    method process_point_cloud (line 40) | def process_point_cloud(self, scan_id, plydata, annotations):
    method scene_proc (line 76) | def scene_proc(self, scan_id):
    method process_scans (line 92) | def process_scans(self):

FILE: preprocess/utils/align_utils.py
  function compute_box_3d (line 4) | def compute_box_3d(size, center, rotmat):
  function rotate_z_axis_by_degrees (line 28) | def rotate_z_axis_by_degrees(pointcloud, theta, clockwise=True):
  function eulerAnglesToRotationMatrix (line 40) | def eulerAnglesToRotationMatrix(theta):
  function is_axis_aligned (line 79) | def is_axis_aligned(rotated_box, thres=0.05):
  function calc_align_matrix (line 85) | def calc_align_matrix(bbox_list):

FILE: preprocess/utils/constant.py
  class PromptType (line 641) | class PromptType(Enum):

FILE: run.py
  function main (line 13) | def main(cfg):

FILE: trainer/build.py
  class Tracker (line 29) | class Tracker():
    method __init__ (line 30) | def __init__(self, cfg):
    method step (line 33) | def step(self):
    method reset (line 36) | def reset(self, cfg):
    method state_dict (line 41) | def state_dict(self):
    method load_state_dict (line 44) | def load_state_dict(self, state_dict):
  class BaseTrainer (line 48) | class BaseTrainer():
    method __init__ (line 49) | def __init__(self, cfg):
    method forward (line 135) | def forward(self, data_dict):
    method backward (line 138) | def backward(self, loss):
    method log (line 147) | def log(self, results, mode="train"):
    method save (line 160) | def save(self, name):
    method resume (line 164) | def resume(self):
    method load_pretrain (line 174) | def load_pretrain(self):
    method save_func (line 186) | def save_func(self, path):
  function build_trainer (line 190) | def build_trainer(cfg):

FILE: trainer/debug_trainer.py
  class DebugTrainer (line 10) | class DebugTrainer(BaseTrainer):
    method __init__ (line 11) | def __init__(self, cfg):
    method forward (line 15) | def forward(self, data_dict):
    method backward (line 18) | def backward(self, loss):
    method train_step (line 26) | def train_step(self, epoch):
    method eval_step (line 39) | def eval_step(self, epoch):
    method test_step (line 48) | def test_step(self):
    method run (line 56) | def run(self):

FILE: trainer/default_trainer.py
  class DefaultTrainer (line 10) | class DefaultTrainer(BaseTrainer):
    method __init__ (line 11) | def __init__(self, cfg):
    method forward (line 15) | def forward(self, data_dict):
    method backward (line 18) | def backward(self, loss):
    method train_step (line 26) | def train_step(self, epoch):
    method eval_step (line 51) | def eval_step(self, epoch):
    method test_step (line 67) | def test_step(self):
    method run (line 80) | def run(self):

FILE: trainer/objpretrain_trainer.py
  class ObjPretrainTrainer (line 10) | class ObjPretrainTrainer(BaseTrainer):
    method __init__ (line 11) | def __init__(self, cfg):
    method forward (line 15) | def forward(self, data_dict):
    method backward (line 18) | def backward(self, loss):
    method train_step (line 26) | def train_step(self, epoch):
    method eval_step (line 49) | def eval_step(self, epoch):
    method test_step (line 70) | def test_step(self):
    method run (line 87) | def run(self):

FILE: trainer/openvocab_trainer.py
  class OpenVocabTrainer (line 10) | class OpenVocabTrainer(BaseTrainer):
    method __init__ (line 11) | def __init__(self, cfg):
    method forward (line 15) | def forward(self, data_dict):
    method backward (line 18) | def backward(self, loss):
    method train_step (line 26) | def train_step(self, epoch):
    method eval_step (line 49) | def eval_step(self, epoch):
    method test_step (line 65) | def test_step(self):
    method run (line 82) | def run(self):

FILE: visualize_data.py
  function convert_pc_to_box (line 11) | def convert_pc_to_box(obj_pc):
  function load_scan (line 23) | def load_scan(pcd_path, inst2label_path, scene_name):
  function visualize_one_scene (line 31) | def visualize_one_scene(obj_pcds, points, colors, caption):
  function visualize_data (line 47) | def visualize_data(save_root, scene_name, vis_obj=True):
  function visualize_refer (line 70) | def visualize_refer(save_root, anno_file):