SYMBOL INDEX (154 symbols across 17 files)

FILE: code/mesh_operations.py
  function read_off_ (line 12) | def read_off_(off_file_name):
  function write_off_ (line 15) | def write_off_(off_file_name,v,face_=numpy.zeros((1))):
  function write_obj_ (line 28) | def write_obj_(obj_write_name,v,face_=numpy.zeros((1)),color_=numpy.zero...
  function read_obj_ (line 45) | def read_obj_(obj_write_name):
  function write_xyz_ (line 50) | def write_xyz_(xyz_write_name,v,normal_=numpy.zeros((1))):
  function read_xyz_ (line 59) | def read_xyz_(xyz_name):
  function convert_obj_to_off_ (line 78) | def convert_obj_to_off_(obj_path_in, off_path_out):
  function normalize_points_to_sphere_ (line 84) | def normalize_points_to_sphere_(v_in):
  function normalize_points_to_sphere_with_given_center_and_factor_ (line 93) | def normalize_points_to_sphere_with_given_center_and_factor_(v_in, cente...

FILE: code/torch_tensor_functions.py
  function compute_sqrdis_map (line 14) | def compute_sqrdis_map(points_x, points_y):
  function draw_tensor_point_xyz_with_normal (line 28) | def draw_tensor_point_xyz_with_normal(save_path, torch_tensor_points, to...
  function draw_tensor_point_xyz_with_normal_by_threshold (line 43) | def draw_tensor_point_xyz_with_normal_by_threshold(save_path, torch_tens...
  function draw_tensor_point_obj_with_color (line 60) | def draw_tensor_point_obj_with_color(save_path, torch_tensor_points, tor...
  function draw_tensor_point_batch_xyz_with_normal (line 75) | def draw_tensor_point_batch_xyz_with_normal(save_batch_path, torch_tenso...
  function euler2rot (line 89) | def euler2rot(euler_angle):
  function get_neighbor_index (line 115) | def get_neighbor_index(vertices: "(bs, vertice_num, 3)",  neighbor_num: ...
  function indexing_neighbor (line 127) | def indexing_neighbor(tensor: "(bs, vertice_num, dim)", index: "(bs, que...
  function indexing_by_id (line 135) | def indexing_by_id(tensor: "(bs, vertice_num, dim)", index: "(bs, query_...

FILE: model/conpu_v6/chamfer_distance/chamfer_distance.cpp
  function chamfer_distance_forward_cuda (line 27) | void chamfer_distance_forward_cuda(
  function chamfer_distance_backward_cuda (line 46) | void chamfer_distance_backward_cuda(
  function nnsearch (line 64) | void nnsearch(
  function chamfer_distance_forward (line 95) | void chamfer_distance_forward(
  function chamfer_distance_backward (line 119) | void chamfer_distance_backward(
  function PYBIND11_MODULE (line 185) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {

FILE: model/conpu_v6/chamfer_distance/chamfer_distance.py
  class ChamferDistanceFunction (line 10) | class ChamferDistanceFunction(torch.autograd.Function):
    method forward (line 12) | def forward(ctx, xyz1, xyz2):
    method backward (line 36) | def backward(ctx, graddist1, graddist2):
  class ChamferDistance (line 55) | class ChamferDistance(torch.nn.Module):
    method forward (line 56) | def forward(self, xyz1, xyz2):

FILE: model/conpu_v6/loss.py
  class Loss (line 34) | class Loss(nn.Module):
    method __init__ (line 35) | def __init__(self, args):
    method loss_on_cd (line 39) | def loss_on_cd(self, deformation_p, p1):
    method loss_on_proj (line 46) | def loss_on_proj(self, p0, p1):
    method loss_on_normal (line 82) | def loss_on_normal(self, p0, p1, n0, n1):
    method loss_on_reg (line 108) | def loss_on_reg(self, gen_points_batch, train_points_sparse_batch):
    method loss_on_arap (line 122) | def loss_on_arap(self, gen_points_batch, uv_sampling_coors):
    method loss_on_overlap (line 143) | def loss_on_overlap(self, gen_points_batch, train_points_sparse_batch):
    method loss_on_ndirection (line 156) | def loss_on_ndirection(self, gen_points_batch, uv_sampling_coors, gen_...
    method forward (line 182) | def forward(self, gen_points_batch, gen_normals_batch, uv_sampling_coo...

FILE: model/conpu_v6/network.py
  class DGCNN_multi_knn_c5 (line 33) | class DGCNN_multi_knn_c5(nn.Module):
    method __init__ (line 34) | def __init__(self, emb_dims=512, args=None):
    method forward (line 52) | def forward(self, x, if_relu_atlast = False):
  function knn (line 77) | def knn(x, k):
  function get_graph_feature (line 84) | def get_graph_feature(x, k=4):
  class MLPNet_relu (line 102) | class MLPNet_relu(torch.nn.Module):
    method __init__ (line 107) | def __init__(self, nch_input, nch_layers, b_shared=True, bn_momentum=0...
    method forward (line 111) | def forward(self, inp):
  function mlp_layers_relu (line 115) | def mlp_layers_relu(nch_input, nch_layers, b_shared=True, bn_momentum=0....
  class MLPNet (line 144) | class MLPNet(torch.nn.Module):
    method __init__ (line 149) | def __init__(self, nch_input, nch_layers, b_shared=True, bn_momentum=0...
    method forward (line 153) | def forward(self, inp):
  function mlp_layers (line 157) | def mlp_layers(nch_input, nch_layers, b_shared=True, bn_momentum=0.1, dr...
  class Sine (line 182) | class Sine(nn.Module):
    method __init (line 183) | def __init(self):
    method forward (line 185) | def forward(self, input):
  class Net_conpu_v7 (line 192) | class Net_conpu_v7(nn.Module):
    method __init__ (line 193) | def __init__(self, args):
    method forward (line 213) | def forward(self, points_sparse):
    method project_3d_query_point_to_patches (line 268) | def project_3d_query_point_to_patches(self, querying_points_3d, queryi...
    method convert_uv_to_xyzn (line 341) | def convert_uv_to_xyzn(self, uv_coor, uv_coor_idx_in_sparse, sparse_em...
    method convert_uv_to_xyz (line 369) | def convert_uv_to_xyz(self, uv_coor, uv_coor_idx_in_sparse, sparse_emb...
    method convert_xyz_to_uv (line 386) | def convert_xyz_to_uv(self, xyz_coor, xyz_coor_idx_in_sparse, sparse_e...
  function convert_str_2_list (line 406) | def convert_str_2_list(str_):
  function position_encoding (line 411) | def position_encoding(input_uv, pe_out_L):
  function fix_sample (line 421) | def fix_sample(thisbatchsize, num_point, up_ratio, if_random=False):
  function uniform_random_sample (line 489) | def uniform_random_sample(thisbatchsize, num_point, up_ratio):
  function cal_grad (line 500) | def cal_grad(inputs, outputs):

FILE: model/conpu_v6/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, np...
  class PointnetSAModuleMSG (line 59) | class PointnetSAModuleMSG(_PointnetSAModuleBase):
    method __init__ (line 62) | def __init__(self, *, npoint: int, radii: List[float], nsamples: List[...
  class PointnetSAModule (line 99) | class PointnetSAModule(PointnetSAModuleMSG):
    method __init__ (line 102) | def __init__(self, *, mlp: List[int], npoint: int = None, radius: floa...
  class PointNetSSG_Base (line 120) | class PointNetSSG_Base(PointnetSAModuleMSG):
    method __init__ (line 121) | def __init__(self, npoint, nsample, radius, in_channel, out_channel, b...
  class PointnetFPModule (line 127) | class PointnetFPModule(nn.Module):
    method __init__ (line 130) | def __init__(self, *, mlp: List[int], bn: bool = True):
    method forward (line 138) | def forward(

FILE: model/conpu_v6/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 157) | class GroupingOperation(Function):
    method forward (line 160) | def forward(ctx, features: torch.Tensor, idx: torch.Tensor) -> torch.T...
    method backward (line 181) | def backward(ctx, grad_out: torch.Tensor) -> Tuple[torch.Tensor, torch...
  class BallQuery (line 201) | class BallQuery(Function):
    method forward (line 204) | def forward(ctx, radius: float, nsample: int, xyz: torch.Tensor, new_x...
    method backward (line 225) | def backward(ctx, a=None):
  class QueryAndGroup (line 232) | class QueryAndGroup(nn.Module):
    method __init__ (line 233) | def __init__(self, radius: float, nsample: int, use_xyz: bool = True):
    method forward (line 242) | def forward(self, xyz: torch.Tensor, new_xyz: torch.Tensor, features: ...

FILE: model/conpu_v6/pointnet2/pytorch_utils.py
  class EmptyModule (line 5) | class EmptyModule(nn.Module):
    method __init__ (line 6) | def __init__(self):
    method forward (line 9) | def forward(self, x):
  class SharedResMLP (line 13) | class SharedResMLP(nn.Module):
    method __init__ (line 14) | def __init__(
    method forward (line 37) | def forward(self, x):
  class SharedMLP (line 45) | class SharedMLP(nn.Sequential):
    method __init__ (line 47) | def __init__(
  class _ConvBase (line 74) | class _ConvBase(nn.Sequential):
    method __init__ (line 76) | def __init__(
  class _BNBase (line 143) | class _BNBase(nn.Sequential):
    method __init__ (line 145) | def __init__(self, in_size, batch_norm=None, name=""):
  class BatchNorm1d (line 153) | class BatchNorm1d(_BNBase):
    method __init__ (line 155) | def __init__(self, in_size: int, *, name: str = ""):
  class BatchNorm2d (line 159) | class BatchNorm2d(_BNBase):
    method __init__ (line 161) | def __init__(self, in_size: int, name: str = ""):
  class BatchNorm3d (line 165) | class BatchNorm3d(_BNBase):
    method __init__ (line 167) | def __init__(self, in_size: int, name: str = ""):
  class Conv1d (line 171) | class Conv1d(_ConvBase):
    method __init__ (line 173) | def __init__(
  class Conv2d (line 208) | class Conv2d(_ConvBase):
    method __init__ (line 210) | def __init__(
  class Conv3d (line 244) | class Conv3d(_ConvBase):
    method __init__ (line 246) | def __init__(
  class FC (line 281) | class FC(nn.Sequential):
    method __init__ (line 283) | def __init__(

FILE: model/conpu_v6/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: model/conpu_v6/pointnet2/src/cuda_utils.h
  function opt_n_threads (line 10) | inline int opt_n_threads(int work_size) {

FILE: model/conpu_v6/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: model/conpu_v6/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: model/conpu_v6/pointnet2/src/pointnet2_api.cpp
  function PYBIND11_MODULE (line 10) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {

FILE: model/conpu_v6/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: model/conpu_v6/train_view_toy.py
  class GradScaler (line 56) | class GradScaler:
    method __init__ (line 57) | def __init__(self):
    method scale (line 59) | def scale(self, loss):
    method unscale_ (line 61) | def unscale_(self, optimizer):
    method step (line 63) | def step(self, optimizer):
    method update (line 65) | def update(self):
  function update_test_cache (line 148) | def update_test_cache(used_samples_num, model, loss_obj, args):
  function update_pics (line 166) | def update_pics():
  function update_visualization (line 265) | def update_visualization(model,  args):
  function stophere (line 376) | def stophere():
  function run_train_val (line 380) | def run_train_val(model, optimizer, loss_obj,  args):
  function train_one_batch (line 425) | def train_one_batch(model, optimizer, loss_obj, start_pos, end_pos, args):
  function test_one_batch (line 524) | def test_one_batch(model, loss_obj, start_pos, end_pos, args):
  function compute_test_loss_values (line 588) | def compute_test_loss_values(model, loss_obj, args):
  function show_parameter_by_name (line 615) | def show_parameter_by_name(net_name, layer_name):
  function get_para_of_one_layer_from_another_net (line 621) | def get_para_of_one_layer_from_another_net(net_source, net_to_be_changed...

FILE: utils/config.py
  function parse_args (line 5) | def parse_args():