SYMBOL INDEX (276 symbols across 30 files)

FILE: lib/pointops/functions/pointops.py
  class FurthestSampling (line 10) | class FurthestSampling(Function):
    method forward (line 12) | def forward(ctx, xyz, m):
    method backward (line 25) | def backward(xyz, a=None):
  class Gathering (line 31) | class Gathering(Function):
    method forward (line 33) | def forward(ctx, features, idx):
    method backward (line 48) | def backward(ctx, grad_out):
  class NearestNeighbor (line 59) | class NearestNeighbor(Function):
    method forward (line 61) | def forward(ctx, unknown: torch.Tensor, known: torch.Tensor) -> Tuple[...
    method backward (line 78) | def backward(ctx, a=None, b=None):
  class Interpolation (line 84) | class Interpolation(Function):
    method forward (line 86) | def forward(ctx, features: torch.Tensor, idx: torch.Tensor, weight: to...
    method backward (line 105) | def backward(ctx, grad_out: torch.Tensor) -> Tuple[torch.Tensor, torch...
  class Grouping (line 120) | class Grouping(Function):
    method forward (line 122) | def forward(ctx, features: torch.Tensor, idx: torch.Tensor) -> torch.T...
    method backward (line 137) | def backward(ctx, grad_out: torch.Tensor) -> Tuple[torch.Tensor, torch...
  class GroupingInt (line 152) | class GroupingInt(Function):
    method forward (line 154) | def forward(ctx, features: torch.Tensor, idx: torch.Tensor) -> torch.T...
    method backward (line 168) | def backward(ctx, a=None):
  class BallQuery (line 174) | class BallQuery(Function):
    method forward (line 176) | def forward(ctx, radius: float, nsample: int, xyz: torch.Tensor, new_x...
    method backward (line 193) | def backward(ctx, a=None):
  class FeatureDistribute (line 199) | class FeatureDistribute(Function):
    method forward (line 201) | def forward(ctx, max_xyz: torch.Tensor, xyz: torch.Tensor) -> torch.Te...
    method backward (line 217) | def backward(ctx, a=None):
  class FeatureGather (line 223) | class FeatureGather(Function):
    method forward (line 225) | def forward(ctx, max_feature: torch.Tensor, distribute_idx: torch.Tens...
    method backward (line 242) | def backward(ctx, grad_distribute_feature: torch.Tensor):
  class LabelStatBallRange (line 258) | class LabelStatBallRange(Function):
    method forward (line 260) | def forward(ctx, radius: float, xyz: torch.Tensor, new_xyz: torch.Tens...
    method backward (line 281) | def backward(ctx, a=None):
  class LabelStatIdx (line 287) | class LabelStatIdx(Function):
    method forward (line 289) | def forward(ctx, nsample: int, label_stat: torch.Tensor, idx: torch.Te...
    method backward (line 308) | def backward(ctx, a=None):
  class LabelStatAndBallQuery (line 314) | class LabelStatAndBallQuery(Function):
    method forward (line 316) | def forward(ctx, radius: float, nsample: int, xyz: torch.Tensor, new_x...
    method backward (line 340) | def backward(ctx, a=None, b=None):
  function pairwise_distances (line 346) | def pairwise_distances(x, y=None):
  class KNNQueryNaive (line 366) | class KNNQueryNaive(Function):
    method forward (line 368) | def forward(ctx, nsample: int, xyz: torch.Tensor, new_xyz: torch.Tenso...
    method backward (line 400) | def backward(ctx):
  class KNNQuery (line 406) | class KNNQuery(Function):
    method forward (line 408) | def forward(ctx, nsample: int, xyz: torch.Tensor, new_xyz: torch.Tenso...
    method backward (line 429) | def backward(ctx, a=None):
  class KNNQueryExclude (line 435) | class KNNQueryExclude(Function):
    method forward (line 437) | def forward(ctx, nsample: int, xyz: torch.Tensor, new_xyz: torch.Tenso...
    method backward (line 469) | def backward(ctx):
  class QueryAndGroup (line 475) | class QueryAndGroup(nn.Module):
    method __init__ (line 482) | def __init__(self, radius=None, nsample=32, use_xyz=True):
    method forward (line 486) | def forward(self, xyz: torch.Tensor, new_xyz: torch.Tensor = None, fea...
  class GroupAll (line 521) | class GroupAll(nn.Module):
    method __init__ (line 525) | def __init__(self, use_xyz: bool = True):
    method forward (line 529) | def forward(self, xyz: torch.Tensor, new_xyz: torch.Tensor, features: ...

FILE: lib/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: lib/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: lib/pointops/src/featuredistribute/featuredistribute_cuda.cpp
  function featuredistribute_cuda (line 15) | void featuredistribute_cuda(int b, int n, int m, at::Tensor max_xyz_tens...
  function featuregather_forward_cuda (line 30) | void featuregather_forward_cuda(int b, int n, int m, int c, at::Tensor m...
  function featuregather_backward_cuda (line 45) | void featuregather_backward_cuda(int b, int n, int m, int c, at::Tensor ...

FILE: lib/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: lib/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: lib/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: lib/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: lib/pointops/src/labelstat/labelstat_cuda.cpp
  function labelstat_idx_cuda_fast (line 14) | void labelstat_idx_cuda_fast(int b, int n, int m, int nsample, int nclass,
  function labelstat_ballrange_cuda_fast (line 29) | void labelstat_ballrange_cuda_fast(int b, int n, int m, float radius, in...
  function labelstat_and_ballquery_cuda_fast (line 46) | void labelstat_and_ballquery_cuda_fast(int b, int n, int m, float radius...

FILE: lib/pointops/src/pointops_api.cpp
  function PYBIND11_MODULE (line 15) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {

FILE: lib/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: lib/sync_bn/batchnorm.py
  function _sum_ft (line 24) | def _sum_ft(tensor):
  function _unsqueeze_ft (line 29) | def _unsqueeze_ft(tensor):
  class _SynchronizedBatchNorm (line 38) | class _SynchronizedBatchNorm(_BatchNorm):
    method __init__ (line 39) | def __init__(self, num_features, eps=1e-5, momentum=0.1, affine=True):
    method forward (line 48) | def forward(self, input):
    method __data_parallel_replicate__ (line 80) | def __data_parallel_replicate__(self, ctx, copy_id):
    method _data_parallel_master (line 90) | def _data_parallel_master(self, intermediates):
    method _compute_mean_std (line 113) | def _compute_mean_std(self, sum_, ssum, size):
  class SynchronizedBatchNorm1d (line 128) | class SynchronizedBatchNorm1d(_SynchronizedBatchNorm):
    method _check_input_dim (line 184) | def _check_input_dim(self, input):
  class SynchronizedBatchNorm2d (line 191) | class SynchronizedBatchNorm2d(_SynchronizedBatchNorm):
    method _check_input_dim (line 247) | def _check_input_dim(self, input):
  class SynchronizedBatchNorm3d (line 254) | class SynchronizedBatchNorm3d(_SynchronizedBatchNorm):
    method _check_input_dim (line 311) | def _check_input_dim(self, input):

FILE: lib/sync_bn/comm.py
  class FutureResult (line 18) | class FutureResult(object):
    method __init__ (line 21) | def __init__(self):
    method put (line 26) | def put(self, result):
    method get (line 32) | def get(self):
  class SlavePipe (line 46) | class SlavePipe(_SlavePipeBase):
    method run_slave (line 49) | def run_slave(self, msg):
  class SyncMaster (line 56) | class SyncMaster(object):
    method __init__ (line 67) | def __init__(self, master_callback):
    method __getstate__ (line 78) | def __getstate__(self):
    method __setstate__ (line 81) | def __setstate__(self, state):
    method register_slave (line 84) | def register_slave(self, identifier):
    method run_master (line 102) | def run_master(self, master_msg):
    method nr_slaves (line 136) | def nr_slaves(self):

FILE: lib/sync_bn/replicate.py
  class CallbackContext (line 23) | class CallbackContext(object):
  function execute_replication_callbacks (line 27) | def execute_replication_callbacks(modules):
  class DataParallelWithCallback (line 50) | class DataParallelWithCallback(DataParallel):
    method replicate (line 64) | def replicate(self, module, device_ids):
  function patch_replication_callback (line 70) | def patch_replication_callback(data_parallel):

FILE: lib/sync_bn/unittest.py
  function as_numpy (line 17) | def as_numpy(v):
  class TorchTestCase (line 23) | class TorchTestCase(unittest.TestCase):
    method assertTensorClose (line 24) | def assertTensorClose(self, a, b, atol=1e-3, rtol=1e-3):

FILE: model/pointnet/pointnet.py
  class STN3D (line 6) | class STN3D(nn.Module):
    method __init__ (line 7) | def __init__(self, c):
    method forward (line 24) | def forward(self, x):
  class PointNetFeat (line 43) | class PointNetFeat(nn.Module):
    method __init__ (line 44) | def __init__(self, c=3, global_feat=True):
    method forward (line 62) | def forward(self, x):
  class PointNetCls (line 81) | class PointNetCls(nn.Module):
    method __init__ (line 82) | def __init__(self, c=3, k=40, dropout=0.3, sync_bn=False):
    method forward (line 93) | def forward(self, x):
  class PointNetSeg (line 104) | class PointNetSeg(nn.Module):
    method __init__ (line 105) | def __init__(self, c=9, k=13, sync_bn=False):
    method forward (line 119) | def forward(self, x):

FILE: model/pointnet2/pointnet2_modules.py
  class _PointNet2SAModuleBase (line 11) | class _PointNet2SAModuleBase(nn.Module):
    method __init__ (line 12) | def __init__(self):
    method forward (line 18) | def forward(self, xyz: torch.Tensor, features: torch.Tensor = None) ->...
  class PointNet2SAModuleMSG (line 48) | class PointNet2SAModuleMSG(_PointNet2SAModuleBase):
    method __init__ (line 63) | def __init__(self, *, npoint: int, radii: List[float], nsamples: List[...
  class PointNet2SAModule (line 82) | class PointNet2SAModule(PointNet2SAModuleMSG):
    method __init__ (line 97) | def __init__(self, *, mlp: List[int], npoint: int = None, radius: floa...
  class PointNet2FPModule (line 101) | class PointNet2FPModule(nn.Module):
    method __init__ (line 110) | def __init__(self, *, mlp: List[int], bn: bool = True):
    method forward (line 114) | def forward(self, unknown: torch.Tensor, known: torch.Tensor, unknow_f...

FILE: model/pointnet2/pointnet2_seg.py
  class PointNet2SSGSeg (line 10) | class PointNet2SSGSeg(nn.Module):
    method __init__ (line 25) | def __init__(self, c=3, k=13, use_xyz=True):
    method _break_up_pc (line 39) | def _break_up_pc(self, pc):
    method forward (line 44) | def forward(self, pointcloud: torch.cuda.FloatTensor):
  class PointNet2MSGSeg (line 67) | class PointNet2MSGSeg(PointNet2SSGSeg):
    method __init__ (line 82) | def __init__(self, k, c=6, use_xyz=True):
  function model_fn_decorator (line 105) | def model_fn_decorator(criterion):

FILE: model/pointweb/pointweb_module.py
  class _AFAModule (line 11) | class _AFAModule(nn.Module):
    method __init__ (line 12) | def __init__(self, mlp, use_softmax=False):
    method forward (line 21) | def forward(self, feature: torch.Tensor) -> torch.Tensor:
  class _PointWebSAModuleBase (line 43) | class _PointWebSAModuleBase(nn.Module):
    method __init__ (line 44) | def __init__(self):
    method forward (line 51) | def forward(self, xyz: torch.Tensor, features: torch.Tensor = None) ->...
  class PointWebSAModule (line 81) | class PointWebSAModule(_PointWebSAModuleBase):
    method __init__ (line 96) | def __init__(self, *, npoint: int = None, nsample: int = None, mlp: Li...

FILE: model/pointweb/pointweb_seg.py
  class PointWebSeg (line 11) | class PointWebSeg(nn.Module):
    method __init__ (line 26) | def __init__(self, c=3, k=13, use_xyz=True):
    method _break_up_pc (line 41) | def _break_up_pc(self, pc):
    method forward (line 46) | def forward(self, pointcloud: torch.cuda.FloatTensor):
  function model_fn_decorator (line 68) | def model_fn_decorator(criterion):
  function model_fn_decorator (line 84) | def model_fn_decorator(criterion):

FILE: tool/test_s3dis.py
  function get_parser (line 22) | def get_parser():
  function get_logger (line 34) | def get_logger():
  function main (line 45) | def main():
  function data_prepare (line 77) | def data_prepare(room_path):
  function test (line 119) | def test(model, criterion, names):

FILE: tool/test_scannet.py
  function get_parser (line 22) | def get_parser():
  function get_logger (line 34) | def get_logger():
  function main (line 45) | def main():
  function data_prepare (line 77) | def data_prepare(points, labels):
  function test (line 116) | def test(model, criterion, names):

FILE: tool/train.py
  function get_parser (line 23) | def get_parser():
  function get_logger (line 35) | def get_logger():
  function worker_init_fn (line 46) | def worker_init_fn(worker_id):
  function init (line 50) | def init():
  function main (line 68) | def main():
  function train (line 154) | def train(train_loader, model, criterion, optimizer, epoch):
  function validate (line 220) | def validate(val_loader, model, criterion):

FILE: util/config.py
  class CfgNode (line 10) | class CfgNode(dict):
    method __init__ (line 16) | def __init__(self, init_dict=None, key_list=None, new_allowed=False):
    method __getattr__ (line 26) | def __getattr__(self, name):
    method __setattr__ (line 32) | def __setattr__(self, name, value):
    method __str__ (line 35) | def __str__(self):
    method __repr__ (line 56) | def __repr__(self):
  function load_cfg_from_cfg_file (line 60) | def load_cfg_from_cfg_file(file):
  function merge_cfg_from_list (line 76) | def merge_cfg_from_list(cfg, cfg_list):
  function _decode_cfg_value (line 91) | def _decode_cfg_value(v):
  function _check_and_coerce_cfg_value_type (line 120) | def _check_and_coerce_cfg_value_type(replacement, original, key, full_key):
  function _assert_with_logging (line 162) | def _assert_with_logging(cond, msg):

FILE: util/dataset.py
  function make_dataset (line 8) | def make_dataset(split='train', data_root=None, data_list=None):
  class PointData (line 19) | class PointData(Dataset):
    method __init__ (line 20) | def __init__(self, split='train', data_root=None, data_list=None, tran...
    method __len__ (line 28) | def __len__(self):
    method __getitem__ (line 31) | def __getitem__(self, index):

FILE: util/pt_util.py
  class SharedMLP (line 15) | class SharedMLP(nn.Sequential):
    method __init__ (line 17) | def __init__(
  class _BNBase (line 43) | class _BNBase(nn.Sequential):
    method __init__ (line 45) | def __init__(self, in_size, batch_norm=None, name=""):
  class BatchNorm1d (line 53) | class BatchNorm1d(_BNBase):
    method __init__ (line 55) | def __init__(self, in_size: int, *, name: str = ""):
  class BatchNorm2d (line 59) | class BatchNorm2d(_BNBase):
    method __init__ (line 61) | def __init__(self, in_size: int, name: str = ""):
  class BatchNorm3d (line 65) | class BatchNorm3d(_BNBase):
    method __init__ (line 67) | def __init__(self, in_size: int, name: str = ""):
  class _ConvBase (line 71) | class _ConvBase(nn.Sequential):
    method __init__ (line 73) | def __init__(
  class Conv1d (line 127) | class Conv1d(_ConvBase):
    method __init__ (line 129) | def __init__(
  class Conv2d (line 161) | class Conv2d(_ConvBase):
    method __init__ (line 163) | def __init__(
  class Conv3d (line 195) | class Conv3d(_ConvBase):
    method __init__ (line 197) | def __init__(
  class FC (line 229) | class FC(nn.Sequential):
    method __init__ (line 231) | def __init__(
  class _DropoutNoScaling (line 267) | class _DropoutNoScaling(InplaceFunction):
    method _make_noise (line 270) | def _make_noise(input):
    method symbolic (line 274) | def symbolic(g, input, p=0.5, train=False, inplace=False):
    method forward (line 286) | def forward(cls, ctx, input, p=0.5, train=False, inplace=False):
    method backward (line 314) | def backward(ctx, grad_output):
  class _FeatureDropoutNoScaling (line 324) | class _FeatureDropoutNoScaling(_DropoutNoScaling):
    method symbolic (line 327) | def symbolic(input, p=0.5, train=False, inplace=False):
    method _make_noise (line 331) | def _make_noise(input):
  function group_model_params (line 341) | def group_model_params(model: nn.Module, **kwargs):
  function checkpoint_state (line 359) | def checkpoint_state(
  function save_checkpoint (line 380) | def save_checkpoint(
  function load_checkpoint (line 389) | def load_checkpoint(model=None, optimizer=None, filename='checkpoint'):
  function variable_size_collate (line 408) | def variable_size_collate(pad_val=0, use_shared_memory=True):
  class TrainValSplitter (line 478) | class TrainValSplitter():
    method __init__ (line 491) | def __init__(
  function set_bn_momentum_default (line 576) | def set_bn_momentum_default(bn_momentum):
  class BNMomentumScheduler (line 585) | class BNMomentumScheduler(object):
    method __init__ (line 587) | def __init__(
    method step (line 605) | def step(self, epoch=None):
  class Trainer (line 613) | class Trainer(object):
    method __init__ (line 638) | def __init__(
    method _decode_value (line 661) | def _decode_value(v):
    method _train_it (line 680) | def _train_it(self, it, batch):
    method eval_epoch (line 697) | def eval_epoch(self, d_loader):
    method train (line 717) | def train(

FILE: util/s3dis.py
  class S3DIS (line 7) | class S3DIS(Dataset):
    method __init__ (line 8) | def __init__(self, split='train', data_root='trainval_fullarea', num_p...
    method __getitem__ (line 38) | def __getitem__(self, idx):
    method __len__ (line 72) | def __len__(self):
  function worker_init_fn (line 90) | def worker_init_fn(worker_id):

FILE: util/scannet.py
  class ScanNet (line 8) | class ScanNet(Dataset):
    method __init__ (line 9) | def __init__(self, split='train', data_root='scannet', num_point=8192,...
    method __getitem__ (line 47) | def __getitem__(self, idx):
    method __len__ (line 84) | def __len__(self):
  function worker_init_fn (line 96) | def worker_init_fn(worker_id):

FILE: util/transform.py
  class Compose (line 6) | class Compose(object):
    method __init__ (line 7) | def __init__(self, transforms):
    method __call__ (line 10) | def __call__(self, data, label):
  class ToTensor (line 16) | class ToTensor(object):
    method __call__ (line 17) | def __call__(self, data, label):
  class RandomRotate (line 27) | class RandomRotate(object):
    method __init__ (line 28) | def __init__(self, rotate_angle=None, along_z=False):
    method __call__ (line 32) | def __call__(self, data, label):
  class RandomRotatePerturbation (line 48) | class RandomRotatePerturbation(object):
    method __init__ (line 49) | def __init__(self, angle_sigma=0.06, angle_clip=0.18):
    method __call__ (line 53) | def __call__(self, data, label):
  class RandomScale (line 71) | class RandomScale(object):
    method __init__ (line 72) | def __init__(self, scale_low=0.8, scale_high=1.25):
    method __call__ (line 76) | def __call__(self, data, label):
  class RandomShift (line 82) | class RandomShift(object):
    method __init__ (line 83) | def __init__(self, shift_range=0.1):
    method __call__ (line 86) | def __call__(self, data, label):
  class RandomJitter (line 92) | class RandomJitter(object):
    method __init__ (line 93) | def __init__(self, sigma=0.01, clip=0.05):
    method __call__ (line 97) | def __call__(self, data, label):

FILE: util/util.py
  class AverageMeter (line 12) | class AverageMeter(object):
    method __init__ (line 14) | def __init__(self):
    method reset (line 17) | def reset(self):
    method update (line 23) | def update(self, val, n=1):
  function step_learning_rate (line 30) | def step_learning_rate(optimizer, base_lr, epoch, step_epoch, multiplier...
  function poly_learning_rate (line 37) | def poly_learning_rate(optimizer, base_lr, curr_iter, max_iter, power=0.9):
  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 check_mkdir (line 75) | def check_mkdir(dir_name):
  function check_makedirs (line 80) | def check_makedirs(dir_name):
  function init_weights (line 85) | def init_weights(model, conv='kaiming', batchnorm='normal', linear='kaim...
  function convert_to_syncbn (line 136) | def convert_to_syncbn(model):
  function colorize (line 152) | def colorize(gray, palette):