Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/NVlabs/MUNIT
/ functions
Functions
103 in github.com/NVlabs/MUNIT
⨍
Functions
103
◇
Types & classes
21
↓ 32 callers
Method
decode
(self, hiddens)
networks.py:179
↓ 21 callers
Method
encode
(self, images)
networks.py:174
↓ 14 callers
Method
__init__
(self, input_dim, output_dim, dim, n_blk, norm='none', activ='relu')
networks.py:257
↓ 8 callers
Function
get_scheduler
(optimizer, hyperparameters, iterations=-1)
utils.py:261
↓ 8 callers
Method
recon_criterion
(self, input, target)
trainer.py:53
↓ 6 callers
Function
weights_init
(init_type='gaussian')
utils.py:272
↓ 6 callers
Function
write_one_row_html
(html_file, iterations, img_filename, all_size)
utils.py:137
↓ 5 callers
Function
get_data_loader_folder
(input_folder, batch_size, train, new_size=None, height=256, width=256, num_workers
utils.py:86
↓ 5 callers
Method
save
(self, snapshot_dir, iterations)
trainer.py:374
↓ 4 callers
Method
__compute_kl
(self, mu)
trainer.py:249
↓ 4 callers
Method
calc_dis_loss
(self, input_fake, input_real)
networks.py:53
↓ 4 callers
Method
calc_gen_loss
(self, input_fake)
networks.py:71
↓ 4 callers
Function
get_data_loader_list
(root, file_list, batch_size, train, new_size=None, height=256, width=256, num_work
utils.py:73
↓ 4 callers
Function
get_model_list
(dirname, key)
utils.py:209
↓ 4 callers
Function
l2normalize
(v, eps=1e-12)
networks.py:514
↓ 4 callers
Method
recon_criterion
(self, input, target)
trainer.py:237
↓ 4 callers
Function
vgg_preprocess
(batch)
utils.py:248
↓ 3 callers
Method
forward
(self, x)
networks.py:46
↓ 3 callers
Function
get_config
(config)
utils.py:100
↓ 3 callers
Method
sample
(self, x_a, x_b)
trainer.py:315
↓ 3 callers
Function
write_2images
(image_outputs, display_image_num, image_directory, postfix)
utils.py:119
↓ 2 callers
Function
__conversion_core
(state_dict_base, trainer_name)
utils.py:308
↓ 2 callers
Function
__write_images
(image_outputs, display_image_num, file_name)
utils.py:112
↓ 2 callers
Method
compute_vgg_loss
(self, vgg, img, target)
trainer.py:118
↓ 2 callers
Method
compute_vgg_loss
(self, vgg, img, target)
trainer.py:308
↓ 2 callers
Function
load_inception
(model_path)
utils.py:237
↓ 2 callers
Function
load_vgg16
Use the model from https://github.com/abhiskk/fast-neural-style/blob/master/neural_style/utils.py
utils.py:221
↓ 2 callers
Function
pytorch03_to_pytorch04
(state_dict_base, trainer_name)
utils.py:307
↓ 1 callers
Method
__init__
(self, hyperparameters)
trainer.py:203
↓ 1 callers
Method
_made_params
(self)
networks.py:545
↓ 1 callers
Method
_make_net
(self)
networks.py:35
↓ 1 callers
Method
_make_params
(self)
networks.py:555
↓ 1 callers
Method
_update_u_v
(self)
networks.py:531
↓ 1 callers
Method
assign_adain_params
(self, adain_params, model)
networks.py:130
↓ 1 callers
Method
decode
(self, content, style)
networks.py:123
↓ 1 callers
Method
dis_update
(self, x_a, x_b, hyperparameters)
trainer.py:331
↓ 1 callers
Method
encode
(self, images)
networks.py:117
↓ 1 callers
Method
forward
(self, x)
networks.py:267
↓ 1 callers
Method
gen_update
(self, x_a, x_b, hyperparameters)
trainer.py:259
↓ 1 callers
Function
get_all_data_loaders
(conf)
utils.py:41
↓ 1 callers
Method
get_num_adain_params
(self, model)
networks.py:141
↓ 1 callers
Function
is_image_file
(filename)
data.py:85
↓ 1 callers
Function
make_dataset
(dir)
data.py:89
↓ 1 callers
Function
prepare_sub_folder
(output_directory)
utils.py:125
↓ 1 callers
Method
resume
(self, checkpoint_dir, hyperparameters)
trainer.py:352
↓ 1 callers
Function
slerp
original: Animating Rotation with Quaternion Curves, Ken Shoemake https://arxiv.org/abs/1609.04468 Code: https://github.com/soumith/dcgan
utils.py:179
↓ 1 callers
Method
update_learning_rate
(self)
trainer.py:346
↓ 1 callers
Function
write_html
(filename, iterations, image_save_iterations, image_directory, all_size=1536)
utils.py:148
↓ 1 callers
Function
write_loss
(iterations, trainer, train_writer)
utils.py:172
Method
__enter__
(self)
utils.py:300
Method
__exit__
(self, exc_type, exc_value, exc_tb)
utils.py:303
Method
__getitem__
(self, index)
data.py:33
Method
__getitem__
(self, index)
data.py:56
Method
__getitem__
(self, index)
data.py:118
Method
__init__
(self, msg)
utils.py:296
Method
__init__
(self, input_dim, params)
networks.py:20
Method
__init__
(self, input_dim, params)
networks.py:91
Method
__init__
(self, input_dim, params)
networks.py:152
Method
__init__
(self, n_downsample, input_dim, dim, style_dim, norm, activ, pad_type)
networks.py:189
Method
__init__
(self, n_downsample, n_res, input_dim, dim, norm, activ, pad_type)
networks.py:207
Method
__init__
(self, n_upsample, n_res, dim, output_dim, res_norm='adain', activ='relu', pad_type='zero')
networks.py:224
Method
__init__
(self, num_blocks, dim, norm='in', activation='relu', pad_type='zero')
networks.py:246
Method
__init__
(self, dim, norm='in', activation='relu', pad_type='zero')
networks.py:274
Method
__init__
(self, input_dim ,output_dim, kernel_size, stride, padding=0, norm='none', activation='relu',
networks.py:289
Method
__init__
(self, input_dim, output_dim, norm='none', activation='relu')
networks.py:350
Method
__init__
(self)
networks.py:400
Method
__init__
(self, num_features, eps=1e-5, momentum=0.1)
networks.py:454
Method
__init__
(self, num_features, eps=1e-5, affine=True)
networks.py:486
Method
__init__
(self, module, name='weight', power_iterations=1)
networks.py:523
Method
__init__
(self, root, flist, transform=None, flist_reader=default_flist_reader, loader=default_loader)
data.py:26
Method
__init__
(self, root, flist, transform=None, flist_reader=default_flist_reader, loader=default_loader)
data.py:46
Method
__init__
(self, root, transform=None, return_paths=False, loader=default_loader)
data.py:104
Method
__init__
(self, hyperparameters)
trainer.py:13
Method
__len__
(self)
data.py:41
Method
__len__
(self)
data.py:63
Method
__len__
(self)
data.py:128
Method
__repr__
(self)
networks.py:481
Function
default_flist_reader
flist format: impath label\nimpath label\n ...(same to caffe's filelist)
data.py:12
Function
default_loader
(path)
data.py:8
Method
dis_update
(self, x_a, x_b, hyperparameters)
trainer.py:147
Function
eformat
(f, prec)
utils.py:105
Method
forward
(self, images)
networks.py:111
Method
forward
(self, images)
networks.py:164
Method
forward
(self, x)
networks.py:203
Method
forward
(self, x)
networks.py:220
Method
forward
(self, x)
networks.py:239
Method
forward
(self, x)
networks.py:253
Method
forward
(self, x)
networks.py:282
Method
forward
(self, x)
networks.py:341
Method
forward
(self, x)
networks.py:388
Method
forward
(self, X)
networks.py:420
Method
forward
(self, x)
networks.py:466
Method
forward
(self, x)
networks.py:496
Method
forward
(self, *args)
networks.py:574
Method
forward
(self, x_a, x_b)
trainer.py:56
Method
forward
(self, x_a, x_b)
trainer.py:240
Method
gen_update
(self, x_a, x_b, hyperparameters)
trainer.py:67
Function
get_slerp_interp
modified from: PyTorch inference for "Progressive Growing of GANs" with CelebA snapshot https://github.com/ptrblck/prog_gans_pytorch_inferenc
utils.py:190
Function
init_fun
(m)
utils.py:273
Method
resume
(self, checkpoint_dir, hyperparameters)
trainer.py:170
next →
1–100 of 103, ranked by callers