MCPcopy Create free account

hub / github.com/CVMI-Lab/DODA / functions

Functions369 in github.com/CVMI-Lab/DODA

⨍ Functions 369◇ Types & classes 63
↓ 1 callersFunctionget_coords_color
(opt)
util/visualize_scannet.py:28
↓ 1 callersFunctionget_coords_color
(opt)
util/visualize_3dfront.py:29
↓ 1 callersFunctionget_coords_color
(opt)
util/visualize_s3dis.py:41
↓ 1 callersMethodget_data_list
(self)
dataset/dataset.py:49
↓ 1 callersFunctionget_dataset
(cfg, args, dist_train, logger, pin_memory=False)
dataset/__init__.py:153
↓ 1 callersFunctionget_git_commit_id
()
util/common_utils.py:146
↓ 1 callersMethodget_item
(self, n)
dataset/augmentor/data_augmentor.py:34
↓ 1 callersFunctionget_label_confidence
(cfg, logger, train_loader, model, test_model_fn_st, epoch, rank, dist=False)
util/pseudo_labels_util.py:80
↓ 1 callersFunctionget_perclass_thres
(cfg, logger, train_loader, model, model_fn, epoch, rank, dist=False)
util/pseudo_labels_util.py:147
↓ 1 callersMethodget_split
(self, n)
dataset/augmentor/data_augmentor.py:79
↓ 1 callersFunctionget_tar_train_dataset
(cfg, args, dist_train, logger, pin_memory=False, src_train_data=None)
dataset/__init__.py:140
↓ 1 callersFunctionget_thres_per_class_on_thres_ratio
given per class thres ratio, get per class thres
util/pseudo_labels_util.py:128
↓ 1 callersFunctionget_view_range_mask
(_xyz_f, _camera_f, mode=4, **kwargs)
dataset/augmentor/augmentor_utils.py:211
↓ 1 callersMethodinit_class_ratio
(self, cfg)
dataset/augmentor/data_augmentor.py:51
↓ 1 callersMethodinit_queue
(self)
dataset/augmentor/data_augmentor.py:67
↓ 1 callersMethodload_data
(self, index)
dataset/front3d.py:40
↓ 1 callersMethodload_data
(self, index)
dataset/scannet.py:36
↓ 1 callersMethodload_data
(self, index)
dataset/s3dis.py:35
↓ 1 callersMethodload_mapper_file
(map_file)
dataset/dataset.py:53
↓ 1 callersMethodload_sampler
(self, path)
dataset/augmentor/data_augmentor.py:96
↓ 1 callersFunctionlovasz_softmax
Multi-class Lovasz-Softmax loss probas: [B, C, H, W] Variable, class probabilities at each prediction (between 0 and 1). Inte
util/lovasz_loss.py:129
↓ 1 callersFunctionmain
()
tool/train.py:271
↓ 1 callersFunctionmain
()
tool/test.py:225
↓ 1 callersFunctionmain
()
tool/st.py:408
↓ 1 callersFunctionmerge_new_config
(config, new_config)
util/config.py:56
↓ 1 callersFunctionnoise_simulation
(param, xyz)
dataset/augmentor/augmentor_utils.py:169
↓ 1 callersFunctionocclusion_simulation
(param, xyz, labels, class_names, ignore_label=255)
dataset/augmentor/augmentor_utils.py:116
↓ 1 callersFunctionparse_config
()
tool/train.py:29
↓ 1 callersFunctionparse_config
()
tool/test.py:33
↓ 1 callersFunctionparse_config
()
tool/st.py:33
↓ 1 callersFunctionpoly_learning_rate
poly learning rate policy
util/common_utils.py:161
↓ 1 callersMethodread_data
(self)
util/common_utils.py:31
↓ 1 callersMethodregister_slave
Register an slave device. Args: identifier: an identifier, usually is the device id. Returns: a `SlavePipe` obj
lib/sync_bn/comm.py:84
↓ 1 callersMethodreset
(self)
util/common_utils.py:46
↓ 1 callersMethodreset_parameters
(self)
model/dsnorm.py:54
↓ 1 callersMethodreset_running_stats
(self)
model/dsnorm.py:43
↓ 1 callersMethodrun_master
Main entry for the master device in each forward pass. The messages were first collected from each devices (including the master devi
lib/sync_bn/comm.py:102
↓ 1 callersMethodrun_slave
(self, msg)
lib/sync_bn/comm.py:49
↓ 1 callersFunctionsave_grad
(name)
lib/pointgroup_ops/functions/pointgroup_ops.py:388
↓ 1 callersMethodsave_sampler
(self, path)
dataset/augmentor/data_augmentor.py:104
↓ 1 callersFunctionsec_mean
lib/pointgroup_ops/src/sec_mean/sec_mean.cpp:9
↓ 1 callersMethodset_cur_epoch
(self, cur_epoch)
util/common_utils.py:28
↓ 1 callersFunctionset_pseudo_labels
( args, cfg, logger, pseudo_labels_dir, train_loader, model, model_fn, epoch, rank=0, dist=False )
util/pseudo_labels_util.py:157
↓ 1 callersMethodset_pseudo_labels_dir
(self, pseudo_labels_dir)
dataset/dataset.py:79
↓ 1 callersMethodshuffle
(self, cfg=None, data_dict=None)
dataset/augmentor/data_augmentor.py:232
↓ 1 callersFunctionsoftmax
(x)
util/common_utils.py:464
↓ 1 callersFunctionstep_learning_rate
Sets the learning rate to the base LR decayed by 10 every step epochs
util/common_utils.py:154
↓ 1 callersMethodtacm
(self, cfg=None, data_dict=None)
dataset/augmentor/data_augmentor.py:207
↓ 1 callersFunctiontail_cuboids_from_sampler
(param, n, split_status, split_sampler, **args)
dataset/augmentor/augmentor_utils.py:400
↓ 1 callersFunctiontest
(model, model_fn, test_loader, eval_output_dir, ckpt_dir, dist_test, dataset_cfg)
tool/test.py:204
↓ 1 callersMethodtest_collate_fn
(self, items)
dataset/dataset.py:189
↓ 1 callersFunctiontest_one_epoch
(test_loader, model, model_fn, epoch, dataset_cfg, dist_test=False, result_dir=None)
tool/test.py:103
↓ 1 callersFunctiontrain
( model, model_fn, model_fn_test, train_loader, val_loader, optimizer, scheduler, ckpt_dir, rank, dist
tool/train.py:235
↓ 1 callersFunctiontrain
(model, model_fn, test_model_fn, source_reader, train_loader, val_loader, optimizer, scheduler, ckpt
tool/st.py:344
↓ 1 callersFunctiontrain_epoch
(train_loader, model, model_fn, optimizer, scheduler, epoch, rank, dist_train)
tool/train.py:69
↓ 1 callersFunctiontrain_epoch
(source_reader, train_loader, model, model_fn, optimizer, scheduler, epoch, rank, dist_train)
tool/st.py:100
↓ 1 callersFunctionupdate_checkpoint
(checkpoint)
util/model_utils.py:20
↓ 1 callersMethodupdate_class_ratio
(self, class_ratio)
dataset/augmentor/data_augmentor.py:90
↓ 1 callersFunctionupdate_meter
(intersection_meter, union_meter, target_meter, loss_meter, preds, labels, loss, dist_test)
tool/test.py:81
↓ 1 callersMethodupdate_queue
(self, items)
dataset/augmentor/data_augmentor.py:24
↓ 1 callersFunctionupdate_split_sampler
(train_loader, tar_tail_splits, tar_splits_class_ratio, num_c, rank, dist_train=False)
tool/st.py:82
↓ 1 callersFunctionvalidate_epoch
(val_loader, model, model_fn, epoch, rank, dist_train)
tool/train.py:161
↓ 1 callersFunctionvisualize_o3d
(xyz, color)
util/visualize_scannet.py:20
↓ 1 callersFunctionvisualize_o3d
(xyz, color)
util/visualize_3dfront.py:21
↓ 1 callersFunctionvisualize_o3d
(xyz, color)
util/visualize_s3dis.py:33
↓ 1 callersFunctionxyz_idx_in_split
(xyz, split_max, range)
dataset/augmentor/augmentor_utils.py:444
MethodConnectedComponent
lib/pointgroup_ops/src/datatype/datatype.cpp:12
FunctionPYBIND11_MODULE
lib/pointops2/src/pointops_api.cpp:12
FunctionPYBIND11_MODULE
lib/pointgroup_ops/src/pointgroup_ops_api.cpp:6
MethodSparseGrid
lib/pointgroup_ops/src/datatype/datatype.cpp:3
Method__getitem__
(self, item)
dataset/front3d.py:59
Method__getitem__
(self, item)
dataset/scannet.py:60
Method__getitem__
(self, item)
dataset/dataset.py:115
Method__getitem__
(self, item)
dataset/mix_dataset.py:63
Method__getitem__
(self, item)
dataset/s3dis.py:54
Method__getstate__
(self)
lib/sync_bn/comm.py:78
Method__init__
(self, cfg, class_names, batch_size, split='training', training=True, logger=None)
dataset/front3d.py:20
Method__init__
(self, cfg, class_names, batch_size, split='training', training=True, logger=None)
dataset/scannet.py:13
Method__init__
(self, dataset, num_replicas=None, rank=None, shuffle=True)
dataset/__init__.py:23
Method__init__
(self, cfg, class_names, batch_size, split='training', training=True, logger=None)
dataset/dataset.py:20
Method__init__
(self, dataset1, dataset2)
dataset/mix_dataset.py:60
Method__init__
(self, cfg, class_names, batch_size, split='training', training=True, logger=None)
dataset/s3dis.py:13
Method__init__
(self, size)
dataset/augmentor/data_augmentor.py:16
Method__init__
(self, cfg)
dataset/augmentor/data_augmentor.py:44
Method__init__
(self, aug_cfg, dataset_name, class_names, ignore_label, voxel_scale, voxel_mode, full_scale,
dataset/augmentor/data_augmentor.py:112
Method__init__
(self)
lib/sync_bn/comm.py:21
Method__init__
(self, num_features, eps=1e-5, momentum=0.1, affine=True)
lib/sync_bn/batchnorm.py:39
Method__init__
(self, cfg)
model/unet.py:16
Method__init__
(self, in_channels, out_channels, norm_fn, indice_key=None)
model/unet_block.py:11
Method__init__
(self, in_channels, out_channels, norm_fn, indice_key=None)
model/unet_block.py:42
Method__init__
(self, num_features, eps=1e-5, momentum=0.1, affine=True, track_running_stats=True)
model/dsnorm.py:13
Method__init__
(self)
util/lovasz_loss.py:112
Method__init__
(self, ignore=None)
util/loss_utils.py:10
Method__init__
(self, dataloader, sampler)
util/common_utils.py:21
Method__init__
(self)
util/common_utils.py:43
Method__iter__
(self)
dataset/__init__.py:27
Method__setstate__
(self, state)
lib/sync_bn/comm.py:81
Method_check_input_dim
(self, input)
lib/sync_bn/batchnorm.py:247
Method_check_input_dim
(self, input)
lib/sync_bn/batchnorm.py:311
Method_check_input_dim
(self, input)
model/dsnorm.py:270
← previousnext →101–200 of 369, ranked by callers