MCPcopy Create free account

hub / github.com/OpenTalker/StyleHEAT / types & classes

Types & classes196 in github.com/OpenTalker/StyleHEAT

↓ 24 callersClassbottleneck_IR
models/e4e/helpers.py:76
↓ 17 callersClassAttrDict
Dict as attribute trick.
configs/config.py:9
↓ 9 callersClassEqualConv2d
models/stylegan2/model.py:91
↓ 8 callersClassConvLayer
models/stylegan2/model.py:583
↓ 7 callersClassADAIN
models/styleheat/base_function.py:160
↓ 7 callersClassNetLinLayer
A single linear layer which does a 1x1 conv
third_part/PerceptualSimilarity/models/networks_basic.py:215
↓ 7 callersClass_PerceptualNetwork
r"""The network that extracts features to compute the perceptual loss. Args: network (nn.Sequential) : The network that extracts features
loss/perceptual.py:170
↓ 6 callersClassPerceptualLoss
r"""Perceptual loss initialization. Args: network (str) : The name of the loss network: 'vgg16' | 'vgg19'. layers (str or list of
loss/perceptual.py:25
↓ 6 callersClassScaledLeakyReLU
models/stylegan2/model.py:166
↓ 5 callersClassConvBlock
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/backbones/mobilefacenet.py:16
↓ 5 callersClassEqualLinear
models/stylegan2/model.py:129
↓ 4 callersClassDepthWise
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/backbones/mobilefacenet.py:41
↓ 4 callersClassMeter
Meter is to keep track of statistics along steps. Meters write values for purpose like printing average values. Meters can be flushed to log f
utils/meters.py:103
↓ 3 callersClassBlur
models/stylegan2/model.py:72
↓ 3 callersClassFinalBlock2d
Define the output layer
models/styleheat/base_function.py:399
↓ 3 callersClassFusedLeakyReLU
models/stylegan2/op/fused_act.py:72
↓ 3 callersClassGradualStyleBlock
models/e4e/e4e.py:32
↓ 3 callersClassNoWDataset
third_part/decalib/datasets/now.py:12
↓ 3 callersClassResNet
third_part/decalib/models/resnet.py:23
↓ 3 callersClassResidual
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/backbones/mobilefacenet.py:63
↓ 3 callersClassStyledConv
models/stylegan2/model.py:305
↓ 3 callersClassVGGFace2Dataset
third_part/decalib/datasets/vggface.py:12
↓ 2 callersClassAFLW2000
third_part/decalib/datasets/aflw2000.py:13
↓ 2 callersClassArcFaceORT
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/onnx_helper.py:15
↓ 2 callersClassAudioWarper
models/styleheat/warper.py:48
↓ 2 callersClassBottleneck
A named tuple describing a ResNet block.
models/e4e/helpers.py:22
↓ 2 callersClassCOCODataset
third_part/decalib/datasets/train_datasets.py:206
↓ 2 callersClassCalibrationNet
models/styleheat/calibration_net.py:127
↓ 2 callersClassCelebAHQDataset
third_part/decalib/datasets/train_datasets.py:299
↓ 2 callersClassConfig
r"""Configuration class. This should include every human specifiable hyperparameter values for your training.
configs/config.py:66
↓ 2 callersClassConvUpLayer
Conv Up Layer. Bilinear upsample + Conv. Args: in_channels (int): Channel number of the input. out_channels (int): Channel number
models/styleheat/calibration_net.py:40
↓ 2 callersClassDoubleConv
(convolution => [BN] => ReLU) * 2
third_part/decalib/models/resnet.py:193
↓ 2 callersClassEmbedding
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/eval_ijbc.py:54
↓ 2 callersClassEthnicityDataset
third_part/decalib/datasets/ethnicity.py:12
↓ 2 callersClassFineADAINResBlock2d
Define an Residual block for different types
models/styleheat/base_function.py:375
↓ 2 callersClassFineADAINResBlocks
models/styleheat/base_function.py:343
↓ 2 callersClassGMM
third_part/Deep3DFaceRecon_pytorch/util/skin_mask.py:9
↓ 2 callersClassJump
models/styleheat/base_function.py:359
↓ 2 callersClassLFold
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/eval/verification.py:41
↓ 2 callersClassLinearBlock
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/backbones/mobilefacenet.py:29
↓ 2 callersClassMappingNet
models/styleheat/warper.py:79
↓ 2 callersClassModulatedConv2d
models/stylegan2/model.py:178
↓ 2 callersClassPytorch3dRasterizer
Borrowed from https://github.com/facebookresearch/pytorch3d Notice: x,y,z are in image space, normalized can only render squared
third_part/decalib/utils/renderer.py:108
↓ 2 callersClassResBlock
models/stylegan2/model.py:632
↓ 2 callersClassResnetEncoder
third_part/decalib/models/encoders.py:22
↓ 2 callersClassStandardRasterizer
Alg: https://www.scratchapixel.com/lessons/3d-basic-rendering/rasterization-practical-implementation Notice: x,y,z are in image space, no
third_part/decalib/utils/renderer.py:47
↓ 2 callersClassTempVideoDataset
data/inference_dataset.py:60
↓ 2 callersClassToRGB
models/stylegan2/model.py:342
↓ 2 callersClassUpBlock2d
models/styleheat/base_function.py:328
↓ 2 callersClassVGGFace2HQDataset
third_part/decalib/datasets/vggface.py:124
↓ 2 callersClassVoxelDataset
third_part/decalib/datasets/train_datasets.py:51
↓ 2 callersClassWarpingNet
models/styleheat/warper.py:107
↓ 1 callersClassADAINDecoder
docstring for ADAINDecoder
models/styleheat/base_function.py:61
↓ 1 callersClassADAINDecoderBlock
models/styleheat/base_function.py:111
↓ 1 callersClassADAINEncoder
models/styleheat/base_function.py:39
↓ 1 callersClassADAINEncoderBlock
models/styleheat/base_function.py:92
↓ 1 callersClassADAINHourglass
models/styleheat/base_function.py:28
↓ 1 callersClassAlexNet
loss/lpips.py:89
↓ 1 callersClassAlignedDataSet
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/onnx_ijbc.py:28
↓ 1 callersClassArcFace
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/losses.py:29
↓ 1 callersClassAudioDataset
data/audio_dataset.py:23
↓ 1 callersClassAverageMeter
Computes and stores the average and current value
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/utils/utils_logging.py:6
↓ 1 callersClassBackgroundGenerator
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/dataset.py:13
↓ 1 callersClassC
third_part/decalib/utils/util.py:550
↓ 1 callersClassCallBackLogging
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/utils/utils_callbacks.py:51
↓ 1 callersClassCallBackModelCheckpoint
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/utils/utils_callbacks.py:105
↓ 1 callersClassCallBackVerification
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/utils/utils_callbacks.py:12
↓ 1 callersClassConstantInput
models/stylegan2/model.py:292
↓ 1 callersClassCosFace
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/losses.py:14
↓ 1 callersClassCroper
utils/video_preprocess/crop_videos.py:35
↓ 1 callersClassCroper
utils/video_preprocess/crop_videos_inference.py:18
↓ 1 callersClassDataLoaderX
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/dataset.py:41
↓ 1 callersClassDiscriminator
models/stylegan2/model.py:653
↓ 1 callersClassDist2LogitLayer
takes 2 distances, puts through fc layers, spits out value between [0,1] (if use_sigmoid is True)
third_part/PerceptualSimilarity/models/networks_basic.py:180
↓ 1 callersClassDistModel
third_part/PerceptualSimilarity/models/dist_model.py:25
↓ 1 callersClassDownBlock2d
models/styleheat/base_function.py:310
↓ 1 callersClassE4eEncoder
models/e4e/e4e.py:135
↓ 1 callersClassEmbeddingLossBuilder
loss/iterative_inversion_loss.py:11
↓ 1 callersClassEncoder4Editing
models/e4e/e4e.py:56
↓ 1 callersClassExtract3dmm
utils/video_preprocess/extract_3dmm.py:20
↓ 1 callersClassFLAME
borrowed from https://github.com/soubhiksanyal/FLAME_PyTorch/blob/master/FLAME.py Given flame parameters this class generates a differentiabl
third_part/decalib/models/FLAME.py:37
↓ 1 callersClassFLAMETex
FLAME texture: https://github.com/TimoBolkart/TF_FLAME/blob/ade0ab152300ec5f0e8555d6765411555c5ed43d/sample_texture.py#L64 FLAME texture
third_part/decalib/models/FLAME.py:218
↓ 1 callersClassFineDecoderV2
docstring for FineDecoder
models/styleheat/base_function.py:253
↓ 1 callersClassFineEncoder
docstring for Encoder
models/styleheat/base_function.py:193
↓ 1 callersClassFirstBlock2d
Downsampling block for use in encoder.
models/styleheat/base_function.py:290
↓ 1 callersClassFlatten
r"""Flatten the tensor
loss/perceptual.py:367
↓ 1 callersClassFlatten
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/backbones/mobilefacenet.py:11
↓ 1 callersClassGANLoss
loss/gan_loss.py:11
↓ 1 callersClassGDC
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/backbones/mobilefacenet.py:75
↓ 1 callersClassGenerator
models/stylegan2/model.py:364
↓ 1 callersClassGenerator
third_part/decalib/models/decoders.py:19
↓ 1 callersClassHFGI
models/hfgi/hfgi.py:19
↓ 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
third_part/Deep3DFaceRecon_pytorch/util/html.py:6
↓ 1 callersClassHTML
third_part/PerceptualSimilarity/util/html.py:6
↓ 1 callersClassIResNet
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/backbones/iresnet.py:60
↓ 1 callersClassIResNet
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/backbones/iresnet2060.py:64
↓ 1 callersClassImageDataset
data/inference_dataset.py:13
↓ 1 callersClassLPIPS
r"""Creates a criterion that measures Learned Perceptual Image Patch Similarity (LPIPS). Arguments: net_type (str): the network type t
loss/lpips.py:123
↓ 1 callersClassLatentEditor
models/hfgi/editing.py:25
↓ 1 callersClassLinLayers
loss/lpips.py:35
next →1–100 of 196, ranked by callers