SYMBOL INDEX (392 symbols across 48 files)

FILE: datasets/LabeledHybrid.py
  class LabeledHybrid (line 10) | class LabeledHybrid(data.Dataset):
    method __init__ (line 11) | def __init__(self, config):
    method pc_norm (line 40) | def pc_norm(self, pc):
    method random_sample (line 49) | def random_sample(self, pc, num):
    method __getitem__ (line 55) | def __getitem__(self, idx):
    method __len__ (line 66) | def __len__(self):

FILE: datasets/ModelNetDataset.py
  function pc_normalize (line 20) | def pc_normalize(pc):
  function farthest_point_sample (line 29) | def farthest_point_sample(point, npoint):
  class ModelNet (line 53) | class ModelNet(Dataset):
    method __init__ (line 54) | def __init__(self, config):
    method __len__ (line 118) | def __len__(self):
    method _get_item (line 121) | def _get_item(self, index):
    method __getitem__ (line 142) | def __getitem__(self, index):

FILE: datasets/ModelNetDatasetFewShot.py
  function pc_normalize (line 21) | def pc_normalize(pc):
  class ModelNetFewShot (line 29) | class ModelNetFewShot(Dataset):
    method __init__ (line 30) | def __init__(self, config):
    method __len__ (line 56) | def __len__(self):
    method __getitem__ (line 59) | def __getitem__(self, index):

FILE: datasets/ScanObjectNNDataset.py
  class ScanObjectNN (line 12) | class ScanObjectNN(Dataset):
    method __init__ (line 13) | def __init__(self, config, **kwargs):
    method __getitem__ (line 33) | def __getitem__(self, idx):
    method __len__ (line 46) | def __len__(self):
  class ScanObjectNN_hardest (line 52) | class ScanObjectNN_hardest(Dataset):
    method __init__ (line 53) | def __init__(self, config, **kwargs):
    method __getitem__ (line 73) | def __getitem__(self, idx):
    method __len__ (line 86) | def __len__(self):

FILE: datasets/ShapeNet55Dataset.py
  class ShapeNet (line 11) | class ShapeNet(data.Dataset):
    method __init__ (line 12) | def __init__(self, config):
    method pc_norm (line 52) | def pc_norm(self, pc):
    method random_sample (line 60) | def random_sample(self, pc, num):
    method __getitem__ (line 65) | def __getitem__(self, idx):
    method __len__ (line 76) | def __len__(self):

FILE: datasets/UnlabeledHybrid.py
  class UnlabeledHybrid (line 11) | class UnlabeledHybrid(data.Dataset):
    method __init__ (line 12) | def __init__(self, config):
    method pc_norm (line 52) | def pc_norm(self, pc):
    method random_sample (line 60) | def random_sample(self, pc, num):
    method __getitem__ (line 66) | def __getitem__(self, idx):
    method __len__ (line 78) | def __len__(self):

FILE: datasets/build.py
  function build_dataset_from_cfg (line 7) | def build_dataset_from_cfg(cfg, default_args = None):

FILE: datasets/data_transforms.py
  class PointcloudRotate (line 6) | class PointcloudRotate(object):
    method __call__ (line 7) | def __call__(self, pc):
  class PointcloudScaleAndTranslate (line 20) | class PointcloudScaleAndTranslate(object):
    method __init__ (line 21) | def __init__(self, scale_low=2. / 3., scale_high=3. / 2., translate_ra...
    method __call__ (line 26) | def __call__(self, pc):
  class PointcloudJitter (line 36) | class PointcloudJitter(object):
    method __init__ (line 37) | def __init__(self, std=0.01, clip=0.05):
    method __call__ (line 40) | def __call__(self, pc):
  class PointcloudScale (line 50) | class PointcloudScale(object):
    method __init__ (line 51) | def __init__(self, scale_low=2. / 3., scale_high=3. / 2.):
    method __call__ (line 55) | def __call__(self, pc):
  class PointcloudTranslate (line 64) | class PointcloudTranslate(object):
    method __init__ (line 65) | def __init__(self, translate_range=0.2):
    method __call__ (line 68) | def __call__(self, pc):
  class PointcloudRandomInputDropout (line 78) | class PointcloudRandomInputDropout(object):
    method __init__ (line 79) | def __init__(self, max_dropout_ratio=0.5):
    method __call__ (line 83) | def __call__(self, pc):
  class RandomHorizontalFlip (line 95) | class RandomHorizontalFlip(object):
    method __init__ (line 98) | def __init__(self, upright_axis = 'z', is_temporal=False):
    method __call__ (line 109) | def __call__(self, coords):

FILE: datasets/generate_few_shot_data.py
  function generate_fewshot_data (line 20) | def generate_fewshot_data(way, shot, prefix_ind, eval_sample=20):

FILE: datasets/io.py
  class IO (line 6) | class IO:
    method get (line 8) | def get(cls, file_path):
    method _read_npy (line 24) | def _read_npy(cls, file_path):
    method _read_txt (line 36) | def _read_txt(cls, file_path):
    method _read_h5 (line 40) | def _read_h5(cls, file_path):

FILE: extensions/chamfer_dist/__init__.py
  class ChamferFunction (line 13) | class ChamferFunction(torch.autograd.Function):
    method forward (line 15) | def forward(ctx, xyz1, xyz2):
    method backward (line 22) | def backward(ctx, grad_dist1, grad_dist2):
  class ChamferDistanceL2 (line 28) | class ChamferDistanceL2(torch.nn.Module):
    method __init__ (line 31) | def __init__(self, ignore_zeros=False):
    method forward (line 35) | def forward(self, xyz1, xyz2):
  class ChamferDistanceL2_split (line 46) | class ChamferDistanceL2_split(torch.nn.Module):
    method __init__ (line 49) | def __init__(self, ignore_zeros=False):
    method forward (line 53) | def forward(self, xyz1, xyz2):
  class ChamferDistanceL1 (line 64) | class ChamferDistanceL1(torch.nn.Module):
    method __init__ (line 67) | def __init__(self, ignore_zeros=False):
    method forward (line 71) | def forward(self, xyz1, xyz2):

FILE: extensions/chamfer_dist/chamfer_cuda.cpp
  function chamfer_forward (line 22) | std::vector<torch::Tensor> chamfer_forward(torch::Tensor xyz1,
  function chamfer_backward (line 27) | std::vector<torch::Tensor> chamfer_backward(torch::Tensor xyz1,
  function PYBIND11_MODULE (line 36) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {

FILE: extensions/chamfer_dist/test.py
  class ChamferDistanceTestCase (line 23) | class ChamferDistanceTestCase(unittest.TestCase):
    method test_chamfer_dist (line 24) | def test_chamfer_dist(self):

FILE: extensions/emd/cuda/emd.cpp
  function PYBIND11_MODULE (line 23) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {

FILE: extensions/emd/emd.py
  class EarthMoverDistanceFunction (line 5) | class EarthMoverDistanceFunction(torch.autograd.Function):
    method forward (line 7) | def forward(ctx, xyz1, xyz2):
    method backward (line 17) | def backward(ctx, grad_cost):
  class earth_mover_distance (line 26) | class earth_mover_distance(torch.nn.Module):
    method __init__ (line 29) | def __init__(self):
    method forward (line 32) | def forward(self, xyz1, xyz2, transpose=False):

FILE: main.py
  function main (line 14) | def main():

FILE: main_vis.py
  function main (line 11) | def main():

FILE: models/GPT.py
  class Block (line 8) | class Block(nn.Module):
    method __init__ (line 9) | def __init__(self, embed_dim, num_heads):
    method forward (line 20) | def forward(self, x, attn_mask):
  class GPT_extractor (line 31) | class GPT_extractor(nn.Module):
    method __init__ (line 32) | def __init__(
    method forward (line 70) | def forward(self, h, pos, attn_mask, classify=False):
  class GPT_generator (line 104) | class GPT_generator(nn.Module):
    method __init__ (line 105) | def __init__(
    method forward (line 127) | def forward(self, h, pos, attn_mask):

FILE: models/PointGPT.py
  class Encoder_large (line 18) | class Encoder_large(nn.Module):  # Embedding module
    method __init__ (line 19) | def __init__(self, encoder_channel):
    method forward (line 38) | def forward(self, point_groups):
  class Encoder_small (line 55) | class Encoder_small(nn.Module):  # Embedding module
    method __init__ (line 56) | def __init__(self, encoder_channel):
    method forward (line 72) | def forward(self, point_groups):
  class Group (line 90) | class Group(nn.Module):
    method __init__ (line 91) | def __init__(self, num_group, group_size):
    method simplied_morton_sorting (line 98) | def simplied_morton_sorting(self, xyz, center):
    method morton_sorting (line 130) | def morton_sorting(self, xyz, center):
    method forward (line 148) | def forward(self, xyz):
  class Mlp (line 188) | class Mlp(nn.Module):
    method __init__ (line 189) | def __init__(self, in_features, hidden_features=None, out_features=Non...
    method forward (line 198) | def forward(self, x):
  class Attention (line 207) | class Attention(nn.Module):
    method __init__ (line 208) | def __init__(self, dim, num_heads=8, qkv_bias=False, qk_scale=None, at...
    method forward (line 218) | def forward(self, x):
  class Block (line 235) | class Block(nn.Module):
    method __init__ (line 236) | def __init__(self, dim, num_heads, mlp_ratio=4., qkv_bias=False, qk_sc...
    method forward (line 251) | def forward(self, x):
  class PositionEmbeddingCoordsSine (line 257) | class PositionEmbeddingCoordsSine(nn.Module):
    method __init__ (line 268) | def __init__(self, n_dim: int = 1, d_model: int = 256, temperature=100...
    method forward (line 280) | def forward(self, xyz: torch.Tensor) -> torch.Tensor:
  class GPT_Transformer (line 307) | class GPT_Transformer(nn.Module):
    method __init__ (line 308) | def __init__(self, config, **kwargs):
    method _init_weights (line 360) | def _init_weights(self, m):
    method forward (line 373) | def forward(self, neighborhood, center, noaug=False, classify=False):
  class PointGPT (line 442) | class PointGPT(nn.Module):
    method __init__ (line 443) | def __init__(self, config):
    method build_loss_func (line 463) | def build_loss_func(self, loss_type):
    method forward (line 475) | def forward(self, pts, vis=False, **kwargs):
  class PointTransformer (line 503) | class PointTransformer(nn.Module):
    method __init__ (line 504) | def __init__(self, config, **kwargs):
    method build_loss_func (line 561) | def build_loss_func(self, loss_type='cdl12'):
    method get_loss_acc (line 574) | def get_loss_acc(self, ret, gt):
    method load_model_from_ckpt (line 580) | def load_model_from_ckpt(self, bert_ckpt_path):
    method _init_weights (line 618) | def _init_weights(self, m):
    method forward (line 631) | def forward(self, pts):

FILE: models/build.py
  function build_model_from_cfg (line 7) | def build_model_from_cfg(cfg, **kwargs):

FILE: models/z_order.py
  function round_to_int_32 (line 4) | def round_to_int_32(data):
  function split_by_3 (line 23) | def split_by_3(x):
  function get_z_order (line 50) | def get_z_order(x, y, z):
  function get_z_values (line 72) | def get_z_values(data):

FILE: segmentation/dataset.py
  class ModelNetDataLoader (line 9) | class ModelNetDataLoader(Dataset):
    method __init__ (line 10) | def __init__(self, root, npoint=1024, split='train', uniform=False, no...
    method __len__ (line 36) | def __len__(self):
    method _get_item (line 39) | def _get_item(self, index):
    method __getitem__ (line 62) | def __getitem__(self, index):
  class PartNormalDataset (line 66) | class PartNormalDataset(Dataset):
    method __init__ (line 67) | def __init__(self, root='/data/cgy/ShapenetPart/shapenetcore_partanno_...
    method __getitem__ (line 138) | def __getitem__(self, index):
    method __len__ (line 163) | def __len__(self):

FILE: segmentation/extensions/chamfer_dist/__init__.py
  class ChamferFunction (line 13) | class ChamferFunction(torch.autograd.Function):
    method forward (line 15) | def forward(ctx, xyz1, xyz2):
    method backward (line 22) | def backward(ctx, grad_dist1, grad_dist2):
  class ChamferDistanceL2 (line 28) | class ChamferDistanceL2(torch.nn.Module):
    method __init__ (line 31) | def __init__(self, ignore_zeros=False):
    method forward (line 35) | def forward(self, xyz1, xyz2):
  class ChamferDistanceL2_split (line 46) | class ChamferDistanceL2_split(torch.nn.Module):
    method __init__ (line 49) | def __init__(self, ignore_zeros=False):
    method forward (line 53) | def forward(self, xyz1, xyz2):
  class ChamferDistanceL1 (line 64) | class ChamferDistanceL1(torch.nn.Module):
    method __init__ (line 67) | def __init__(self, ignore_zeros=False):
    method forward (line 71) | def forward(self, xyz1, xyz2):

FILE: segmentation/extensions/chamfer_dist/chamfer_cuda.cpp
  function chamfer_forward (line 22) | std::vector<torch::Tensor> chamfer_forward(torch::Tensor xyz1,
  function chamfer_backward (line 27) | std::vector<torch::Tensor> chamfer_backward(torch::Tensor xyz1,
  function PYBIND11_MODULE (line 36) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {

FILE: segmentation/extensions/chamfer_dist/test.py
  class ChamferDistanceTestCase (line 23) | class ChamferDistanceTestCase(unittest.TestCase):
    method test_chamfer_dist (line 24) | def test_chamfer_dist(self):

FILE: segmentation/extensions/emd/cuda/emd.cpp
  function PYBIND11_MODULE (line 23) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {

FILE: segmentation/extensions/emd/emd.py
  class EarthMoverDistanceFunction (line 5) | class EarthMoverDistanceFunction(torch.autograd.Function):
    method forward (line 7) | def forward(ctx, xyz1, xyz2):
    method backward (line 17) | def backward(ctx, grad_cost):
  class earth_mover_distance (line 26) | class earth_mover_distance(torch.nn.Module):
    method __init__ (line 29) | def __init__(self):
    method forward (line 32) | def forward(self, xyz1, xyz2, transpose=False):

FILE: segmentation/logger.py
  function get_root_logger (line 18) | def get_root_logger(log_file=None, log_level=logging.INFO, name='main'):
  function get_logger (line 41) | def get_logger(name, log_file=None, log_level=logging.INFO, file_mode='w'):
  function print_log (line 115) | def print_log(msg, logger=None, level=logging.INFO):
  function get_missing_parameters_message (line 141) | def get_missing_parameters_message(keys: List[str]) -> str:
  function get_unexpected_parameters_message (line 158) | def get_unexpected_parameters_message(keys: List[str]) -> str:
  function _strip_prefix_if_present (line 175) | def _strip_prefix_if_present(state_dict: Dict[str, Any], prefix: str) ->...
  function _group_checkpoint_keys (line 208) | def _group_checkpoint_keys(keys: List[str]) -> Dict[str, List[str]]:
  function _group_to_str (line 229) | def _group_to_str(group: List[str]) -> str:
  function _named_modules_with_dup (line 246) | def _named_modules_with_dup(

FILE: segmentation/main.py
  function inplace_relu (line 35) | def inplace_relu(m):
  function to_categorical (line 41) | def to_categorical(y, num_classes):
  function parse_args (line 49) | def parse_args():
  function get_model_loss (line 79) | def get_model_loss(MODEL, args, num_part):
  function main (line 97) | def main(args):

FILE: segmentation/misc.py
  function fps (line 13) | def fps(data, number):
  function worker_init_fn (line 23) | def worker_init_fn(worker_id):
  function build_lambda_sche (line 27) | def build_lambda_sche(opti, config):
  function build_lambda_bnsche (line 36) | def build_lambda_bnsche(model, config):
  function set_random_seed (line 45) | def set_random_seed(seed, deterministic=False):
  function is_seq_of (line 72) | def is_seq_of(seq, expected_type, seq_type=None):
  function set_bn_momentum_default (line 94) | def set_bn_momentum_default(bn_momentum):
  class BNMomentumScheduler (line 102) | class BNMomentumScheduler(object):
    method __init__ (line 104) | def __init__(
    method step (line 122) | def step(self, epoch=None):
    method get_momentum (line 129) | def get_momentum(self, epoch=None):
  function seprate_point_cloud (line 135) | def seprate_point_cloud(xyz, num_points, crop, fixed_points=None, paddin...
  function get_ptcloud_img (line 191) | def get_ptcloud_img(ptcloud):
  function visualize_KITTI (line 211) | def visualize_KITTI(path, data_list, titles=['input', 'pred'], cmap=['bw...
  function random_dropping (line 241) | def random_dropping(pc, e):
  function random_scale (line 251) | def random_scale(partial, scale_range=[0.8, 1.2]):

FILE: segmentation/models/gpt2_seg.py
  class Block (line 7) | class Block(nn.Module):
    method __init__ (line 8) | def __init__(self, embed_dim, num_heads, drop_path):
    method forward (line 21) | def forward(self, x):
  class GPT_extractor (line 35) | class GPT_extractor(nn.Module):
    method __init__ (line 36) | def __init__(
    method forward (line 62) | def forward(self, h, pos, classify=False):
  class GPT_generator (line 95) | class GPT_generator(nn.Module):
    method __init__ (line 96) | def __init__(
    method forward (line 123) | def forward(self, h, pos):

FILE: segmentation/models/pointnet2_utils.py
  function timeit (line 7) | def timeit(tag, t):
  function pc_normalize (line 11) | def pc_normalize(pc):
  function square_distance (line 19) | def square_distance(src, dst):
  function index_points (line 41) | def index_points(points, idx):
  function farthest_point_sample (line 60) | def farthest_point_sample(xyz, npoint):
  function query_ball_point (line 84) | def query_ball_point(radius, nsample, xyz, new_xyz):
  function sample_and_group (line 107) | def sample_and_group(npoint, radius, nsample, xyz, points, returnfps=Fal...
  function sample_and_group_all (line 138) | def sample_and_group_all(xyz, points):
  class PointNetSetAbstraction (line 158) | class PointNetSetAbstraction(nn.Module):
    method __init__ (line 159) | def __init__(self, npoint, radius, nsample, in_channel, mlp, group_all):
    method forward (line 173) | def forward(self, xyz, points):
  class PointNetSetAbstractionMsg (line 202) | class PointNetSetAbstractionMsg(nn.Module):
    method __init__ (line 203) | def __init__(self, npoint, radius_list, nsample_list, in_channel, mlp_...
    method forward (line 221) | def forward(self, xyz, points):
  class PointNetFeaturePropagation (line 262) | class PointNetFeaturePropagation(nn.Module):
    method __init__ (line 263) | def __init__(self, in_channel, mlp):
    method forward (line 273) | def forward(self, xyz1, xyz2, points1, points2):

FILE: segmentation/models/pt.py
  function fps (line 17) | def fps(data, number):
  class Group (line 28) | class Group(nn.Module):
    method __init__ (line 29) | def __init__(self, num_group, group_size):
    method simplied_morton_sorting (line 36) | def simplied_morton_sorting(self, xyz, center):
    method morton_sorting (line 65) | def morton_sorting(self, xyz, center):
    method forward (line 83) | def forward(self, xyz):
  class Encoder_small (line 121) | class Encoder_small(nn.Module):
    method __init__ (line 122) | def __init__(self, encoder_channel):
    method forward (line 138) | def forward(self, point_groups):
  class Encoder_large (line 154) | class Encoder_large(nn.Module):  # Embedding module
    method __init__ (line 155) | def __init__(self, encoder_channel):
    method forward (line 174) | def forward(self, point_groups):
  class Mlp (line 191) | class Mlp(nn.Module):
    method __init__ (line 192) | def __init__(self, in_features, hidden_features=None, out_features=Non...
    method forward (line 201) | def forward(self, x):
  class Attention (line 210) | class Attention(nn.Module):
    method __init__ (line 211) | def __init__(self, dim, num_heads=8, qkv_bias=False, qk_scale=None, at...
    method forward (line 223) | def forward(self, x):
  class Block (line 240) | class Block(nn.Module):
    method __init__ (line 241) | def __init__(self, dim, num_heads, mlp_ratio=4., qkv_bias=False, qk_sc...
    method forward (line 255) | def forward(self, x):
  class TransformerEncoder (line 261) | class TransformerEncoder(nn.Module):
    method __init__ (line 265) | def __init__(self, embed_dim=768, depth=4, num_heads=12, mlp_ratio=4.,...
    method forward (line 278) | def forward(self, x, pos):
  class PositionEmbeddingCoordsSine (line 288) | class PositionEmbeddingCoordsSine(nn.Module):
    method __init__ (line 299) | def __init__(self, n_dim: int = 1, d_model: int = 256, temperature=100...
    method forward (line 311) | def forward(self, xyz: torch.Tensor) -> torch.Tensor:
  class get_model (line 338) | class get_model(nn.Module):
    method __init__ (line 339) | def __init__(self, cls_dim, trans_dim=384, depth=12, drop_path_rate=0....
    method get_loss_acc (line 416) | def get_loss_acc(self, ret, gt):
    method load_model_from_ckpt (line 422) | def load_model_from_ckpt(self, bert_ckpt_path):
    method forward (line 454) | def forward(self, pts, cls_label):
  class get_loss (line 522) | class get_loss(nn.Module):
    method __init__ (line 523) | def __init__(self):
    method forward (line 526) | def forward(self, pred, target):

FILE: segmentation/models/z_order.py
  function round_to_int_32 (line 4) | def round_to_int_32(data):
  function split_by_3 (line 23) | def split_by_3(x):
  function get_z_order (line 50) | def get_z_order(x, y, z):
  function get_z_values (line 72) | def get_z_values(data):

FILE: segmentation/pointnet_util.py
  function timeit (line 11) | def timeit(tag, t):
  function pc_normalize (line 15) | def pc_normalize(pc):
  function square_distance (line 22) | def square_distance(src, dst):
  function index_points (line 39) | def index_points(points, idx):
  function farthest_point_sample (line 53) | def farthest_point_sample(xyz, npoint):
  function query_ball_point (line 76) | def query_ball_point(radius, nsample, xyz, new_xyz):
  function sample_and_group (line 99) | def sample_and_group(npoint, radius, nsample, xyz, points, returnfps=Fal...
  function sample_and_group_all (line 139) | def sample_and_group_all(xyz, points):
  class PointNetSetAbstraction (line 159) | class PointNetSetAbstraction(nn.Module):
    method __init__ (line 160) | def __init__(self, npoint, radius, nsample, in_channel, mlp, group_all...
    method forward (line 175) | def forward(self, xyz, points):
  class PointNetSetAbstractionMsg (line 199) | class PointNetSetAbstractionMsg(nn.Module):
    method __init__ (line 200) | def __init__(self, npoint, radius_list, nsample_list, in_channel, mlp_...
    method forward (line 219) | def forward(self, xyz, points, seed_idx=None):
  class PointNetFeaturePropagation (line 261) | class PointNetFeaturePropagation(nn.Module):
    method __init__ (line 262) | def __init__(self, in_channel, mlp):
    method forward (line 272) | def forward(self, xyz1, xyz2, points1, points2):

FILE: segmentation/provider.py
  function normalize_data (line 3) | def normalize_data(batch_data):
  function shuffle_data (line 22) | def shuffle_data(data, labels):
  function shuffle_points (line 34) | def shuffle_points(batch_data):
  function rotate_point_cloud (line 46) | def rotate_point_cloud(batch_data):
  function rotate_point_cloud_z (line 66) | def rotate_point_cloud_z(batch_data):
  function rotate_point_cloud_with_normal (line 86) | def rotate_point_cloud_with_normal(batch_xyz_normal):
  function rotate_perturbation_point_cloud_with_normal (line 106) | def rotate_perturbation_point_cloud_with_normal(batch_data, angle_sigma=...
  function rotate_point_cloud_by_angle (line 133) | def rotate_point_cloud_by_angle(batch_data, rotation_angle):
  function rotate_point_cloud_by_angle_with_normal (line 152) | def rotate_point_cloud_by_angle_with_normal(batch_data, rotation_angle):
  function rotate_perturbation_point_cloud (line 176) | def rotate_perturbation_point_cloud(batch_data, angle_sigma=0.06, angle_...
  function jitter_point_cloud (line 201) | def jitter_point_cloud(batch_data, sigma=0.01, clip=0.05):
  function shift_point_cloud (line 214) | def shift_point_cloud(batch_data, shift_range=0.1):
  function random_scale_point_cloud (line 228) | def random_scale_point_cloud(batch_data, scale_low=0.8, scale_high=1.25):
  function random_point_dropout (line 241) | def random_point_dropout(batch_pc, max_dropout_ratio=0.875):

FILE: tools/builder.py
  function dataset_builder (line 16) | def dataset_builder(args, config):
  function model_builder (line 39) | def model_builder(config):
  function build_opti_sche (line 44) | def build_opti_sche(base_model, config):
  function resume_model (line 103) | def resume_model(base_model, args, logger=None):
  function resume_optimizer (line 133) | def resume_optimizer(optimizer, args, logger=None):
  function save_checkpoint (line 147) | def save_checkpoint(base_model, optimizer, epoch, metrics, best_metrics,...
  function load_model (line 160) | def load_model(base_model, ckpt_path, logger=None):

FILE: tools/runner.py
  function test_net (line 14) | def test_net(args, config):
  function test (line 34) | def test(base_model, test_dataloader, args, config, logger=None):

FILE: tools/runner_finetune.py
  class Acc_Metric (line 37) | class Acc_Metric:
    method __init__ (line 38) | def __init__(self, acc=0.):
    method better_than (line 46) | def better_than(self, other):
    method state_dict (line 52) | def state_dict(self):
  function run_net (line 58) | def run_net(args, config, train_writer=None, val_writer=None):
  function validate (line 249) | def validate(base_model, test_dataloader, epoch, val_writer, args, confi...
  function validate_vote (line 293) | def validate_vote(base_model, test_dataloader, epoch, val_writer, args, ...
  function test_net (line 361) | def test_net(args, config):
  function test (line 380) | def test(base_model, test_dataloader, args, config, logger=None):
  function test_vote (line 429) | def test_vote(base_model, test_dataloader, epoch, val_writer, args, conf...

FILE: tools/runner_pretrain.py
  class Acc_Metric (line 31) | class Acc_Metric:
    method __init__ (line 32) | def __init__(self, acc=0.):
    method better_than (line 38) | def better_than(self, other):
    method state_dict (line 44) | def state_dict(self):
  function evaluate_svm (line 50) | def evaluate_svm(train_features, train_labels, test_features, test_labels):
  function run_net (line 57) | def run_net(args, config, train_writer=None, val_writer=None):
  function validate (line 209) | def validate(base_model, extra_train_dataloader, test_dataloader, epoch,...
  function test_net (line 272) | def test_net():

FILE: utils/AverageMeter.py
  class AverageMeter (line 2) | class AverageMeter(object):
    method __init__ (line 3) | def __init__(self, items=None):
    method reset (line 8) | def reset(self):
    method update (line 13) | def update(self, values):
    method val (line 24) | def val(self, idx=None):
    method count (line 30) | def count(self, idx=None):
    method avg (line 36) | def avg(self, idx=None):

FILE: utils/checkpoint.py
  function get_missing_parameters_message (line 16) | def get_missing_parameters_message(keys: List[str]) -> str:
  function get_unexpected_parameters_message (line 33) | def get_unexpected_parameters_message(keys: List[str]) -> str:
  function _strip_prefix_if_present (line 50) | def _strip_prefix_if_present(state_dict: Dict[str, Any], prefix: str) ->...
  function _group_checkpoint_keys (line 83) | def _group_checkpoint_keys(keys: List[str]) -> Dict[str, List[str]]:
  function _group_to_str (line 104) | def _group_to_str(group: List[str]) -> str:
  function _named_modules_with_dup (line 121) | def _named_modules_with_dup(

FILE: utils/config.py
  function log_args_to_file (line 7) | def log_args_to_file(args, pre='args', logger=None):
  function log_config_to_file (line 12) | def log_config_to_file(cfg, pre='cfg', logger=None):
  function merge_new_config (line 21) | def merge_new_config(config, new_config):
  function cfg_from_yaml_file (line 41) | def cfg_from_yaml_file(cfg_file):
  function get_config (line 52) | def get_config(args, logger=None):
  function save_experiment_config (line 66) | def save_experiment_config(args, config, logger=None):

FILE: utils/dist_utils.py
  function init_dist (line 9) | def init_dist(launcher, backend='nccl', **kwargs):
  function _init_dist_pytorch (line 18) | def _init_dist_pytorch(backend, **kwargs):
  function get_dist_info (line 27) | def get_dist_info():
  function reduce_tensor (line 41) | def reduce_tensor(tensor, args):
  function gather_tensor (line 50) | def gather_tensor(tensor, args):

FILE: utils/logger.py
  function get_root_logger (line 6) | def get_root_logger(log_file=None, log_level=logging.INFO, name='main'):
  function get_logger (line 29) | def get_logger(name, log_file=None, log_level=logging.INFO, file_mode='w'):
  function print_log (line 103) | def print_log(msg, logger=None, level=logging.INFO):

FILE: utils/misc.py
  function fps (line 13) | def fps(data, number):
  function worker_init_fn (line 23) | def worker_init_fn(worker_id):
  function build_lambda_sche (line 26) | def build_lambda_sche(opti, config):
  function build_lambda_bnsche (line 34) | def build_lambda_bnsche(model, config):
  function set_random_seed (line 42) | def set_random_seed(seed, deterministic=False):
  function is_seq_of (line 69) | def is_seq_of(seq, expected_type, seq_type=None):
  function set_bn_momentum_default (line 91) | def set_bn_momentum_default(bn_momentum):
  class BNMomentumScheduler (line 97) | class BNMomentumScheduler(object):
    method __init__ (line 99) | def __init__(
    method step (line 117) | def step(self, epoch=None):
    method get_momentum (line 124) | def get_momentum(self, epoch=None):
  function seprate_point_cloud (line 131) | def seprate_point_cloud(xyz, num_points, crop, fixed_points = None, padd...
  function get_ptcloud_img (line 186) | def get_ptcloud_img(ptcloud,roll,pitch):
  function visualize_KITTI (line 207) | def visualize_KITTI(path, data_list, titles = ['input','pred'], cmap=['b...
  function random_dropping (line 236) | def random_dropping(pc, e):
  function random_scale (line 246) | def random_scale(partial, scale_range=[0.8, 1.2]):

FILE: utils/parser.py
  function get_args (line 5) | def get_args():
  function create_experiment_dir (line 104) | def create_experiment_dir(args):

FILE: utils/registry.py
  class Registry (line 6) | class Registry:
    method __init__ (line 32) | def __init__(self, name, build_func=None, parent=None, scope=None):
    method __len__ (line 56) | def __len__(self):
    method __contains__ (line 59) | def __contains__(self, key):
    method __repr__ (line 62) | def __repr__(self):
    method infer_scope (line 69) | def infer_scope():
    method split_scope_key (line 89) | def split_scope_key(key):
    method name (line 108) | def name(self):
    method scope (line 112) | def scope(self):
    method module_dict (line 116) | def module_dict(self):
    method children (line 120) | def children(self):
    method get (line 123) | def get(self, key):
    method build (line 146) | def build(self, *args, **kwargs):
    method _add_children (line 149) | def _add_children(self, registry):
    method _register_module (line 168) | def _register_module(self, module_class, module_name=None, force=False):
    method deprecated_register_module (line 183) | def deprecated_register_module(self, cls=None, force=False):
    method register_module (line 193) | def register_module(self, name=None, force=False, module=None):
  function build_from_cfg (line 246) | def build_from_cfg(cfg, registry, default_args=None):