Functions — github.com/TEA-Lab/TwoByTwo

hub / github.com/TEA-Lab/TwoByTwo / functions

Functions212 in github.com/TEA-Lab/TwoByTwo

⨍ Functions 212◇ Types & classes 42

↓ 13 callersMethod__init__

(self, in_channels, out_channels)

src/shape_assembly/models/encoder/vn_layers.py:24

↓ 12 callersMethodbgs

(d6s)

src/shape_assembly/datasets/dataloader/dataloader_A.py:57

↓ 12 callersFunctionget_graph_feature

(x, k=20, idx=None, x_coord=None)

src/shape_assembly/models/encoder/vn_dgcnn.py:21

↓ 10 callersFunctionbgs

(d6s)

src/shape_assembly/models/train/network_vnn_A.py:30

↓ 9 callersMethod__init__

(self, layer, N)

src/shape_assembly/models/train/transformer.py:38

↓ 9 callersFunctionbgs

(d6s)

src/shape_assembly/utils.py:37

↓ 6 callersMethodload_data

(self)

src/shape_assembly/datasets/dataloader/dataloader_A.py:74

↓ 5 callersMethod__init__

(self, pc_feat_dim, out_dim)

src/shape_assembly/models/train/regressor_CR.py:9

↓ 5 callersFunctionclones

(module, N)

src/shape_assembly/models/train/transformer.py:8

↓ 4 callersFunctionbgs

(d6s)

src/shape_assembly/models/train/network_vnn_B.py:31

↓ 4 callersFunctionbgs

(d6s)

src/shape_assembly/models/train/network_vnn_A_indi.py:30

↓ 4 callersMethodforward_pass

(self, batch_data, device, mode, vis_idx=-1)

src/shape_assembly/models/train/network_vnn_A.py:453

↓ 4 callersFunctionwrite_matrix_to_csv

data_util/generate_pc.cpp:110

↓ 3 callersMethod__init__

(self, feat_dim)

src/shape_assembly/models/encoder/vn_dgcnn.py:53

↓ 3 callersMethodcompute_rot_loss

(self, batch_data, pred_data)

src/shape_assembly/models/train/network_vnn_A.py:345

↓ 3 callersMethodcompute_rot_loss

(self, batch_data, pred_data)

src/shape_assembly/models/train/network_vnn_B.py:311

↓ 3 callersMethodcompute_rot_loss

(self, batch_data, pred_data)

src/shape_assembly/models/train/network_vnn_A_indi.py:319

↓ 3 callersFunctionget_cfg_defaults

()

src/shape_assembly/config.py:75

↓ 3 callersFunctionknn

(x, k)

src/shape_assembly/models/encoder/vn_dgcnn_util.py:11

↓ 3 callersMethodrecover_R_from_6d

(self, R_6d)

src/shape_assembly/models/train/network_vnn_A.py:387

↓ 3 callersMethodrecover_R_from_6d

(self, R_6d)

src/shape_assembly/models/train/network_vnn_B.py:354

↓ 3 callersMethodrecover_R_from_6d

(self, R_6d)

src/shape_assembly/models/train/network_vnn_A_indi.py:361

↓ 2 callersMethodcheck_equiv

(self, x, R, xR, name)

src/shape_assembly/models/train/network_vnn_A.py:236

↓ 2 callersMethodcheck_equiv

(self, x, R, xR, name)

src/shape_assembly/models/train/network_vnn_B.py:216

↓ 2 callersMethodcheck_equiv

(self, x, R, xR, name)

src/shape_assembly/models/train/network_vnn_A_indi.py:209

↓ 2 callersFunctionfile_exists

(filepath)

data_util/preprocess_obj_blender.py:28

↓ 2 callersMethodforward

(self, src_pc, tgt_pc)

src/shape_assembly/models/train/network_vnn_A.py:274

↓ 2 callersMethodforward

(self, tgt_pc)

src/shape_assembly/models/train/network_vnn_B.py:254

↓ 2 callersMethodforward

(self, src_pc, tgt_pc)

src/shape_assembly/models/train/network_vnn_A_indi.py:247

↓ 2 callersMethodforward_pass

(self, batch_data, device, mode, vis_idx=-1)

src/shape_assembly/models/train/network_vnn_B.py:421

↓ 2 callersFunctionknn

(x, k)

src/shape_assembly/models/encoder/vn_dgcnn.py:13

↓ 2 callersFunctionsample_pc_with_blue_noise

data_util/generate_pc.cpp:19

↓ 2 callersMethodtraining_step

(self, batch_data, device, batch_idx)

src/shape_assembly/models/train/network_vnn_A.py:399

↓ 2 callersMethodtransform_pc_to_rot

(self, pcs)

src/shape_assembly/datasets/dataloader/dataloader_A.py:50

↓ 2 callersMethodtransform_pc_to_rot

(self, pcs)

src/shape_assembly/datasets/dataloader/dataloader_B.py:50

↓ 1 callersMethod__init__

(self, out_channels=(32, 64, 128), train_with_norm=True)

src/shape_assembly/models/train/pose_estimator.py:6

↓ 1 callersMethod_recon_pts

(self, Ga, Gb)

src/shape_assembly/models/train/network_vnn_A.py:269

↓ 1 callersMethod_recon_pts

(self, Ga, Gb)

src/shape_assembly/models/train/network_vnn_A_indi.py:242

↓ 1 callersFunctionattention

(query, key, value, mask=None)

src/shape_assembly/models/train/transformer.py:11

↓ 1 callersFunctionbgdR

(Rgts, Rps)

src/shape_assembly/utils.py:46

↓ 1 callersFunctioncalculate_center_and_scale_two_seperate_part

(instance_dir)

data_util/preprocess_obj_blender.py:80

↓ 1 callersMethodcalculate_metrics

(self, batch_data, pred_data, device, mode)

src/shape_assembly/models/train/network_vnn_A.py:409

↓ 1 callersMethodcalculate_metrics

(self, batch_data, pred_data, device, mode)

src/shape_assembly/models/train/network_vnn_B.py:379

↓ 1 callersMethodcalculate_metrics

(self, batch_data, pred_data, device, mode)

src/shape_assembly/models/train/network_vnn_A_indi.py:385

↓ 1 callersMethodcheck_inv

(self, x, R, xR, name)

src/shape_assembly/models/train/network_vnn_A.py:242

↓ 1 callersMethodcheck_inv

(self, x, R, xR, name)

src/shape_assembly/models/train/network_vnn_B.py:222

↓ 1 callersMethodcheck_inv

(self, x, R, xR, name)

src/shape_assembly/models/train/network_vnn_A_indi.py:215

↓ 1 callersMethodcheck_network_property

(self, gt_data, pred_data)

src/shape_assembly/models/train/network_vnn_A.py:248

↓ 1 callersMethodcheck_network_property

(self, gt_data, pred_data)

src/shape_assembly/models/train/network_vnn_A_indi.py:221

↓ 1 callersFunctioncleanup

()

src/script/our_eval.py:40

↓ 1 callersFunctioncleanup

()

src/script/our_train_A.py:44

↓ 1 callersFunctioncleanup

()

src/script/our_train_B.py:40

↓ 1 callersFunctioncombine_meshes

(instance_dir)

data_util/preprocess_obj_blender.py:41

↓ 1 callersMethodcompute_point_loss

(self, batch_data, pred_data)

src/shape_assembly/models/train/network_vnn_B.py:284

↓ 1 callersMethodcompute_recon_loss

(self, batch_data, pred_data)

src/shape_assembly/models/train/network_vnn_A.py:365

↓ 1 callersMethodcompute_recon_loss

(self, batch_data, pred_data)

src/shape_assembly/models/train/network_vnn_A_indi.py:339

↓ 1 callersMethodcompute_trans_loss

(self, batch_data, pred_data)

src/shape_assembly/models/train/network_vnn_A.py:334

↓ 1 callersMethodcompute_trans_loss

(self, batch_data, pred_data)

src/shape_assembly/models/train/network_vnn_B.py:303

↓ 1 callersMethodcompute_trans_loss

(self, batch_data, pred_data)

src/shape_assembly/models/train/network_vnn_A_indi.py:308

↓ 1 callersFunctionconv1x1

(in_channels, out_channels, dim)

src/shape_assembly/models/encoder/vn_layers.py:12

↓ 1 callersFunctionconvert_csv_to_ply

(input_csv, output_ply)

data_util/csv_to_ply.py:27

↓ 1 callersFunctiondebug_vis_gt

(batch_data, cfg, pred_data, iter_counts)

src/shape_assembly/models/train/network_vnn_A.py:128

↓ 1 callersFunctiondebug_vis_gt

(batch_data, cfg, pred_data, iter_counts)

src/shape_assembly/models/train/network_vnn_B.py:108

↓ 1 callersFunctiondebug_vis_gt

(batch_data, cfg, pred_data, iter_counts)

src/shape_assembly/models/train/network_vnn_A_indi.py:107

↓ 1 callersFunctiondebug_vis_input

(batch_data, cfg, prd_data, iter_counts)

src/shape_assembly/models/train/network_vnn_A.py:38

↓ 1 callersFunctiondebug_vis_input

(batch_data, cfg, prd_data, iter_counts)

src/shape_assembly/models/train/network_vnn_B.py:39

↓ 1 callersFunctiondebug_vis_input

(batch_data, cfg, prd_data, iter_counts)

src/shape_assembly/models/train/network_vnn_A_indi.py:38

↓ 1 callersFunctiondebug_vis_output

(batch_data, cfg, pred_data, iter_counts)

src/shape_assembly/models/train/network_vnn_A.py:77

↓ 1 callersFunctiondebug_vis_output

(batch_data, cfg, pred_data, iter_counts)

src/shape_assembly/models/train/network_vnn_B.py:66

↓ 1 callersFunctiondebug_vis_output

(batch_data, cfg, pred_data, iter_counts)

src/shape_assembly/models/train/network_vnn_A_indi.py:67

↓ 1 callersMethoddecode

(self, memory, src_mask, tgt, tgt_mask)

src/shape_assembly/models/train/transformer.py:31

↓ 1 callersMethodencode

(self, src, src_mask)

src/shape_assembly/models/train/transformer.py:28

↓ 1 callersMethodforward_pass

(self, batch_data, device, mode, vis_idx=-1)

src/shape_assembly/models/train/network_vnn_A_indi.py:429

↓ 1 callersMethodinit_decoder

(self)

src/shape_assembly/models/train/network_vnn_A.py:224

↓ 1 callersMethodinit_decoder

(self)

src/shape_assembly/models/train/network_vnn_B.py:201

↓ 1 callersMethodinit_decoder

(self)

src/shape_assembly/models/train/network_vnn_A_indi.py:197

↓ 1 callersMethodinit_encoder

(self)

src/shape_assembly/models/train/network_vnn_A.py:197

↓ 1 callersMethodinit_encoder

(self)

src/shape_assembly/models/train/network_vnn_B.py:173

↓ 1 callersMethodinit_encoder

(self)

src/shape_assembly/models/train/network_vnn_A_indi.py:170

↓ 1 callersMethodinit_pose_predictor_rot

(self)

src/shape_assembly/models/train/network_vnn_A.py:206

↓ 1 callersMethodinit_pose_predictor_rot

(self)

src/shape_assembly/models/train/network_vnn_B.py:182

↓ 1 callersMethodinit_pose_predictor_rot

(self)

src/shape_assembly/models/train/network_vnn_A_indi.py:179

↓ 1 callersMethodinit_pose_predictor_trans

(self)

src/shape_assembly/models/train/network_vnn_A.py:215

↓ 1 callersMethodinit_pose_predictor_trans

(self)

src/shape_assembly/models/train/network_vnn_B.py:192

↓ 1 callersMethodinit_pose_predictor_trans

(self)

src/shape_assembly/models/train/network_vnn_A_indi.py:188

↓ 1 callersMethodinit_transformer

(self)

src/shape_assembly/models/train/network_vnn_A.py:193

↓ 1 callersMethodinit_transformer

(self)

src/shape_assembly/models/train/network_vnn_A_indi.py:166

↓ 1 callersFunctionmain

()

data_util/blender_triangulate.py:20

↓ 1 callersFunctionmain

(cfg)

src/script/our_eval.py:171

↓ 1 callersFunctionmain

(cfg)

src/script/our_train_A.py:276

↓ 1 callersFunctionmain

(cfg)

src/script/our_train_B.py:271

↓ 1 callersFunctionread_csv_remove_duplicates

(input_csv)

data_util/csv_to_ply.py:5

↓ 1 callersFunctionsetup

(rank, world_size, cfg)

src/script/our_eval.py:34

↓ 1 callersFunctionsetup

(rank, world_size, cfg)

src/script/our_train_A.py:38

↓ 1 callersFunctionsetup

(rank, world_size, cfg)

src/script/our_train_B.py:34

↓ 1 callersFunctiontriangulate_mesh_with_pyblender

(instance_dir)

data_util/preprocess_obj_blender.py:33

↓ 1 callersFunctiontriangulate_object

(obj)

data_util/blender_triangulate.py:4

↓ 1 callersFunctionwrite_ply

(df, output_ply)

data_util/csv_to_ply.py:11

Method__getitem__

(self, index)

src/shape_assembly/datasets/dataloader/dataloader_A.py:109

Method__getitem__

(self, index)

src/shape_assembly/datasets/dataloader/dataloader_B.py:108

next →1–100 of 212, ranked by callers