MCPcopy Create free account

hub / github.com/MengyunQ/MeshHeart / functions

Functions55 in github.com/MengyunQ/MeshHeart

↓ 7 callersMethod__init__
(self, input_dim, out_dim, fc_dim, n_fc, weight_norm=False, activation='relu', normalize_mlp=
model/transformer_vae_batch.py:14
↓ 2 callersFunctionnormal
(tensor, mean, std)
model/layers.py:7
↓ 2 callersMethodreparameterize
(self, mu, logvar)
model/transformer_vae_batch.py:307
↓ 1 callersFunctionCham_loss
(v_pre, v_gt)
loss/loss.py:10
↓ 1 callersFunctionMSE_loss
(v_pre, v_gt)
loss/loss.py:6
↓ 1 callersFunctionSmooth_loss
(mesh)
loss/loss.py:14
↓ 1 callersMethod__init__
(self)
model/layers.py:61
↓ 1 callersMethodbackward
(ctx, grad_cost)
loss/earth_mover_distance.py:51
↓ 1 callersMethodbatch_pairwise_dist
(self, x, y)
loss/loss.py:68
↓ 1 callersFunctionload_config
()
config.py:5
↓ 1 callersFunctionmain
(config)
main.py:92
↓ 1 callersFunctionsave_model
(mesh_vae, optimizer, epoch, train_loss, val_loss, checkpoint_name)
main.py:29
↓ 1 callersFunctiontrain
(mesheart, trainloader, optimizer, device, config, writer, epoch)
main.py:39
↓ 1 callersFunctionval
(mesheart, validloader, optimizer, device, config, writer, epoch)
main.py:64
FunctionVAECELoss
(v_pre, v_gt, f, logvar, mu, beta=1e-2, lambd=1, lambd_s=1, loss='cham_smooth')
loss/loss.py:18
Method__getitem__
(self, index)
data/dataloader.py:39
Method__init__
(self, in_channels, out_channels, K, normalization=None, bias=True)
model/layers.py:12
Method__init__
(self, num_channels=None)
model/transformer_vae_batch.py:63
Method__init__
(self, d_model, dropout=0.1, max_len=60)
model/transformer_vae_batch.py:72
Method__init__
(self, d_model, dropout=0.1, max_len=5000)
model/transformer_vae_batch.py:93
Method__init__
(self, dim_in=3, points=10000, seq_len=50, c_dim=12, z_dim=32, ff_size=1024, num_layers=4, num_heads=4, dropou
model/transformer_vae_batch.py:107
Method__init__
(self, dim_in=3, points=10000, seq_len=50, c_dim=16, z_dim=32, ff_size=1024, num_layers=4, n
model/transformer_vae_batch.py:210
Method__init__
(self, dim_in=3, dim_h=128, z_dim = 32, c_dim_in=12, c_dim=16,
model/transformer_vae_batch.py:285
Method__init__
(self, points, C)
model/transformer_vae_batch.py:345
Method__init__
(self, config, data_usage='train')
data/dataloader.py:16
Method__init__
(self)
loss/loss.py:56
Method__init__
(self)
loss/earth_mover_distance.py:8
Method__len__
(self)
data/dataloader.py:71
Functionage_transform
(age)
util/utils.py:11
Functioncondition_normal
(age, gender, dise, hypert, normaltype=0, samesize=False, NUM_AGES=6
util/utils.py:58
Functioncondition_normal_batch
(age, gender, normaltype=0, samesize=True, NUM_AGES=6, N
util/utils.py:21
Methodforward
(self, x, edge_index, norm, edge_weight=None)
model/layers.py:34
Methodforward
(self, x, pool_mat, dtype=None)
model/layers.py:64
Methodforward
(self, input)
model/transformer_vae_batch.py:58
Methodforward
(self, input)
model/transformer_vae_batch.py:68
Methodforward
(self, x)
model/transformer_vae_batch.py:85
Methodforward
(self, x, mask, lengths)
model/transformer_vae_batch.py:97
Methodforward
(self, v, f, edge_list, con)
model/transformer_vae_batch.py:164
Methodforward
(self, z, con)
model/transformer_vae_batch.py:263
Methodforward
(self, v, f, edge_list, c)
model/transformer_vae_batch.py:313
Methodforward
(self, input)
model/transformer_vae_batch.py:350
Methodforward
(self, preds, gts)
loss/loss.py:60
Methodforward
(self, input1, input2)
loss/earth_mover_distance.py:12
Methodforward
(ctx, xyz1, xyz2)
loss/earth_mover_distance.py:18
Methodgenerate_one
(self, z, con)
model/transformer_vae_batch.py:331
Methodgenerate_random_one
(self, con)
model/transformer_vae_batch.py:336
Methodmessage
(self, x_j, norm)
model/layers.py:56
Methodmessage
(self, x_j, norm)
model/layers.py:69
Methodnorm
(edge_index, num_nodes, edge_weight=None, dtype=None)
model/layers.py:21
Functionpet_load
(filename)
util/utils.py:98
Functionpet_save
(pet, filename)
util/utils.py:94
Methodreset_parameters
(self)
model/layers.py:15
Methodreturn_latent
(self, v, f, edge_list, c)
model/transformer_vae_batch.py:324
Functionscipy_to_torch_sparse
(scp_matrix)
util/utils.py:103
Functionsetup_dir
(dir_path)
util/utils.py:6