SYMBOL INDEX (297 symbols across 44 files) FILE: classification/dataset/ScanObjectNNDataLoader.py class ScanObjectNNDataLoader (line 13) | class ScanObjectNNDataLoader(Dataset): method __init__ (line 14) | def __init__(self, root, split='training', bg=True): method __len__ (line 31) | def __len__(self): method __getitem__ (line 34) | def __getitem__(self, index): FILE: classification/models/repsurf/repsurf_ssg_umb.py class Model (line 11) | class Model(nn.Module): method __init__ (line 12) | def __init__(self, args): method forward (line 43) | def forward(self, points): FILE: classification/models/repsurf/repsurf_ssg_umb_2x.py class Model (line 11) | class Model(nn.Module): method __init__ (line 12) | def __init__(self, args): method forward (line 46) | def forward(self, points): FILE: classification/modules/pointnet2_utils.py function square_distance (line 15) | def square_distance(src, dst): function index_points (line 28) | def index_points(points, idx, cuda=False, is_group=False): function farthest_point_sample (line 47) | def farthest_point_sample(xyz, npoint, cuda=False): function query_ball_point (line 78) | def query_ball_point(radius, nsample, xyz, new_xyz, debug=False, cuda=Fa... function query_knn_point (line 102) | def query_knn_point(k, xyz, new_xyz, cuda=False): function sample (line 114) | def sample(nsample, feature, cuda=False): FILE: classification/modules/pointops/functions/pointops.py class FurthestSampling (line 35) | class FurthestSampling(Function): method forward (line 37) | def forward(ctx, xyz, m): method backward (line 50) | def backward(xyz, a=None): class Gathering (line 57) | class Gathering(Function): method forward (line 59) | def forward(ctx, features, idx): method backward (line 74) | def backward(ctx, grad_out): class NearestNeighbor (line 86) | class NearestNeighbor(Function): method forward (line 88) | def forward(ctx, unknown: torch.Tensor, known: torch.Tensor) -> Tuple[... method backward (line 105) | def backward(ctx, a=None, b=None): class Interpolation (line 112) | class Interpolation(Function): method forward (line 114) | def forward(ctx, features: torch.Tensor, idx: torch.Tensor, weight: to... method backward (line 134) | def backward(ctx, grad_out: torch.Tensor) -> Tuple[torch.Tensor, torch... class Grouping (line 150) | class Grouping(Function): method forward (line 152) | def forward(ctx, features: torch.Tensor, idx: torch.Tensor) -> torch.T... method backward (line 167) | def backward(ctx, grad_out: torch.Tensor) -> Tuple[torch.Tensor, torch... class GroupingInt (line 183) | class GroupingInt(Function): method forward (line 185) | def forward(ctx, features: torch.Tensor, idx: torch.Tensor) -> torch.T... method backward (line 199) | def backward(ctx, a=None): class BallQuery (line 206) | class BallQuery(Function): method forward (line 208) | def forward(ctx, radius: float, nsample: int, xyz: torch.Tensor, new_x... method backward (line 225) | def backward(ctx, a=None): function pairwise_distances (line 232) | def pairwise_distances(x, y=None): class KNNQueryNaive (line 252) | class KNNQueryNaive(Function): method forward (line 254) | def forward(ctx, nsample: int, xyz: torch.Tensor, new_xyz: torch.Tenso... method backward (line 287) | def backward(ctx): class KNNQuery (line 294) | class KNNQuery(Function): method forward (line 296) | def forward(ctx, nsample: int, xyz: torch.Tensor, new_xyz: torch.Tenso... method backward (line 319) | def backward(ctx, a=None): class KNNQuery_Heap (line 326) | class KNNQuery_Heap(Function): method forward (line 328) | def forward(ctx, nsample: int, xyz: torch.Tensor, new_xyz: torch.Tenso... method backward (line 350) | def backward(ctx, a=None): class QueryAndGroup (line 357) | class QueryAndGroup(nn.Module): method __init__ (line 365) | def __init__(self, radius=None, nsample=32, use_xyz=True, return_idx=F... method forward (line 370) | def forward(self, xyz: torch.Tensor, new_xyz: torch.Tensor = None, fea... FILE: classification/modules/pointops/src/ballquery/ballquery_cuda.cpp function ballquery_cuda (line 14) | void ballquery_cuda(int b, int n, int m, float radius, int nsample, at::... function ballquery_cuda_fast (line 24) | void ballquery_cuda_fast(int b, int n, int m, float radius, int nsample,... FILE: classification/modules/pointops/src/cuda_utils.h function opt_n_threads (line 15) | inline int opt_n_threads(int work_size) { function dim3 (line 20) | inline dim3 opt_block_config(int x, int y) { FILE: classification/modules/pointops/src/grouping/grouping_cuda.cpp function grouping_forward_cuda (line 10) | void grouping_forward_cuda(int b, int c, int n, int m, int nsample, at::... function grouping_backward_cuda (line 18) | void grouping_backward_cuda(int b, int c, int n, int m, int nsample, at:... function grouping_forward_cuda_fast (line 26) | void grouping_forward_cuda_fast(int b, int c, int n, int npoints, int ns... FILE: classification/modules/pointops/src/grouping_int/grouping_int_cuda.cpp function grouping_int_forward_cuda (line 10) | void grouping_int_forward_cuda(int b, int c, int n, int m, int nsample, ... function grouping_int_forward_cuda_fast (line 18) | void grouping_int_forward_cuda_fast(int b, int c, int n, int m, int nsam... FILE: classification/modules/pointops/src/interpolation/interpolation_cuda.cpp function nearestneighbor_cuda (line 9) | void nearestneighbor_cuda(int b, int n, int m, at::Tensor unknown_tensor... function interpolation_forward_cuda (line 18) | void interpolation_forward_cuda(int b, int c, int m, int n, at::Tensor p... function interpolation_backward_cuda (line 27) | void interpolation_backward_cuda(int b, int c, int n, int m, at::Tensor ... function nearestneighbor_cuda_fast (line 36) | void nearestneighbor_cuda_fast(int b, int n, int m, at::Tensor unknown_t... function interpolation_forward_cuda_fast (line 44) | void interpolation_forward_cuda_fast(int b, int c, int m, int n, at::Ten... FILE: classification/modules/pointops/src/knnquery/knnquery_cuda.cpp function knnquery_cuda (line 15) | void knnquery_cuda(int b, int n, int m, int nsample, at::Tensor xyz_tens... FILE: classification/modules/pointops/src/knnquery_heap/knnquery_heap_cuda.cpp function knnquery_heap_cuda (line 15) | void knnquery_heap_cuda(int b, int n, int m, int nsample, at::Tensor xyz... FILE: classification/modules/pointops/src/pointops_api.cpp function PYBIND11_MODULE (line 13) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) { FILE: classification/modules/pointops/src/sampling/sampling_cuda.cpp function gathering_forward_cuda (line 9) | void gathering_forward_cuda(int b, int c, int n, int m, at::Tensor point... function gathering_backward_cuda (line 17) | void gathering_backward_cuda(int b, int c, int n, int m, at::Tensor grad... function furthestsampling_cuda (line 26) | void furthestsampling_cuda(int b, int n, int m, at::Tensor points_tensor... FILE: classification/modules/polar_utils.py function xyz2sphere (line 10) | def xyz2sphere(xyz, normalize=True): function xyz2cylind (line 34) | def xyz2cylind(xyz, normalize=True): FILE: classification/modules/ptaug_utils.py function get_aug_args (line 13) | def get_aug_args(args): function transform_point_cloud (line 22) | def transform_point_cloud(batch, args, aug_args, train=True, label=None): function shift_point_cloud (line 37) | def shift_point_cloud(batch_data, shift_range=0.2): function scale_point_cloud (line 53) | def scale_point_cloud(batch_data, scale_range=0.2): FILE: classification/modules/recons_utils.py function _recons_factory (line 11) | def _recons_factory(type): function knn_recons (line 18) | def knn_recons(k, center, context, cuda=False): function cal_normal (line 27) | def cal_normal(group_xyz, random_inv=False, is_group=False): function pca (line 60) | def pca(X, k, center=True): function cal_center (line 82) | def cal_center(group_xyz): function cal_area (line 93) | def cal_area(group_xyz): function cal_const (line 108) | def cal_const(normal, center, is_normalize=True): function check_nan (line 127) | def check_nan(normal, center, pos=None): function check_nan_umb (line 152) | def check_nan_umb(normal, center, pos=None): class SurfaceConstructor (line 179) | class SurfaceConstructor(nn.Module): method __init__ (line 188) | def __init__(self, r=None, k=3, recons_type='knn', return_dist=False, ... method forward (line 198) | def forward(self, center, context): FILE: classification/modules/repsurface_utils.py function sample_and_group (line 15) | def sample_and_group(npoint, radius, nsample, center, normal, feature, r... function sample_and_group_all (line 62) | def sample_and_group_all(center, normal, feature, return_normal=True, re... function resort_points (line 91) | def resort_points(points, idx): function group_by_umbrella (line 112) | def group_by_umbrella(xyz, new_xyz, k=9, cuda=False): class SurfaceAbstraction (line 135) | class SurfaceAbstraction(nn.Module): method __init__ (line 141) | def __init__(self, npoint, radius, nsample, in_channel, mlp, group_all... method forward (line 159) | def forward(self, center, normal, feature): class SurfaceAbstractionCD (line 186) | class SurfaceAbstractionCD(nn.Module): method __init__ (line 192) | def __init__(self, npoint, radius, nsample, feat_channel, pos_channel,... method forward (line 218) | def forward(self, center, normal, feature): class UmbrellaSurfaceConstructor (line 252) | class UmbrellaSurfaceConstructor(nn.Module): method __init__ (line 258) | def __init__(self, k, in_channel, aggr_type='sum', return_dist=False, ... method forward (line 276) | def forward(self, center): FILE: classification/tool/train_cls_scanobjectnn.py function parse_args (line 22) | def parse_args(): function test (line 70) | def test(model, loader, num_class=15, num_point=1024, num_votes=10, tota... function main (line 111) | def main(args): FILE: classification/util/utils.py function set_seed (line 11) | def set_seed(seed): function weight_init (line 27) | def weight_init(m, init_type): class ClsLoss (line 45) | class ClsLoss(nn.Module): method __init__ (line 46) | def __init__(self): method forward (line 49) | def forward(self, pred, target): class SmoothClsLoss (line 55) | class SmoothClsLoss(nn.Module): method __init__ (line 56) | def __init__(self, smoothing_ratio=0.1): method forward (line 60) | def forward(self, pred, target): function get_model (line 72) | def get_model(args): function get_loss (line 77) | def get_loss(): function get_test_args (line 81) | def get_test_args(): FILE: segmentation/dataset/S3DISDataLoader.py class S3DIS (line 16) | class S3DIS(Dataset): method __init__ (line 17) | def __init__(self, args, split, coord_transform=None, rgb_transform=None, method __getitem__ (line 37) | def __getitem__(self, idx): method __len__ (line 47) | def __len__(self): method print_weight (line 51) | def print_weight(data_root, data_list): method print_mean_std (line 72) | def print_mean_std(data_root, data_list): FILE: segmentation/models/pointnet2/pointnet2_ssg.py class Model (line 11) | class Model(nn.Module): method __init__ (line 12) | def __init__(self, args): method forward (line 32) | def forward(self, pos_feat_off0): FILE: segmentation/models/pointtransformer/pointtransformer.py class Model (line 6) | class Model(nn.Module): method __init__ (line 7) | def __init__(self, args): method _make_enc (line 33) | def _make_enc(self, block, planes, blocks, share_planes=8, stride=1, n... method _make_dec (line 40) | def _make_dec(self, block, planes, blocks, share_planes=8, nsample=16,... method forward (line 47) | def forward(self, pxo, *args): FILE: segmentation/models/repsurf/repsurf_umb_ssg.py class Model (line 11) | class Model(nn.Module): method __init__ (line 12) | def __init__(self, args): method forward (line 42) | def forward(self, pos_feat_off0): FILE: segmentation/modules/aug_utils.py function transform_point_cloud_coord (line 9) | def transform_point_cloud_coord(args): function transform_point_cloud_rgb (line 38) | def transform_point_cloud_rgb(args): class Compose (line 54) | class Compose(object): method __init__ (line 55) | def __init__(self, transforms): method __call__ (line 58) | def __call__(self, coord, feat, label, mask=None): class RandomRotate (line 64) | class RandomRotate(object): method __init__ (line 65) | def __init__(self, rot=(np.pi/24, np.pi/24, np.pi/4), prob=1.): method __call__ (line 69) | def __call__(self, coord, feat, label, mask=None): class RandomRotateAligned (line 85) | class RandomRotateAligned(object): method __init__ (line 86) | def __init__(self, rot=np.pi, prob=1.): method __call__ (line 90) | def __call__(self, coord, feat, label, mask=None): class RandomRotatePerturb (line 99) | class RandomRotatePerturb(object): method __init__ (line 100) | def __init__(self, sigma=0.03, clip=0.09, prob=1.): method __call__ (line 105) | def __call__(self, coord, feat, label, mask=None): class RandomRotatePerturbAligned (line 121) | class RandomRotatePerturbAligned(object): method __init__ (line 122) | def __init__(self, sigma=0.03, clip=0.09, prob=1.): method __call__ (line 127) | def __call__(self, coord, feat, label, mask=None): class RandomScale (line 136) | class RandomScale(object): method __init__ (line 137) | def __init__(self, scale=0.1, anisotropic=False, prob=1.): method __call__ (line 142) | def __call__(self, coord, feat, label, mask=None): class RandomShift (line 149) | class RandomShift(object): method __init__ (line 150) | def __init__(self, shift=(0.2, 0.2, 0), p=0.95): method __call__ (line 154) | def __call__(self, coord, feat, label, mask=None): class RandomFlip (line 163) | class RandomFlip(object): method __init__ (line 164) | def __init__(self, p=1.): method __call__ (line 167) | def __call__(self, coord, feat, label, mask=None): class RandomJitter (line 176) | class RandomJitter(object): method __init__ (line 177) | def __init__(self, sigma=0.01, clip=0.05, p=1., is_lidar=False): method __call__ (line 183) | def __call__(self, coord, feat, label, mask=None): class ChromaticAutoContrast (line 193) | class ChromaticAutoContrast(object): method __init__ (line 194) | def __init__(self, p=0.2, blend_factor=None): method __call__ (line 198) | def __call__(self, coord, feat, label, mask=None): class ChromaticTranslation (line 214) | class ChromaticTranslation(object): method __init__ (line 215) | def __init__(self, p=0.95, ratio=0.05): method __call__ (line 219) | def __call__(self, coord, feat, label, mask=None): class ChromaticJitter (line 228) | class ChromaticJitter(object): method __init__ (line 229) | def __init__(self, p=0.95, std=0.005): method __call__ (line 233) | def __call__(self, coord, feat, label, mask=None): class HueSaturationTranslation (line 243) | class HueSaturationTranslation(object): method rgb_to_hsv (line 245) | def rgb_to_hsv(rgb): method hsv_to_rgb (line 270) | def hsv_to_rgb(hsv): method __init__ (line 290) | def __init__(self, hue_max=0.5, saturation_max=0.2, p=1.): method __call__ (line 295) | def __call__(self, coord, feat, label, mask=None): class RandomDropColor (line 312) | class RandomDropColor(object): method __init__ (line 313) | def __init__(self, p=0.2): method __call__ (line 316) | def __call__(self, coord, feat, label, mask=None): FILE: segmentation/modules/pointnet2_utils.py function sample_and_group (line 13) | def sample_and_group(stride, nsample, xyz, points, offset, return_idx=Fa... class PointNetSetAbstraction (line 49) | class PointNetSetAbstraction(nn.Module): method __init__ (line 55) | def __init__(self, stride, nsample, in_channel, mlp, num_sector=1): method forward (line 69) | def forward(self, pos_feat_off): class PointNetFeaturePropagation (line 86) | class PointNetFeaturePropagation(nn.Module): method __init__ (line 92) | def __init__(self, in_channel, mlp): method forward (line 102) | def forward(self, pos_feat_off1, pos_feat_off2): FILE: segmentation/modules/pointops/functions/pointops.py class FurthestSampling (line 31) | class FurthestSampling(Function): method forward (line 33) | def forward(ctx, xyz, offset, new_offset): class SectorizedFurthestSampling (line 52) | class SectorizedFurthestSampling(Function): method forward (line 54) | def forward(ctx, xyz, offset, new_offset, num_sectors, min_points=10000): class KNNQuery (line 114) | class KNNQuery(Function): method forward (line 116) | def forward(ctx, nsample, xyz, new_xyz, offset, new_offset): class Grouping (line 133) | class Grouping(Function): method forward (line 135) | def forward(ctx, input, idx): method backward (line 149) | def backward(ctx, grad_output): function queryandgroup (line 165) | def queryandgroup(nsample, xyz, new_xyz, feat, idx, offset, new_offset, ... class Subtraction (line 189) | class Subtraction(Function): method forward (line 191) | def forward(ctx, input1, input2, idx): method backward (line 205) | def backward(ctx, grad_output): class Aggregation (line 221) | class Aggregation(Function): method forward (line 223) | def forward(ctx, input, position, weight, idx): method backward (line 237) | def backward(ctx, grad_output): function interpolation (line 256) | def interpolation(xyz, new_xyz, feat, offset, new_offset, k=3): class Interpolation (line 273) | class Interpolation(Function): method forward (line 275) | def forward(ctx, xyz, new_xyz, input, offset, new_offset, k=3): method backward (line 294) | def backward(ctx, grad_output): FILE: segmentation/modules/pointops/src/aggregation/aggregation_cuda.cpp function aggregation_forward_cuda (line 8) | void aggregation_forward_cuda(int n, int nsample, int c, int w_c, at::Te... function aggregation_backward_cuda (line 18) | void aggregation_backward_cuda(int n, int nsample, int c, int w_c, at::T... FILE: segmentation/modules/pointops/src/cuda_utils.h function opt_n_threads (line 11) | inline int opt_n_threads(int work_size) { function dim3 (line 16) | inline dim3 opt_block_config(int x, int y) { FILE: segmentation/modules/pointops/src/grouping/grouping_cuda.cpp function grouping_forward_cuda (line 8) | void grouping_forward_cuda(int m, int nsample, int c, at::Tensor input_t... function grouping_backward_cuda (line 16) | void grouping_backward_cuda(int m, int nsample, int c, at::Tensor grad_o... FILE: segmentation/modules/pointops/src/interpolation/interpolation_cuda.cpp function interpolation_forward_cuda (line 8) | void interpolation_forward_cuda(int n, int c, int k, at::Tensor input_te... function interpolation_backward_cuda (line 17) | void interpolation_backward_cuda(int n, int c, int k, at::Tensor grad_ou... FILE: segmentation/modules/pointops/src/knnquery/knnquery_cuda.cpp function knnquery_cuda (line 8) | void knnquery_cuda(int m, int nsample, at::Tensor xyz_tensor, at::Tensor... FILE: segmentation/modules/pointops/src/pointops_api.cpp function PYBIND11_MODULE (line 12) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) { FILE: segmentation/modules/pointops/src/sampling/sampling_cuda.cpp function furthestsampling_cuda (line 8) | void furthestsampling_cuda(int b, int n, at::Tensor xyz_tensor, at::Tens... FILE: segmentation/modules/pointops/src/subtraction/subtraction_cuda.cpp function subtraction_forward_cuda (line 8) | void subtraction_forward_cuda(int n, int nsample, int c, at::Tensor inpu... function subtraction_backward_cuda (line 17) | void subtraction_backward_cuda(int n, int nsample, int c, at::Tensor idx... FILE: segmentation/modules/pointtransformer_utils.py class PointTransformerLayer (line 7) | class PointTransformerLayer(nn.Module): method __init__ (line 8) | def __init__(self, in_planes, out_planes, share_planes=8, nsample=16): method forward (line 25) | def forward(self, pxo) -> torch.Tensor: class TransitionDown (line 45) | class TransitionDown(nn.Module): method __init__ (line 46) | def __init__(self, in_planes, out_planes, stride=1, nsample=16, num_se... method forward (line 57) | def forward(self, pxo): class TransitionUp (line 79) | class TransitionUp(nn.Module): method __init__ (line 80) | def __init__(self, in_planes, out_planes=None): method forward (line 92) | def forward(self, pxo1, pxo2=None): class PointTransformerBlock (line 113) | class PointTransformerBlock(nn.Module): method __init__ (line 116) | def __init__(self, in_planes, planes, share_planes=8, nsample=16): method forward (line 126) | def forward(self, pxo): FILE: segmentation/modules/polar_utils.py function xyz2sphere (line 10) | def xyz2sphere(xyz, normalize=True): function xyz2cylind (line 34) | def xyz2cylind(xyz, normalize=True): FILE: segmentation/modules/recons_utils.py function cal_normal (line 10) | def cal_normal(group_xyz, offset, random_inv=False, is_group=False): function cal_center (line 48) | def cal_center(group_xyz): function cal_area (line 59) | def cal_area(group_xyz): function cal_const (line 74) | def cal_const(normal, center, is_normalize=True): function check_nan (line 93) | def check_nan(normal, center, pos=None): function check_nan_umb (line 117) | def check_nan_umb(normal, center, pos=None): FILE: segmentation/modules/repsurface_utils.py function sample_and_group (line 15) | def sample_and_group(stride, nsample, center, normal, feature, offset, r... function resort_points (line 54) | def resort_points(points, idx): function _fixed_rotate (line 71) | def _fixed_rotate(xyz): function group_by_umbrella_v2 (line 77) | def group_by_umbrella_v2(xyz, new_xyz, offset, new_offset, k=9): function group_by_umbrella (line 101) | def group_by_umbrella(xyz, new_xyz, offset, new_offset, k=9): function sort_factory (line 125) | def sort_factory(s_type): class SurfaceAbstraction (line 134) | class SurfaceAbstraction(nn.Module): method __init__ (line 140) | def __init__(self, stride, nsample, in_channel, mlp, return_polar=True... method forward (line 155) | def forward(self, pos_nor_feat_off): class SurfaceAbstractionCD (line 176) | class SurfaceAbstractionCD(nn.Module): method __init__ (line 182) | def __init__(self, stride, nsample, feat_channel, pos_channel, mlp, re... method forward (line 205) | def forward(self, pos_nor_feat_off): class SurfaceFeaturePropagationCD (line 233) | class SurfaceFeaturePropagationCD(nn.Module): method __init__ (line 239) | def __init__(self, prev_channel, skip_channel, mlp): method forward (line 257) | def forward(self, pos_feat_off1, pos_feat_off2): class UmbrellaSurfaceConstructor (line 287) | class UmbrellaSurfaceConstructor(nn.Module): method __init__ (line 293) | def __init__(self, k, in_channel, out_channel, random_inv=True, sort='... method forward (line 306) | def forward(self, center, offset): FILE: segmentation/modules/voxelize_utils.py function fnv_hash_vec (line 4) | def fnv_hash_vec(arr): function ravel_hash_vec (line 20) | def ravel_hash_vec(arr): function voxelize (line 40) | def voxelize(coord, voxel_size=0.05, hash_type='fnv', mode=0): FILE: segmentation/tool/test_s3dis.py function parse_args (line 34) | def parse_args(): function main (line 61) | def main(): function data_prepare (line 105) | def data_prepare(): function data_load (line 114) | def data_load(data_name): function data_process (line 133) | def data_process(coord, feat, idx_data): function input_normalize (line 162) | def input_normalize(coord, feat): function visualize_scene (line 177) | def visualize_scene(coord, pred, label, name): function test (line 186) | def test(model): FILE: segmentation/tool/train.py function parse_args (line 33) | def parse_args(): function main_worker (line 106) | def main_worker(gpu, ngpus_per_node, argss): function train (line 261) | def train(train_loader, model, criterion, optimizer, epoch): function validate (line 350) | def validate(val_loader, model, criterion): FILE: segmentation/util/data_util.py function sa_create (line 8) | def sa_create(name, var): function collate_fn (line 15) | def collate_fn(batch): function data_prepare (line 26) | def data_prepare(coord, feat, label, args, split, coord_transform, rgb_t... FILE: segmentation/util/utils.py function main_process (line 16) | def main_process(args): class AverageMeter (line 25) | class AverageMeter(object): method __init__ (line 28) | def __init__(self): method reset (line 31) | def reset(self): method update (line 37) | def update(self, val, n=1): function intersectionAndUnion (line 44) | def intersectionAndUnion(output, target, K, ignore_index=255): function intersectionAndUnionGPU (line 59) | def intersectionAndUnionGPU(output, target, K, ignore_index=255): function find_free_port (line 74) | def find_free_port(): function set_seed (line 85) | def set_seed(seed): function worker_init_fn (line 99) | def worker_init_fn(worker_id, seed=None): function get_logger (line 108) | def get_logger(log_dir, model): function get_aug_args (line 125) | def get_aug_args(args): function get_dataset_obj (line 136) | def get_dataset_obj(args): function get_dataset_description (line 142) | def get_dataset_description(args): function get_loop (line 150) | def get_loop(args): function get_class_weights (line 159) | def get_class_weights(dataset_name): function get_rgb_stat (line 192) | def get_rgb_stat(args): function get_model (line 202) | def get_model(args): function get_optimizer (line 207) | def get_optimizer(args, model): function get_scheduler (line 219) | def get_scheduler(args, optimizer): function get_loss (line 227) | def get_loss(weight=None, ignore_label=None): function get_test_args (line 231) | def get_test_args(): function pc_median_filter_gpu (line 235) | def pc_median_filter_gpu(coord, label, group_size=16):