Functions — github.com/TEA-Lab/TwoByTwo

CodeHub MCPcopy Create free account

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

Functions212 in github.com/TEA-Lab/TwoByTwo

⨍ Functions 212◇ Types & classes 42

(self, data_root_dir, data_csv_file, data_features=[], num_points = 1024 ,num_query_points = 1024 ,data_per_se

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

(self, data_root_dir, data_csv_file, data_features=[], num_points = 1024 ,num_query_points = 1024 ,data_per_se

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

(self, in_channels, share_nonlinearity=False, negative_slope=0.2)

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

(self, in_channels, share_nonlinearity=False, negative_slope=0.2)

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

(self, in_channels, out_channels, dim=5, share_nonlinearity=False, negative_slope=0.2)

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

(self, in_channels, out_channels, dim=5, share_nonlinearity=False, use_batchnorm='norm', nega

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

(self, num_features, dim)

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

(self, in_channels)

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

(self, in_channels, dim=4, normalize_frame=False, share_nonlinearity=False, negative_slope=0.2)

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

( self, in_channels, dim=4, share_nonlinearity=False,

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

( self, in_channels, out_channels, dim=5, share_no

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

(self, in_channels, out_channels, dim=2, use_batchnorm=True, negative_slope=0.2)

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

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

(self, in_channels, dim=4, normalize_frame=False, negative_slope=0.2)

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

(self, in_channels, out_channels, dim=4, use_batchnorm=True, negative_slope=0.2)

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

(self, feat_dim)

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

(self, feat_dim)

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

(self, feat_dim)

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

(self, feat_dim, num_points)

src/shape_assembly/models/decoder/MLPDecoder.py:13

(self, cfg, data_features)

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

(self, pc_feat_dim, out_dim)

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

(self, pc_feat_dim, out_dim)

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

(self, pc_feat_dim)

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

(self, pc_feat_dim, out_dim)

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

(self, pc_feat_dim)

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

(self, cfg, data_features)

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

(self, cfg, data_features)

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

(self, encoder, decoder, src_embed, tgt_embed, generator)

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

(self, layer, N)

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

(self, features, eps=1e-6)

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

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

(self, size, self_attn, feed_forward)

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

(self, size, self_attn, src_attn, feed_forward)

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

(self, h, d_model)

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

(self, d_model, d_ff)

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

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

(self, pointnet_out_dim=128, pose_dim=6, hidden_dims=(512, 256, 128))

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

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

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

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

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

Method_recon_pts

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

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

Methodcheck_network_property

(self, gt_data, pred_data)

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

Functioncompute_distance_between_rotations

src/shape_assembly/utils.py:28

Methodcompute_point_loss

(self, batch_data, pred_data)

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

Methodcompute_point_loss

(self, batch_data, pred_data)

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

Methodcompute_recon_loss

(self, batch_data, pred_data)

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

Methodcompute_rot_loss_symmetry

(self, batch_data, pred_data, device)

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

Methodcompute_rot_loss_symmetry

(self, batch_data, pred_data, device)

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

Methodcompute_rot_loss_symmetry

(self, batch_data, pred_data, device)

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

Methodconfigure_optimizers

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

Methodconfigure_optimizers

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

Methodconfigure_optimizers

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

x: point features of shape [B, N_feat, 3, N_samples, ...]

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

x: point features of shape [B, N_feat, 3, N_samples, ...]

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

x: point features of shape [B, N_feat, 3, N_samples, ...]

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

x: point features of shape [B, N_feat, 3, N_samples, ...]

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

x: point features of shape [B, N_feat, 3, N_samples, ...]

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

x: point features of shape [B, N_feat, 3, N_samples, ...]

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

x: point features of shape [B, N_feat, 3, N_samples, ...]

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

x: point features of shape [B, N_feat, 3, N_samples, ...]

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

Args: x: point features of shape [B, C, 3, N, ...] Returns: rotation invariant features of the same shape

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

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

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

x: point features of shape [B, C, N]

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

x: point features of shape [B, C_in, N]

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

x: point features of shape [B, C_in, N]

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

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

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

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

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

src/shape_assembly/models/decoder/MLPDecoder.py:23

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

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

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

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

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

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

(self, src, tgt, src_mask, tgt_mask)

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

(self, x, mask)

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

(self, x, memory, src_mask, tgt_mask)

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

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

(self, x, sublayer)

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

(self, x, mask)

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

(self, x, memory, src_mask, tgt_mask)

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

(self, query, key, value, mask=None)

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

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

(self, src, tgt)

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

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

(self, point_cloud, poses)

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

Functionget_6d_rot_loss

(gt_6d, pred_6d)

src/shape_assembly/utils.py:54

Functionget_6d_rot_loss_symmetry

(gt_6d, pred_6d, symmetry, device)

src/shape_assembly/utils.py:144

Functionget_6d_rot_loss_symmetry_new

(batch_data, pred_data, device)

src/shape_assembly/utils.py:63

Functionget_cfg_defaults

src/shape_assembly/config_eval.py:42

Functionget_graph_feature

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

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

Functionget_graph_feature_cross

(x, k=20, idx=None)

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

Functionload_data

(file_dir, cat_shape_dict)

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

Functionload_data

(file_dir, cat_shape_dict)

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

Methodload_data

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

← previous next →101–200 of 212, ranked by callers