SYMBOL INDEX (161 symbols across 9 files) FILE: cage_deformer_3d.py function test (line 24) | def test(net=None, save_subdir="test"): function train (line 155) | def train(): FILE: common.py function is_type (line 13) | def is_type(file, file_ext): function find_files (line 19) | def find_files(source, file_ext=["txt",]): function loadInitCage (line 40) | def loadInitCage(templates): function renderMeshes (line 51) | def renderMeshes(shape_dir, forward=(0.5,0.5,0), pos=(-4,-4,0), up=(0,0,... function call_proc (line 122) | def call_proc(cmd): function deform_with_MVC (line 129) | def deform_with_MVC(cage, cage_deformed, cage_face, query, verbose=False): function load_shapenet_cat (line 153) | def load_shapenet_cat(): function build_dataset (line 164) | def build_dataset(opt): function log_outputs (line 254) | def log_outputs(opt, step, all_outputs, all_inputs): function remesh (line 283) | def remesh(path1): function read_trimesh (line 299) | def read_trimesh(path, normal=False, clean=True): function crisscross_input (line 316) | def crisscross_input(data): FILE: datasets.py function _numpy_chamfer (line 27) | def _numpy_chamfer(P1, P2): class FileListDataset (line 49) | class FileListDataset(torch.utils.data.Dataset): method __init__ (line 50) | def __init__(self, opt, normalization_function=lambda x: pc_utils.norm... method __len__ (line 64) | def __len__(self): method __getitem__ (line 67) | def __getitem__(self,idx): method uncollate (line 125) | def uncollate(batch_data): class CoSegDataset (line 152) | class CoSegDataset(torch.utils.data.Dataset): method __init__ (line 153) | def __init__(self, root_dir="/home/mnt/points/data/Coseg_Wang/Coseg_Wa... method normalize (line 193) | def normalize(shape): method __getitem__ (line 197) | def __getitem__(self, idx): method uncollate (line 287) | def uncollate(batch_data): method __len__ (line 303) | def __len__(self): method render_result (line 310) | def render_result(shape_dir, **kwargs): function _unwrap_self (line 317) | def _unwrap_self(arg, **kwarg): class ShapeNetSeg (line 320) | class ShapeNetSeg(torch.utils.data.Dataset): method __init__ (line 321) | def __init__(self, root_dir="/home/mnt/points/data/ShapeNet/PartSeg_v0... method _uniformize_sizes (line 463) | def _uniformize_sizes(knn): method compute_nearest_neighbors_graph (line 487) | def compute_nearest_neighbors_graph(self): method _getitem (line 503) | def _getitem(self, index): method preprocess (line 516) | def preprocess(self): method generate_parts_by_cat (line 557) | def generate_parts_by_cat(self): method normalize (line 569) | def normalize(x, isV2=False): method getAnItem (line 582) | def getAnItem(self, index): method __getitem__ (line 644) | def __getitem__(self, index): method uncollate (line 670) | def uncollate(batch_data): method __len__ (line 688) | def __len__(self): method render_result (line 700) | def render_result(shape_dir, **kwargs): class ShapeNetV2 (line 709) | class ShapeNetV2(ShapeNetSeg): method __init__ (line 710) | def __init__(self, root_dir="/home/mnt/points/data/ShapeNet/ShapeNetCo... method normalize (line 858) | def normalize(x): method getAnItem (line 862) | def getAnItem(self, index): method __getitem__ (line 903) | def __getitem__(self, index): method render_result (line 928) | def render_result(shape_dir, **kwargs): class PairedSurreal (line 935) | class PairedSurreal(torch.utils.data.Dataset): method __init__ (line 936) | def __init__(self, root_dir, phase="train", npoints=6890, regular_samp... method getPairIdx (line 995) | def getPairIdx(self, index): method __getitem__ (line 1000) | def __getitem__(self, index): method uncollate (line 1036) | def uncollate(batch_data): method __len__ (line 1059) | def __len__(self): method render_result (line 1063) | def render_result(shape_dir, **kwargs): FILE: deformer_3d.py function test (line 23) | def test(net=None, subdir="test"): function test_all (line 108) | def test_all(net=None, subdir="test"): function train (line 186) | def train(): FILE: losses.py class AllLosses (line 15) | class AllLosses(torch.nn.Module): method __init__ (line 16) | def __init__(self, opt): method forward (line 51) | def forward(self, all_inputs, all_outputs, progress=1.0): class FaceNormalLoss (line 193) | class FaceNormalLoss(torch.nn.Module): method __init__ (line 194) | def __init__(self, n_faces=100): method forward (line 199) | def forward(self, ref_mesh_V, mesh_V, mesh_F): class GroundingLoss (line 208) | class GroundingLoss(torch.nn.Module): method __init__ (line 209) | def __init__(self, up_dim=1): method forward (line 213) | def forward(self, source, deformed): class ExtPointToNearestFaceDistance (line 229) | class ExtPointToNearestFaceDistance(torch.nn.Module): method __init__ (line 233) | def __init__(self, min_dist=0.1, reduction="mean"): method forward (line 238) | def forward(self, mesh_V, mesh_F, points, exterior_flag, mesh_FN=None): class MVCRegularizer (line 288) | class MVCRegularizer(torch.nn.Module): method __init__ (line 293) | def __init__(self, alpha=1.0, beta=1.0, threshold=5.0): method forward (line 299) | def forward(self, weights): class LabeledChamferDistance (line 314) | class LabeledChamferDistance(torch.nn.Module): method __init__ (line 323) | def __init__(self, beta=1.0, gamma=1, delta=0): method forward (line 329) | def forward(self, xyz1, xyz2, label1=None, label2=None): class SymmetryLoss (line 340) | class SymmetryLoss(torch.nn.Module): method __init__ (line 350) | def __init__(self, sym_plane=("yz",), NCHW=True): method get_mirror_multiplier (line 369) | def get_mirror_multiplier(self, dim_id): method forward (line 377) | def forward(self, xyz): class ConditionNumberLoss (line 385) | class ConditionNumberLoss(torch.nn.Module): method __init__ (line 393) | def __init__(self, ball_size, metric, reduction="mean"): method forward (line 400) | def forward(self, ref_points, points, *args, **kwargs): class InsideLoss2D (line 432) | class InsideLoss2D(torch.nn.Module): method __init__ (line 433) | def __init__(self, reduction="mean"): method forward (line 437) | def forward(self, cage, shape, shape_normals, epsilon=0.01, interpolat... class InterpolatedCDTriMesh (line 478) | class InterpolatedCDTriMesh(torch.nn.Module): method __init__ (line 483) | def __init__(self, interpolate_n=4, beta=1.0, gamma=1, delta=0): method forward (line 501) | def forward(self, cage_v, cage_f, shape, interpolate=True): class InsideLoss3DTriMesh (line 520) | class InsideLoss3DTriMesh(torch.nn.Module): method __init__ (line 529) | def __init__(self, reduction="mean", interpolate_n=4): method forward (line 542) | def forward(self, cage_v, cage_f, shape, shape_vn, epsilon=0.01, inter... class MeshDihedralAngleLoss (line 583) | class MeshDihedralAngleLoss(torch.nn.Module): method __init__ (line 591) | def __init__(self, threshold=np.pi/6, edge_points=None, reduction="mea... method forward (line 597) | def forward(self, vert1, vert2=None, edge_points=None): class GTNormalLoss (line 621) | class GTNormalLoss(torch.nn.Module): method __init__ (line 629) | def __init__(self, nn_size=10, NCHW=True): method forward (line 635) | def forward(self, pred, gt_normals): class MeshSmoothLoss (line 641) | class MeshSmoothLoss(torch.nn.Module): method __init__ (line 649) | def __init__(self, metric, use_cot=False, use_norm=False): method forward (line 657) | def forward(self, vert1, face=None): class LocalFeatureLoss (line 662) | class LocalFeatureLoss(torch.nn.Module): method __init__ (line 670) | def __init__(self, nn_size=10, metric=torch.nn.MSELoss("mean"), **kwar... method forward (line 675) | def forward(self, xyz1, xyz2, **kwargs): FILE: network2.py class DeformationSharedMLP (line 19) | class DeformationSharedMLP(nn.Module): method __init__ (line 22) | def __init__(self,dim=3, residual=True, normalization="none"): method forward (line 31) | def forward(self, x): class MLPDeformer (line 40) | class MLPDeformer(nn.Module): method __init__ (line 41) | def __init__(self, dim, bottleneck_size, npoint, residual=True, normal... method forward (line 52) | def forward(self, code, template): class FixedSourceDeformer (line 64) | class FixedSourceDeformer(torch.nn.Module): method __init__ (line 65) | def __init__(self, opt, dim, num_points, bottleneck_size, method initialize_buffers (line 103) | def initialize_buffers(self, template_vertices=None, template_faces=No... method forward (line 134) | def forward(self, target_shape, sample_idx=None, alpha=1.0, cage_only=... FILE: networks.py class STN (line 28) | class STN(nn.Module): method __init__ (line 29) | def __init__(self, num_points = 2500, dim=3): method forward (line 41) | def forward(self, x): class PointNet2feat (line 61) | class PointNet2feat(nn.Module): method __init__ (line 66) | def __init__(self, dim=3, num_points=2048, num_levels=3, bottleneck_si... method _break_up_pc (line 101) | def _break_up_pc(self, pc): method forward (line 110) | def forward(self, pointcloud: torch.cuda.FloatTensor, return_all=False): class PointNetfeat3DCoded (line 135) | class PointNetfeat3DCoded(nn.Module): method __init__ (line 136) | def __init__(self, npoint=2500, nlatent=1024): method forward (line 155) | def forward(self, x): class UnetCageGen (line 167) | class UnetCageGen(nn.Module): method __init__ (line 179) | def __init__(self, bottleneck_size, dim=3, knn_k=3, method interpolate_features (line 187) | def interpolate_features(self, query, points, feats, q_normals=None, p... method forward (line 225) | def forward(self, template, l_xyz, l_features, return_aux=False): class UnetDeformGen (line 249) | class UnetDeformGen(UnetCageGen): method interpolate_features (line 260) | def interpolate_features(self, query_feats, feats, points): method forward (line 292) | def forward(self, template, template_features, l_xyz, l_features, retu... class PointNetfeat (line 317) | class PointNetfeat(nn.Module): method __init__ (line 318) | def __init__(self, dim=3, num_points=2500, global_feat=True, trans=Fal... method forward (line 331) | def forward(self, x): class PointGenCon (line 352) | class PointGenCon(nn.Module): method __init__ (line 353) | def __init__(self, bottleneck_size, out_dim, prim_dim, normalization=N... method forward (line 373) | def forward(self, x, primative): class MultiFoldPointGen (line 386) | class MultiFoldPointGen(nn.Module): method __init__ (line 396) | def __init__(self, bottleneck_size, out_dim=3, prim_dim=3, method forward (line 410) | def forward(self, code, primative): class MLPDeformer (line 426) | class MLPDeformer(nn.Module): method __init__ (line 427) | def __init__(self, dim, bottleneck_size, npoint, residual=True, normal... method forward (line 437) | def forward(self, code, template): class NetworkFull (line 450) | class NetworkFull(nn.Module): method __init__ (line 451) | def __init__(self, opt, dim, bottleneck_size, method set_up_template (line 500) | def set_up_template(self, template_vertices, template_faces): method forward (line 512) | def forward(self, source_shape, target_shape, alpha=1.0): FILE: optimize_cage.py class MyOptions (line 30) | class MyOptions(DeformationOptions): method initialize (line 31) | def initialize(self, parser): method parse (line 40) | def parse(self): function visualize_correspondence (line 48) | def visualize_correspondence(opt, source_shape, source_face, target_shap... function optimize (line 75) | def optimize(opt): function test_one (line 272) | def test_one(opt, cage_shape, new_source, new_source_face, new_target, n... function test_all (line 305) | def test_all(opt, new_cage_shape): FILE: option.py class BaseOptions (line 8) | class BaseOptions(): method __init__ (line 14) | def __init__(self): method initialize (line 18) | def initialize(self, parser): method gather_options (line 97) | def gather_options(self): method print_options (line 115) | def print_options(self, opt, output_file=None): method parse (line 143) | def parse(self): class DeformationOptions (line 185) | class DeformationOptions(BaseOptions): method parse (line 189) | def parse(self):