MCPcopy Create free account

hub / github.com/VSAnimator/stgan / types & classes

Types & classes49 in github.com/VSAnimator/stgan

↓ 16 callersClassUnetBranchedSkipConnectionBlockUp
Defines the Unet submodule with skip connection. X -------------------identity---------------------- |-- downsampling -- |submodule| -
models/networks_branched.py:1009
↓ 8 callersClassUnetBranchedSkipConnectionBlockDown
Defines the Unet submodule with skip connection. X -------------------identity---------------------- |-- downsampling -- |submodule| -
models/networks_branched.py:940
↓ 8 callersClassUnetIndependentBranchedSkipConnectionBlockDown
Defines the Unet submodule with skip connection. X -------------------identity---------------------- |-- downsampling -- |submodule| -
models/networks_branched.py:800
↓ 8 callersClassUnetSkipConnectionBlockDown
Defines the Unet submodule with skip connection. X -------------------identity---------------------- |-- downsampling -- |submodule| -
models/networks_branched.py:596
↓ 8 callersClassUnetSkipConnectionBlockUp
Defines the Unet submodule with skip connection. X -------------------identity---------------------- |-- downsampling -- |submodule| -
models/networks_branched.py:662
↓ 6 callersClassUnetSkipConnectionBlock
Defines the Unet submodule with skip connection. X -------------------identity---------------------- |-- downsampling -- |submodule| -
models/networks.py:462
↓ 2 callersClassImagePool
This class implements an image buffer that stores previously generated images. This buffer enables us to update discriminators using a history of
util/image_pool.py:5
↓ 2 callersClassNLayerDiscriminator
Defines a PatchGAN discriminator
models/networks.py:532
↓ 2 callersClassNLayerDiscriminator
Defines a PatchGAN discriminator
models/networks_branched.py:1148
↓ 2 callersClassResnetBlock
Define a Resnet block
models/networks_branched.py:468
↓ 2 callersClassResnetGenerator
Resnet-based generator that consists of Resnet blocks between a few downsampling/upsampling operations. We adapt Torch code and idea from Justin
models/networks.py:309
↓ 2 callersClassResnetGenerator
Resnet-based generator that consists of Resnet blocks between a few downsampling/upsampling operations. We adapt Torch code and idea from Justin
models/networks_branched.py:392
↓ 2 callersClassUnetGenerator
Create a Unet-based generator
models/networks.py:430
↓ 1 callersClassCustomDatasetDataLoader
Wrapper class of Dataset class that performs multi-threaded data loading
data/__init__.py:62
↓ 1 callersClassHTML
This HTML class allows us to save images and write texts into a single HTML file. It consists of functions such as <add_header> (add a text head
util/html.py:6
↓ 1 callersClassIndependentResnetGenerator
Resnet-based generator that consists of Resnet blocks between a few downsampling/upsampling operations. We adapt Torch code and idea from Justin
models/networks_branched.py:312
↓ 1 callersClassPixelDiscriminator
Defines a 1x1 PatchGAN discriminator (pixelGAN)
models/networks.py:580
↓ 1 callersClassPixelDiscriminator
Defines a 1x1 PatchGAN discriminator (pixelGAN)
models/networks_branched.py:1196
↓ 1 callersClassResnetBlock
Define a Resnet block
models/networks.py:370
↓ 1 callersClassTestOptions
This class includes test options. It also includes shared options defined in BaseOptions.
options/test_options.py:4
↓ 1 callersClassTrainOptions
This class includes training options. It also includes shared options defined in BaseOptions.
options/train_options.py:4
↓ 1 callersClassUnetBranchedGenerator
Create a Unet-based generator
models/networks_branched.py:871
↓ 1 callersClassUnetGenerator
Create a Unet-based generator
models/networks_branched.py:527
↓ 1 callersClassUnetIndependentBranchedGenerator
Create a Unet-based generator
models/networks_branched.py:731
↓ 1 callersClassVisualizer
This class includes several functions that can display/save images and print/save logging information. It uses a Python library 'visdom' for disp
util/visualizer.py:52
ClassAlignedDataset
A dataset class for paired image dataset. It assumes that the directory '/path/to/data/train' contains image pairs in the form of {A,B}. Duri
data/aligned_dataset.py:9
ClassAlignedIrDataset
A dataset class for paired image dataset. It assumes that the directory '/path/to/data/train' contains image pairs in the form of {A,B}. Duri
data/aligned_ir_dataset.py:9
ClassBaseDataset
This class is an abstract base class (ABC) for datasets. To create a subclass, you need to implement the following four functions: -- <__init
data/base_dataset.py:13
ClassBaseModel
This class is an abstract base class (ABC) for models. To create a subclass, you need to implement the following five functions: -- <__ini
models/base_model.py:8
ClassBaseOptions
This class defines options used during both training and test time. It also implements several helper functions such as parsing, printing, and sa
options/base_options.py:9
ClassCycleGANModel
This class implements the CycleGAN model, for learning image-to-image translation without paired data. The model training requires '--datase
models/cycle_gan_model.py:8
ClassGANLoss
Define different GAN objectives. The GANLoss class abstracts away the need to create the target label tensor that has the same size as the in
models/networks.py:202
ClassGANLoss
Define different GAN objectives. The GANLoss class abstracts away the need to create the target label tensor that has the same size as the in
models/networks_branched.py:206
ClassGetData
A Python script for downloading CycleGAN or pix2pix datasets. Parameters: technique (str) -- One of: 'cyclegan' or 'pix2pix'. ver
util/get_data.py:11
ClassImageFolder
data/image_folder.py:39
ClassPix2PixIrModel
This class implements the pix2pix model, for learning a mapping from input images to output images given paired data. The model training require
models/pix2pix_ir_model.py:6
ClassPix2PixModel
This class implements the pix2pix model, for learning a mapping from input images to output images given paired data. The model training require
models/pix2pix_model.py:6
ClassSingleDataset
This dataset class can load a set of images specified by the path --dataroot /path/to/data. It can be used for generating CycleGAN results only f
data/single_dataset.py:6
ClassTemplateDataset
A template dataset class for you to implement custom datasets.
data/template_dataset.py:19
ClassTemplateModel
models/template_model.py:23
ClassTemporalBranchedIrModel
This class implements the pix2pix model, for learning a mapping from input images to output images given paired data. The model training require
models/temporal_branched_ir_model.py:6
ClassTemporalBranchedModel
This class implements the pix2pix model, for learning a mapping from input images to output images given paired data. The model training require
models/temporal_branched_model.py:6
ClassTemporalDataset
A dataset class for temporal image dataset. It assumes that the directory '/path/to/data/train' contains image pairs in the form of {{A_0, A_1, A
data/temporal_dataset.py:9
ClassTemporalIrDataset
A dataset class for temporal image dataset. It assumes that the directory '/path/to/data/train' contains image pairs in the form of {{A_0, A_1, A
data/temporal_ir_dataset.py:9
ClassTemporalModel
This class implements the pix2pix model, for learning a mapping from input images to output images given paired data. The model training require
models/temporal_ir_model.py:6
ClassTemporalModel
This class implements the pix2pix model, for learning a mapping from input images to output images given paired data. The model training require
models/temporal_model.py:6
ClassTestModel
This TesteModel can be used to generate CycleGAN results for only one direction. This model will automatically set '--dataset_mode single', which
models/test_model.py:5
ClassUnalignedDataset
This dataset class can load unaligned/unpaired datasets. It requires two directories to host training images from domain A '/path/to/data/tr
data/unaligned_dataset.py:8
ClassUnetSkipConnectionBlock
Defines the Unet submodule with skip connection. X -------------------identity---------------------- |-- downsampling -- |submodule| -
models/networks_branched.py:1078