Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/VSAnimator/stgan
/ functions
Functions
228 in github.com/VSAnimator/stgan
⨍
Functions
228
◇
Types & classes
49
Method
__init__
Initialize this dataset class. Parameters: opt (Option class) -- stores all the experiment flags; needs to be a subclass of BaseO
data/temporal_ir_dataset.py:16
Method
__init__
Initialize this class Step 1: create a dataset instance given the name [dataset_mode] Step 2: create a multi-threaded data loader.
data/__init__.py:65
Method
__init__
Initialize this dataset class. Parameters: opt (Option class) -- stores all the experiment flags; needs to be a subclass of BaseO
data/temporal_dataset.py:16
Method
__init__
Initialize this dataset class. Parameters: opt (Option class) -- stores all the experiment flags; needs to be a subclass of BaseO
data/unaligned_dataset.py:19
Method
__init__
Initialize this dataset class. Parameters: opt (Option class) -- stores all the experiment flags; needs to be a subclass of BaseO
data/aligned_ir_dataset.py:16
Method
__init__
Initialize this dataset class. Parameters: opt (Option class) -- stores all the experiment flags; needs to be a subclass of BaseO
data/aligned_dataset.py:16
Method
__init__
(self, root, transform=None, return_paths=False, loader=default_loader)
data/image_folder.py:41
Method
__init__
Initialize this dataset class. Parameters: opt (Option class) -- stores all the experiment flags; needs to be a subclass of BaseO
data/template_dataset.py:36
Method
__init__
Initialize this dataset class. Parameters: opt (Option class) -- stores all the experiment flags; needs to be a subclass of BaseO
data/single_dataset.py:12
Method
__init__
Reset the class; indicates the class hasn't been initailized
options/base_options.py:16
Method
__init__
Initialize the pix2pix class. Parameters: opt (Option class)-- stores all the experiment flags; needs to be a subclass of BaseOpt
models/temporal_branched_model.py:40
Method
__init__
Initialize the pix2pix class. Parameters: opt (Option class)-- stores all the experiment flags; needs to be a subclass of BaseOpt
models/temporal_ir_model.py:40
Method
__init__
Initialize the pix2pix class. Parameters: opt (Option class)-- stores all the experiment flags; needs to be a subclass of BaseOpt
models/temporal_model.py:40
Method
__init__
Initialize the pix2pix class. Parameters: opt (Option class)-- stores all the experiment flags; needs to be a subclass of BaseOpt
models/temporal_branched_ir_model.py:40
Method
__init__
Initialize the CycleGAN class. Parameters: opt (Option class)-- stores all the experiment flags; needs to be a subclass of BaseOp
models/cycle_gan_model.py:47
Method
__init__
Initialize the pix2pix class. Parameters: opt (Option class)-- stores all the experiment flags; needs to be a subclass of BaseOpt
models/test_model.py:31
Method
__init__
Initialize the BaseModel class. Parameters: opt (Option class)-- stores all the experiment flags; needs to be a subclass of BaseO
models/base_model.py:18
Method
__init__
Initialize the pix2pix class. Parameters: opt (Option class)-- stores all the experiment flags; needs to be a subclass of BaseOpt
models/pix2pix_model.py:39
Method
__init__
Construct a Resnet-based generator Parameters: input_nc (int) -- the number of channels in input images output_n
models/networks.py:315
Method
__init__
Initialize the Resnet block A resnet block is a conv block with skip connections We construct a conv block with build_conv_block func
models/networks.py:373
Method
__init__
Construct a Unet generator Parameters: input_nc (int) -- the number of channels in input images output_nc (int) -- th
models/networks.py:433
Method
__init__
Construct a Unet submodule with skip connections. Parameters: outer_nc (int) -- the number of filters in the outer conv layer
models/networks.py:468
Method
__init__
Construct a PatchGAN discriminator Parameters: input_nc (int) -- the number of channels in input images ndf (int)
models/networks.py:535
Method
__init__
Construct a 1x1 PatchGAN discriminator Parameters: input_nc (int) -- the number of channels in input images ndf (int
models/networks.py:583
Method
__init__
Initialize this model class. Parameters: opt -- training/test options A few things can be done here. - (required
models/template_model.py:41
Method
__init__
Initialize the pix2pix class. Parameters: opt (Option class)-- stores all the experiment flags; needs to be a subclass of BaseOpt
models/pix2pix_ir_model.py:39
Method
__init__
Construct a Resnet-based generator Parameters: input_nc (int) -- the number of channels in input images output_n
models/networks_branched.py:318
Method
__init__
Construct a Resnet-based generator Parameters: input_nc (int) -- the number of channels in input images output_n
models/networks_branched.py:398
Method
__init__
Initialize the Resnet block A resnet block is a conv block with skip connections We construct a conv block with build_conv_block func
models/networks_branched.py:471
Method
__init__
Construct a Unet generator Parameters: input_nc (int) -- the number of channels in input images output_nc (int) -- th
models/networks_branched.py:530
Method
__init__
Construct a Unet submodule with skip connections. Parameters: outer_nc (int) -- the number of filters in the outer conv layer
models/networks_branched.py:602
Method
__init__
Construct a Unet submodule with skip connections. Parameters: outer_nc (int) -- the number of filters in the outer conv layer
models/networks_branched.py:668
Method
__init__
Construct a Unet generator Parameters: input_nc (int) -- the number of channels in input images output_nc (int) -- th
models/networks_branched.py:734
Method
__init__
Construct a Unet submodule with skip connections. Parameters: outer_nc (int) -- the number of filters in the outer conv layer
models/networks_branched.py:806
Method
__init__
Construct a Unet generator Parameters: input_nc (int) -- the number of channels in input images output_nc (int) -- th
models/networks_branched.py:874
Method
__init__
Construct a Unet submodule with skip connections. Parameters: outer_nc (int) -- the number of filters in the outer conv layer
models/networks_branched.py:946
Method
__init__
Construct a Unet submodule with skip connections. Parameters: outer_nc (int) -- the number of filters in the outer conv layer
models/networks_branched.py:1015
Method
__init__
Construct a Unet submodule with skip connections. Parameters: outer_nc (int) -- the number of filters in the outer conv layer
models/networks_branched.py:1084
Method
__init__
Construct a PatchGAN discriminator Parameters: input_nc (int) -- the number of channels in input images ndf (int)
models/networks_branched.py:1151
Method
__init__
Construct a 1x1 PatchGAN discriminator Parameters: input_nc (int) -- the number of channels in input images ndf (int
models/networks_branched.py:1199
Method
__init__
(self, technique='cyclegan', verbose=True)
util/get_data.py:27
Method
__init__
Initialize the HTML classes Parameters: web_dir (str) -- a directory that stores the webpage. HTML file will be created at <web_d
util/html.py:14
Method
__init__
Initialize the Visualizer class Parameters: opt -- stores all the experiment flags; needs to be a subclass of BaseOptions
util/visualizer.py:58
Method
__init__
Initialize the ImagePool class Parameters: pool_size (int) -- the size of image buffer, if pool_size=0, no buffer will be created
util/image_pool.py:12
Method
__iter__
Return a batch of data
data/__init__.py:88
Method
__len__
Return the total number of images in the dataset.
data/base_dataset.py:46
Method
__len__
Return the total number of images in the dataset.
data/temporal_ir_dataset.py:71
Method
__len__
Return the number of data in the dataset
data/__init__.py:84
Method
__len__
Return the total number of images in the dataset.
data/temporal_dataset.py:65
Method
__len__
Return the total number of images in the dataset. As we have two datasets with potentially different number of images, we take a maxi
data/unaligned_dataset.py:65
Method
__len__
Return the total number of images in the dataset.
data/aligned_ir_dataset.py:64
Method
__len__
Return the total number of images in the dataset.
data/aligned_dataset.py:60
Method
__len__
(self)
data/image_folder.py:65
Method
__len__
Return the total number of images.
data/template_dataset.py:73
Method
__len__
Return the total number of images in the dataset.
data/single_dataset.py:38
Function
cal_gradient_penalty
Calculate the gradient penalty loss, used in WGAN-GP paper https://arxiv.org/abs/1704.00028 Arguments: netD (network) -- dis
models/networks.py:271
Function
cal_gradient_penalty
Calculate the gradient penalty loss, used in WGAN-GP paper https://arxiv.org/abs/1704.00028 Arguments: netD (network) -- dis
models/networks_branched.py:275
Function
default_loader
(path)
data/image_folder.py:35
Function
define_D
Create a discriminator Parameters: input_nc (int) -- the number of channels in input images ndf (int) -- the number
models/networks.py:155
Function
define_D
Create a discriminator Parameters: input_nc (int) -- the number of channels in input images ndf (int) -- the number
models/networks_branched.py:159
Function
define_G
Create a generator Parameters: input_nc (int) -- the number of channels in input images output_nc (int) -- the number of channels
models/networks.py:112
Function
define_G
Create a generator Parameters: input_nc (int) -- the number of channels in input images output_nc (int) -- the number of channels
models/networks_branched.py:112
Function
diagnose_network
Calculate and print the mean of average absolute(gradients) Parameters: net (torch network) -- Torch network name (str) -- the na
util/util.py:30
Method
forward
Run forward pass.
models/test_model.py:63
Method
forward
Standard forward
models/networks.py:365
Method
forward
Forward function (with skip connections)
models/networks.py:424
Method
forward
Standard forward
models/networks.py:457
Method
forward
(self, x)
models/networks.py:525
Method
forward
Standard forward.
models/networks.py:575
Method
forward
Standard forward.
models/networks.py:607
Method
forward
Standard forward
models/networks_branched.py:375
Method
forward
Standard forward
models/networks_branched.py:451
Method
forward
Forward function (with skip connections)
models/networks_branched.py:522
Method
forward
Standard forward
models/networks_branched.py:576
Method
forward
(self, x)
models/networks_branched.py:659
Method
forward
(self, x1, x2)
models/networks_branched.py:725
Method
forward
Standard forward
models/networks_branched.py:780
Method
forward
(self, x)
models/networks_branched.py:866
Method
forward
Standard forward
models/networks_branched.py:920
Method
forward
(self, x)
models/networks_branched.py:1004
Method
forward
(self, x1, x2)
models/networks_branched.py:1072
Method
forward
(self, x)
models/networks_branched.py:1141
Method
forward
Standard forward.
models/networks_branched.py:1191
Method
forward
Standard forward.
models/networks_branched.py:1223
Function
get_option_setter
Return the static method <modify_commandline_options> of the dataset class.
data/__init__.py:41
Function
get_option_setter
Return the static method <modify_commandline_options> of the model class.
models/__init__.py:48
Function
get_scheduler
Return a learning rate scheduler Parameters: optimizer -- the optimizer of the network opt (option class) -- stores all
models/networks.py:31
Function
get_scheduler
Return a learning rate scheduler Parameters: optimizer -- the optimizer of the network opt (option class) -- stores all
models/networks_branched.py:31
Function
init_func
(m)
models/networks.py:71
Function
init_func
(m)
models/networks_branched.py:71
Method
initialize
(self, parser)
options/train_options.py:10
Method
initialize
(self, parser)
options/test_options.py:10
Function
lambda_rule
(epoch)
models/networks.py:45
Function
lambda_rule
(epoch)
models/networks_branched.py:45
Function
mkdirs
create empty directories if they don't exist Parameters: paths (str list) -- a list of directory paths
util/util.py:76
Method
modify_commandline_options
Add new dataset-specific options, and rewrite default values for existing options. Parameters: parser -- original option
data/base_dataset.py:33
Method
modify_commandline_options
Add new dataset-specific options, and rewrite default values for existing options. Parameters: parser -- original option
data/template_dataset.py:22
Method
modify_commandline_options
Add new dataset-specific options, and rewrite default values for existing options. Parameters: parser -- original option
models/temporal_branched_model.py:17
Method
modify_commandline_options
Add new dataset-specific options, and rewrite default values for existing options. Parameters: parser -- original option
models/temporal_ir_model.py:17
Method
modify_commandline_options
Add new dataset-specific options, and rewrite default values for existing options. Parameters: parser -- original option
models/temporal_model.py:17
← previous
next →
101–200 of 228, ranked by callers