SYMBOL INDEX (107 symbols across 13 files)

FILE: pointnet2/pointnet2_modules.py
  class _PointnetSAModuleBase (line 10) | class _PointnetSAModuleBase(nn.Module):
    method __init__ (line 12) | def __init__(self):
    method forward (line 19) | def forward(self, xyz: torch.Tensor, features: torch.Tensor = None, ne...
  class PointnetSAModuleMSG (line 58) | class PointnetSAModuleMSG(_PointnetSAModuleBase):
    method __init__ (line 61) | def __init__(self, *, npoint: int, radii: List[float], nsamples: List[...
  class PointnetSAModule (line 95) | class PointnetSAModule(PointnetSAModuleMSG):
    method __init__ (line 98) | def __init__(self, *, mlp: List[int], npoint: int = None, radius: floa...
  class PointnetFPModule (line 116) | class PointnetFPModule(nn.Module):
    method __init__ (line 119) | def __init__(self, *, mlp: List[int], bn: bool = True):
    method forward (line 127) | def forward(

FILE: pointnet2/pointnet2_utils.py
  class FurthestPointSampling (line 10) | class FurthestPointSampling(Function):
    method forward (line 12) | def forward(ctx, xyz: torch.Tensor, npoint: int) -> torch.Tensor:
    method backward (line 32) | def backward(xyz, a=None):
  class GatherOperation (line 39) | class GatherOperation(Function):
    method forward (line 42) | def forward(ctx, features: torch.Tensor, idx: torch.Tensor) -> torch.T...
    method backward (line 63) | def backward(ctx, grad_out):
  class ThreeNN (line 76) | class ThreeNN(Function):
    method forward (line 79) | def forward(ctx, unknown: torch.Tensor, known: torch.Tensor) -> Tuple[...
    method backward (line 101) | def backward(ctx, a=None, b=None):
  class ThreeInterpolate (line 108) | class ThreeInterpolate(Function):
    method forward (line 111) | 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 GroupingOperation (line 156) | class GroupingOperation(Function):
    method forward (line 159) | def forward(ctx, features: torch.Tensor, idx: torch.Tensor) -> torch.T...
    method backward (line 180) | def backward(ctx, grad_out: torch.Tensor) -> Tuple[torch.Tensor, torch...
  class BallQuery (line 200) | class BallQuery(Function):
    method forward (line 203) | def forward(ctx, radius: float, nsample: int, xyz: torch.Tensor, new_x...
    method backward (line 224) | def backward(ctx, a=None):
  class QueryAndGroup (line 231) | class QueryAndGroup(nn.Module):
    method __init__ (line 232) | def __init__(self, radius: float, nsample: int, use_xyz: bool = True):
    method forward (line 241) | def forward(self, xyz: torch.Tensor, new_xyz: torch.Tensor, features: ...
  class GroupAll (line 267) | class GroupAll(nn.Module):
    method __init__ (line 268) | def __init__(self, use_xyz: bool = True):
    method forward (line 272) | def forward(self, xyz: torch.Tensor, new_xyz: torch.Tensor, features: ...

FILE: pointnet2/pytorch_utils.py
  class SharedMLP (line 5) | class SharedMLP(nn.Sequential):
    method __init__ (line 7) | def __init__(
  class _ConvBase (line 35) | class _ConvBase(nn.Sequential):
    method __init__ (line 37) | def __init__(
  class _BNBase (line 104) | class _BNBase(nn.Sequential):
    method __init__ (line 106) | def __init__(self, in_size, batch_norm=None, name=""):
  class BatchNorm1d (line 114) | class BatchNorm1d(_BNBase):
    method __init__ (line 116) | def __init__(self, in_size: int, *, name: str = ""):
  class BatchNorm2d (line 120) | class BatchNorm2d(_BNBase):
    method __init__ (line 122) | def __init__(self, in_size: int, name: str = ""):
  class Conv1d (line 126) | class Conv1d(_ConvBase):
    method __init__ (line 128) | def __init__(
  class Conv2d (line 163) | class Conv2d(_ConvBase):
    method __init__ (line 165) | def __init__(
  class FC (line 200) | class FC(nn.Sequential):
    method __init__ (line 202) | def __init__(

FILE: pointnet2/src/ball_query.cpp
  function ball_query_wrapper_fast (line 14) | int ball_query_wrapper_fast(int b, int n, int m, float radius, int nsample,

FILE: pointnet2/src/cuda_utils.h
  function opt_n_threads (line 10) | inline int opt_n_threads(int work_size) {

FILE: pointnet2/src/group_points.cpp
  function group_points_grad_wrapper_fast (line 11) | int group_points_grad_wrapper_fast(int b, int c, int n, int npoints, int...
  function group_points_wrapper_fast (line 25) | int group_points_wrapper_fast(int b, int c, int n, int npoints, int nsam...

FILE: pointnet2/src/interpolate.cpp
  function three_nn_wrapper_fast (line 14) | void three_nn_wrapper_fast(int b, int n, int m, at::Tensor unknown_tensor,
  function three_interpolate_wrapper_fast (line 26) | void three_interpolate_wrapper_fast(int b, int c, int m, int n,
  function three_interpolate_grad_wrapper_fast (line 41) | void three_interpolate_grad_wrapper_fast(int b, int c, int n, int m,

FILE: pointnet2/src/pointnet2_api.cpp
  function PYBIND11_MODULE (line 10) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {

FILE: pointnet2/src/sampling.cpp
  function gather_points_wrapper_fast (line 11) | int gather_points_wrapper_fast(int b, int c, int n, int npoints,
  function gather_points_grad_wrapper_fast (line 23) | int gather_points_grad_wrapper_fast(int b, int c, int n, int npoints,
  function furthest_point_sampling_wrapper (line 36) | int furthest_point_sampling_wrapper(int b, int n, int m,

FILE: tools/dataset.py
  class KittiDataset (line 12) | class KittiDataset(torch_data.Dataset):
    method __init__ (line 13) | def __init__(self, root_dir, split='train', mode='TRAIN'):
    method get_image (line 33) | def get_image(self, idx):
    method get_image_shape (line 38) | def get_image_shape(self, idx):
    method get_lidar (line 45) | def get_lidar(self, idx):
    method get_calib (line 50) | def get_calib(self, idx):
    method get_label (line 55) | def get_label(self, idx):
    method get_valid_flag (line 61) | def get_valid_flag(pts_rect, pts_img, pts_rect_depth, img_shape):
    method filtrate_objects (line 68) | def filtrate_objects(self, obj_list):
    method __len__ (line 83) | def __len__(self):
    method __getitem__ (line 86) | def __getitem__(self, index):
    method generate_training_labels (line 155) | def generate_training_labels(pts_rect, gt_boxes3d):
    method collate_batch (line 173) | def collate_batch(self, batch):

FILE: tools/kitti_utils.py
  function cls_type_to_id (line 6) | def cls_type_to_id(cls_type):
  class Object3d (line 13) | class Object3d(object):
    method __init__ (line 14) | def __init__(self, line):
    method get_obj_level (line 33) | def get_obj_level(self):
    method generate_corners3d (line 49) | def generate_corners3d(self):
    method to_str (line 67) | def to_str(self):
    method to_kitti_format (line 73) | def to_kitti_format(self):
  function get_calib_from_file (line 81) | def get_calib_from_file(calib_file):
  class Calibration (line 100) | class Calibration(object):
    method __init__ (line 101) | def __init__(self, calib_file):
    method cart_to_hom (line 111) | def cart_to_hom(self, pts):
    method lidar_to_rect (line 119) | def lidar_to_rect(self, pts_lidar):
    method rect_to_img (line 128) | def rect_to_img(self, pts_rect):
    method lidar_to_img (line 139) | def lidar_to_img(self, pts_lidar):
  function get_objects_from_label (line 149) | def get_objects_from_label(label_file):
  function objs_to_boxes3d (line 156) | def objs_to_boxes3d(obj_list):
  function boxes3d_to_corners3d (line 164) | def boxes3d_to_corners3d(boxes3d, rotate=True):
  function enlarge_box3d (line 202) | def enlarge_box3d(boxes3d, extra_width):
  function in_hull (line 215) | def in_hull(p, hull):

FILE: tools/pointnet2_msg.py
  function get_model (line 7) | def get_model(input_channels=0):
  class Pointnet2MSG (line 21) | class Pointnet2MSG(nn.Module):
    method __init__ (line 22) | def __init__(self, input_channels=6):
    method _break_up_pc (line 66) | def _break_up_pc(self, pc):
    method forward (line 75) | def forward(self, pointcloud: torch.cuda.FloatTensor):

FILE: tools/train_and_eval.py
  function log_print (line 39) | def log_print(info, log_f=None):
  class DiceLoss (line 45) | class DiceLoss(nn.Module):
    method __init__ (line 46) | def __init__(self, ignore_target=-1):
    method forward (line 50) | def forward(self, input, target):
  function train_one_epoch (line 62) | def train_one_epoch(model, train_loader, optimizer, epoch, lr_scheduler,...
  function eval_one_epoch (line 102) | def eval_one_epoch(model, eval_loader, epoch, tb_log=None, log_f=None):
  function save_checkpoint (line 133) | def save_checkpoint(model, epoch, ckpt_name):
  function load_checkpoint (line 144) | def load_checkpoint(model, filename):
  function train_and_eval (line 157) | def train_and_eval(model, train_loader, eval_loader, tb_log, ckpt_dir, l...