Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/WeilunWang/NEGCUT
/ types & classes
Types & classes
62 in github.com/WeilunWang/NEGCUT
⨍
Functions
264
◇
Types & classes
62
↓ 8 callers
Class
ConvLayer
models/stylegan_networks.py:621
↓ 6 callers
Class
UnetSkipConnectionBlock
Defines the Unet submodule with skip connection. X -------------------identity---------------------- |-- downsampling -- |submodule|
models/generator.py:195
↓ 4 callers
Class
EqualLinear
models/stylegan_networks.py:198
↓ 4 callers
Class
Normalize
models/feature_extractor.py:25
↓ 4 callers
Class
ResBlock
models/stylegan_networks.py:670
↓ 4 callers
Class
StyledConv
models/stylegan_networks.py:377
↓ 3 callers
Class
Blur
models/stylegan_networks.py:139
↓ 3 callers
Class
Downsample
models/utils.py:35
↓ 3 callers
Class
ResnetGenerator
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 callers
Class
FusedLeakyReLU
models/stylegan_networks.py:21
↓ 2 callers
Class
GANLoss
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 callers
Class
ModulatedConv2d
models/stylegan_networks.py:247
↓ 2 callers
Class
NLayerDiscriminator
Defines a PatchGAN discriminator
models/discriminator.py:56
↓ 2 callers
Class
Negative_Generator
models/negative_generator.py:53
↓ 2 callers
Class
Normalize
models/negative_generator.py:18
↓ 2 callers
Class
PatchSampleF
models/feature_extractor.py:107
↓ 2 callers
Class
StyleGAN2Generator
models/stylegan_networks.py:897
↓ 2 callers
Class
ToRGB
models/stylegan_networks.py:417
↓ 2 callers
Class
UnetGenerator
Create a Unet-based generator
models/generator.py:163
↓ 1 callers
Class
ConstantInput
models/stylegan_networks.py:364
↓ 1 callers
Class
CustomDatasetDataLoader
Wrapper class of Dataset class that performs multi-threaded data loading
data/__init__.py:62
↓ 1 callers
Class
EqualConv2d
models/stylegan_networks.py:158
↓ 1 callers
Class
HTML
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 callers
Class
Identity
models/utils.py:205
↓ 1 callers
Class
LearnedPatchNCELoss
models/losses.py:58
↓ 1 callers
Class
Negative_Placeholder
models/negative_generator.py:29
↓ 1 callers
Class
NoiseInjection
models/stylegan_networks.py:350
↓ 1 callers
Class
PatchNCELoss
models/losses.py:7
↓ 1 callers
Class
PixelDiscriminator
Defines a 1x1 PatchGAN discriminator (pixelGAN)
models/discriminator.py:123
↓ 1 callers
Class
PixelNorm
models/stylegan_networks.py:78
↓ 1 callers
Class
PoolingF
models/feature_extractor.py:37
↓ 1 callers
Class
ReshapeF
models/feature_extractor.py:48
↓ 1 callers
Class
ResnetBlock
Define a Resnet block
models/utils.py:213
↓ 1 callers
Class
ScaledLeakyReLU
models/stylegan_networks.py:235
↓ 1 callers
Class
StridedConvF
models/feature_extractor.py:61
↓ 1 callers
Class
StyleGAN2Decoder
models/stylegan_networks.py:851
↓ 1 callers
Class
StyleGAN2Discriminator
models/stylegan_networks.py:695
↓ 1 callers
Class
StyleGAN2Encoder
models/stylegan_networks.py:797
↓ 1 callers
Class
TestOptions
This class includes test options. It also includes shared options defined in BaseOptions.
options/test_options.py:4
↓ 1 callers
Class
TrainOptions
This class includes training options. It also includes shared options defined in BaseOptions.
options/train_options.py:4
↓ 1 callers
Class
Upsample
models/utils.py:61
↓ 1 callers
Class
Upsample
models/stylegan_networks.py:97
↓ 1 callers
Class
Visualizer
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
Class
BaseDataset
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
Class
BaseModel
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
Class
BaseOptions
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
Class
CUTModel
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
Class
Downsample
models/stylegan_networks.py:118
Class
Generator
models/stylegan_networks.py:439
Class
GetData
A Python script for downloading CycleGAN or pix2pix datasets. Parameters: technique (str) -- One of: 'cyclegan' or 'pix2pix'. ver
util/get_data.py:11
Class
ImageFolder
data/image_folder.py:40
Class
ImagePool
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
Class
NEGCUTModel
models/negcut_model.py:20
Class
PatchDiscriminator
Defines a PatchGAN discriminator
models/discriminator.py:155
Class
SinCUTModel
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
Class
SinNEGCUTModel
models/sinnegcut_model.py:5
Class
SingleDataset
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
Class
SingleImageDataset
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
Class
TemplateDataset
A template dataset class for you to implement custom datasets.
data/template_dataset.py:19
Class
TemplateModel
models/template_model.py:23
Class
TileStyleGAN2Discriminator
models/stylegan_networks.py:786
Class
UnalignedDataset
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