MCPcopy Create free account

hub / github.com/WeilunWang/NEGCUT / types & classes

Types & classes62 in github.com/WeilunWang/NEGCUT

↓ 8 callersClassConvLayer
models/stylegan_networks.py:621
↓ 6 callersClassUnetSkipConnectionBlock
Defines the Unet submodule with skip connection. X -------------------identity---------------------- |-- downsampling -- |submodule|
models/generator.py:195
↓ 4 callersClassEqualLinear
models/stylegan_networks.py:198
↓ 4 callersClassNormalize
models/feature_extractor.py:25
↓ 4 callersClassResBlock
models/stylegan_networks.py:670
↓ 4 callersClassStyledConv
models/stylegan_networks.py:377
↓ 3 callersClassBlur
models/stylegan_networks.py:139
↓ 3 callersClassDownsample
models/utils.py:35
↓ 3 callersClassResnetGenerator
Resnet-based generator that consists of Resnet blocks between a few downsampling/upsampling operations. We adapt Torch code and idea from Justi
models/generator.py:62
↓ 2 callersClassFusedLeakyReLU
models/stylegan_networks.py:21
↓ 2 callersClassGANLoss
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/losses.py:114
↓ 2 callersClassModulatedConv2d
models/stylegan_networks.py:247
↓ 2 callersClassNLayerDiscriminator
Defines a PatchGAN discriminator
models/discriminator.py:56
↓ 2 callersClassNegative_Generator
models/negative_generator.py:53
↓ 2 callersClassNormalize
models/negative_generator.py:18
↓ 2 callersClassPatchSampleF
models/feature_extractor.py:107
↓ 2 callersClassStyleGAN2Generator
models/stylegan_networks.py:897
↓ 2 callersClassToRGB
models/stylegan_networks.py:417
↓ 2 callersClassUnetGenerator
Create a Unet-based generator
models/generator.py:163
↓ 1 callersClassConstantInput
models/stylegan_networks.py:364
↓ 1 callersClassCustomDatasetDataLoader
Wrapper class of Dataset class that performs multi-threaded data loading
data/__init__.py:62
↓ 1 callersClassEqualConv2d
models/stylegan_networks.py:158
↓ 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 callersClassIdentity
models/utils.py:205
↓ 1 callersClassLearnedPatchNCELoss
models/losses.py:58
↓ 1 callersClassNegative_Placeholder
models/negative_generator.py:29
↓ 1 callersClassNoiseInjection
models/stylegan_networks.py:350
↓ 1 callersClassPatchNCELoss
models/losses.py:7
↓ 1 callersClassPixelDiscriminator
Defines a 1x1 PatchGAN discriminator (pixelGAN)
models/discriminator.py:123
↓ 1 callersClassPixelNorm
models/stylegan_networks.py:78
↓ 1 callersClassPoolingF
models/feature_extractor.py:37
↓ 1 callersClassReshapeF
models/feature_extractor.py:48
↓ 1 callersClassResnetBlock
Define a Resnet block
models/utils.py:213
↓ 1 callersClassScaledLeakyReLU
models/stylegan_networks.py:235
↓ 1 callersClassStridedConvF
models/feature_extractor.py:61
↓ 1 callersClassStyleGAN2Decoder
models/stylegan_networks.py:851
↓ 1 callersClassStyleGAN2Discriminator
models/stylegan_networks.py:695
↓ 1 callersClassStyleGAN2Encoder
models/stylegan_networks.py:797
↓ 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 callersClassUpsample
models/utils.py:61
↓ 1 callersClassUpsample
models/stylegan_networks.py:97
↓ 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:46
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
ClassCUTModel
This class implements CUT and FastCUT model, described in the paper Contrastive Learning for Unpaired Image-to-Image Translation Taesung Park
models/cut_model.py:11
ClassDownsample
models/stylegan_networks.py:118
ClassGenerator
models/stylegan_networks.py:439
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:40
ClassImagePool
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
ClassNEGCUTModel
models/negcut_model.py:20
ClassPatchDiscriminator
Defines a PatchGAN discriminator
models/discriminator.py:155
ClassSinCUTModel
This class implements the single image translation model (Fig 9) of Contrastive Learning for Unpaired Image-to-Image Translation Taesung Park
models/sincut_model.py:5
ClassSinNEGCUTModel
models/sinnegcut_model.py:5
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
ClassSingleImageDataset
This dataset class can load unaligned/unpaired datasets. It requires two directories to host training images from domain A '/path/to/data/tr
data/singleimage_dataset.py:10
ClassTemplateDataset
A template dataset class for you to implement custom datasets.
data/template_dataset.py:19
ClassTemplateModel
models/template_model.py:23
ClassTileStyleGAN2Discriminator
models/stylegan_networks.py:786
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:9