MCPcopy Create free account

hub / github.com/Bei-Jin/STMFANet / functions

Functions87 in github.com/Bei-Jin/STMFANet

⨍ Functions 87◇ Types & classes 17
↓ 9 callersMethod__init__
(self, c_dim, gpu_ids)
wavenet_models/STMF_network.py:590
↓ 6 callersFunctionfixed_unpooling
(x, gpu_ids)
wavenet_models/STMF_network.py:390
↓ 5 callersMethodforward
(self)
wavenet_models/STMF_Model.py:64
↓ 5 callersMethodforward
(self, x)
wavenet_models/STMF_network.py:154
↓ 5 callersFunctioninit_weights
(net, init_type='normal')
wavenet_models/STMF_network.py:324
↓ 3 callersMethod_make_dense
(self, nChannels, growthRate, nDenseBlocks, bottleneck)
wavenet_models/STMF_network.py:172
↓ 3 callersMethodbackward_G
(self)
wavenet_models/STMF_Model.py:93
↓ 2 callersFunction__scale_width
(img, target_width)
data/base_dataset.py:41
↓ 2 callersMethodbackward_D
(self)
wavenet_models/STMF_Model.py:68
↓ 2 callersFunctiondraw_frame_tensor
(img, K, T)
util/util.py:105
↓ 2 callersMethodinitialize
(self, opt)
data/data_loader.py:22
↓ 2 callersFunctioninverse_transform
(images)
util/util.py:75
↓ 2 callersMethodname
(self)
data/data_loader.py:19
↓ 1 callersFunctionCreateDataset
(opt)
data/data_loader.py:5
↓ 1 callersFunctioncreate_model
(opt)
wavenet_models/create_model.py:1
↓ 1 callersFunctiondefine_convLstm_cell
(feature_size, num_features, forget_bias=1, gpu_ids=[], init_type="STMF", bias=True)
wavenet_models/STMF_network.py:64
↓ 1 callersFunctiondefine_decoder
(c_dim, gf_dim, init_type="STMF", gpu_ids=[])
wavenet_models/STMF_network.py:80
↓ 1 callersFunctiondefine_encoder
(opt, init_type = 'STMF')
wavenet_models/STMF_network.py:53
↓ 1 callersFunctionfore_transform
(images)
util/util.py:71
↓ 1 callersMethodget_current_errors
(self)
wavenet_models/STMF_Model.py:175
↓ 1 callersMethodget_current_visuals
(self)
wavenet_models/base_model.py:32
↓ 1 callersMethodinitialize
(self)
options/train_options.py:7
↓ 1 callersMethodinitialize
(self, opt)
wavenet_models/STMF_Model.py:14
↓ 1 callersMethodload
(self, epoch_label)
wavenet_models/STMF_Model.py:222
↓ 1 callersFunctionmain
()
train_kth.py:11
↓ 1 callersMethodname
(self)
wavenet_models/STMF_Model.py:11
↓ 1 callersMethodoptimize_parameters
(self)
wavenet_models/STMF_Model.py:132
↓ 1 callersMethodparse
(self)
options/train_options.py:57
↓ 1 callersMethodprint_current_errors
(self, epoch, i, errors)
util/visualizer.py:35
↓ 1 callersMethodsave
(self, label, epoch)
wavenet_models/STMF_Model.py:204
↓ 1 callersMethodset_inputs
(self, input)
wavenet_models/STMF_Model.py:189
FunctionCreateDataLoader
(opt)
data/data_loader.py:35
Method__call__
(self, output, target)
wavenet_models/STMF_network.py:602
Method__init__
(self)
data/base_data_loader.py:2
Method__init__
(self)
data/base_dataset.py:7
Method__init__
(self)
options/train_options.py:4
Method__init__
(self, opt)
wavenet_models/STMF_network.py:21
Method__init__
(self, opt)
wavenet_models/STMF_network.py:97
Method__init__
(self, nChannels, growthRate)
wavenet_models/STMF_network.py:183
Method__init__
(self, nChannels, growthRate)
wavenet_models/STMF_network.py:200
Method__init__
(self, nChannels, nOutChannels)
wavenet_models/STMF_network.py:214
Method__init__
(self, outC1, outC2, outC3)
wavenet_models/STMF_network.py:227
Method__init__
(self, feature_size, num_features, gpu_ids, forget_bias=1, bias=True)
wavenet_models/STMF_network.py:294
Method__init__
(self, c_dim, gf_dim, gpu_ids)
wavenet_models/STMF_network.py:343
Method__init__
(self, img_size, c_dim, in_num, out_num, df_dim, gpu_ids)
wavenet_models/STMF_network.py:514
Method__init__
(self, opt)
util/visualizer.py:11
Method__len__
(self)
data/data_loader.py:32
Functiondefine_discriminator
(img_size, c_dim, in_num, out_num, df_dim, init_type="STMF", gpu_ids=[])
wavenet_models/STMF_network.py:499
Functiondefine_gdl
(c_dim, gpu_ids=[])
wavenet_models/STMF_network.py:573
Functiondefine_generator
(opt)
wavenet_models/STMF_network.py:7
Methodforward
(self)
wavenet_models/base_model.py:19
Methodforward
(self, inputs, state)
wavenet_models/STMF_network.py:35
Methodforward
(self, x)
wavenet_models/STMF_network.py:193
Methodforward
(self, x)
wavenet_models/STMF_network.py:207
Methodforward
(self, x)
wavenet_models/STMF_network.py:221
Methodforward
(self, x)
wavenet_models/STMF_network.py:261
Methodforward
(self, input, state)
wavenet_models/STMF_network.py:306
Methodforward
(self, x)
wavenet_models/STMF_network.py:370
Methodforward
(self, input, batch_size)
wavenet_models/STMF_network.py:559
Methodget_current_errors
(self)
wavenet_models/base_model.py:35
Methodget_image_paths
(self)
wavenet_models/base_model.py:26
Functionget_minibatches_idx
Used to shuffle the dataset at each iteration.
util/util.py:5
Functionget_scheduler
(optimizer, opt)
wavenet_models/STMF_network.py:618
Functionget_transform
(opt)
data/base_dataset.py:16
Methodinitialize
(self, opt)
data/base_data_loader.py:5
Methodinitialize
(self, opt)
data/base_dataset.py:13
Methodinitialize
(self, opt)
wavenet_models/base_model.py:9
Functionlambda_rule
(epoch)
wavenet_models/STMF_network.py:620
Methodload_data
(self)
data/base_data_loader.py:9
Methodload_data
(self)
data/data_loader.py:29
Functionload_kth_data
(f_name, data_path, image_size, K, T)
util/util.py:33
Methodname
(self)
data/base_dataset.py:10
Methodname
(self)
wavenet_models/base_model.py:6
Methodoptimize_parameters
(self)
wavenet_models/base_model.py:29
Methodplot_current_errors
(self, epoch, counter_ratio, errors)
util/visualizer.py:21
Functionprint_current_errors
(epoch, i, errors, checkpoints_dir, name)
util/util.py:125
Methodsave
(self, label, epoch)
wavenet_models/base_model.py:38
Methodsave_images
:param visuals: dict, diff_in: K-1 [batch_size, h, w, c], ndarray, [0,255]; targets: K+T ndarrays, pred: T ndarray :param video_name:
util/visualizer.py:47
Methodset_input
(self, input)
wavenet_models/base_model.py:16
Methodtest
(self)
wavenet_models/base_model.py:23
Functionvisual_grid
(seq_batch, pred, K, T)
util/util.py:79
Functionweights_init_kaiming
(m)
wavenet_models/STMF_network.py:431
Functionweights_init_normal
(m)
wavenet_models/STMF_network.py:407
Functionweights_init_orthogonal
(m)
wavenet_models/STMF_network.py:443
Functionweights_init_stmf
(m)
wavenet_models/STMF_network.py:477
Functionweights_init_xavier
(m)
wavenet_models/STMF_network.py:419
Functionweights_init_zeros
(m)
wavenet_models/STMF_network.py:455