Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/OpenTalker/StyleHEAT
/ types & classes
Types & classes
196 in github.com/OpenTalker/StyleHEAT
⨍
Functions
1,100
◇
Types & classes
196
↓ 24 callers
Class
bottleneck_IR
models/e4e/helpers.py:76
↓ 17 callers
Class
AttrDict
Dict as attribute trick.
configs/config.py:9
↓ 9 callers
Class
EqualConv2d
models/stylegan2/model.py:91
↓ 8 callers
Class
ConvLayer
models/stylegan2/model.py:583
↓ 7 callers
Class
ADAIN
models/styleheat/base_function.py:160
↓ 7 callers
Class
NetLinLayer
A single linear layer which does a 1x1 conv
third_part/PerceptualSimilarity/models/networks_basic.py:215
↓ 7 callers
Class
_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 callers
Class
PerceptualLoss
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 callers
Class
ScaledLeakyReLU
models/stylegan2/model.py:166
↓ 5 callers
Class
ConvBlock
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/backbones/mobilefacenet.py:16
↓ 5 callers
Class
EqualLinear
models/stylegan2/model.py:129
↓ 4 callers
Class
DepthWise
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/backbones/mobilefacenet.py:41
↓ 4 callers
Class
Meter
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 callers
Class
Blur
models/stylegan2/model.py:72
↓ 3 callers
Class
FinalBlock2d
Define the output layer
models/styleheat/base_function.py:399
↓ 3 callers
Class
FusedLeakyReLU
models/stylegan2/op/fused_act.py:72
↓ 3 callers
Class
GradualStyleBlock
models/e4e/e4e.py:32
↓ 3 callers
Class
NoWDataset
third_part/decalib/datasets/now.py:12
↓ 3 callers
Class
ResNet
third_part/decalib/models/resnet.py:23
↓ 3 callers
Class
Residual
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/backbones/mobilefacenet.py:63
↓ 3 callers
Class
StyledConv
models/stylegan2/model.py:305
↓ 3 callers
Class
VGGFace2Dataset
third_part/decalib/datasets/vggface.py:12
↓ 2 callers
Class
AFLW2000
third_part/decalib/datasets/aflw2000.py:13
↓ 2 callers
Class
ArcFaceORT
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/onnx_helper.py:15
↓ 2 callers
Class
AudioWarper
models/styleheat/warper.py:48
↓ 2 callers
Class
Bottleneck
A named tuple describing a ResNet block.
models/e4e/helpers.py:22
↓ 2 callers
Class
COCODataset
third_part/decalib/datasets/train_datasets.py:206
↓ 2 callers
Class
CalibrationNet
models/styleheat/calibration_net.py:127
↓ 2 callers
Class
CelebAHQDataset
third_part/decalib/datasets/train_datasets.py:299
↓ 2 callers
Class
Config
r"""Configuration class. This should include every human specifiable hyperparameter values for your training.
configs/config.py:66
↓ 2 callers
Class
ConvUpLayer
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 callers
Class
DoubleConv
(convolution => [BN] => ReLU) * 2
third_part/decalib/models/resnet.py:193
↓ 2 callers
Class
Embedding
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/eval_ijbc.py:54
↓ 2 callers
Class
EthnicityDataset
third_part/decalib/datasets/ethnicity.py:12
↓ 2 callers
Class
FineADAINResBlock2d
Define an Residual block for different types
models/styleheat/base_function.py:375
↓ 2 callers
Class
FineADAINResBlocks
models/styleheat/base_function.py:343
↓ 2 callers
Class
GMM
third_part/Deep3DFaceRecon_pytorch/util/skin_mask.py:9
↓ 2 callers
Class
Jump
models/styleheat/base_function.py:359
↓ 2 callers
Class
LFold
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/eval/verification.py:41
↓ 2 callers
Class
LinearBlock
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/backbones/mobilefacenet.py:29
↓ 2 callers
Class
MappingNet
models/styleheat/warper.py:79
↓ 2 callers
Class
ModulatedConv2d
models/stylegan2/model.py:178
↓ 2 callers
Class
Pytorch3dRasterizer
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 callers
Class
ResBlock
models/stylegan2/model.py:632
↓ 2 callers
Class
ResnetEncoder
third_part/decalib/models/encoders.py:22
↓ 2 callers
Class
StandardRasterizer
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 callers
Class
TempVideoDataset
data/inference_dataset.py:60
↓ 2 callers
Class
ToRGB
models/stylegan2/model.py:342
↓ 2 callers
Class
UpBlock2d
models/styleheat/base_function.py:328
↓ 2 callers
Class
VGGFace2HQDataset
third_part/decalib/datasets/vggface.py:124
↓ 2 callers
Class
VoxelDataset
third_part/decalib/datasets/train_datasets.py:51
↓ 2 callers
Class
WarpingNet
models/styleheat/warper.py:107
↓ 1 callers
Class
ADAINDecoder
docstring for ADAINDecoder
models/styleheat/base_function.py:61
↓ 1 callers
Class
ADAINDecoderBlock
models/styleheat/base_function.py:111
↓ 1 callers
Class
ADAINEncoder
models/styleheat/base_function.py:39
↓ 1 callers
Class
ADAINEncoderBlock
models/styleheat/base_function.py:92
↓ 1 callers
Class
ADAINHourglass
models/styleheat/base_function.py:28
↓ 1 callers
Class
AlexNet
loss/lpips.py:89
↓ 1 callers
Class
AlignedDataSet
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/onnx_ijbc.py:28
↓ 1 callers
Class
ArcFace
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/losses.py:29
↓ 1 callers
Class
AudioDataset
data/audio_dataset.py:23
↓ 1 callers
Class
AverageMeter
Computes and stores the average and current value
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/utils/utils_logging.py:6
↓ 1 callers
Class
BackgroundGenerator
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/dataset.py:13
↓ 1 callers
Class
C
third_part/decalib/utils/util.py:550
↓ 1 callers
Class
CallBackLogging
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/utils/utils_callbacks.py:51
↓ 1 callers
Class
CallBackModelCheckpoint
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/utils/utils_callbacks.py:105
↓ 1 callers
Class
CallBackVerification
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/utils/utils_callbacks.py:12
↓ 1 callers
Class
ConstantInput
models/stylegan2/model.py:292
↓ 1 callers
Class
CosFace
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/losses.py:14
↓ 1 callers
Class
Croper
utils/video_preprocess/crop_videos.py:35
↓ 1 callers
Class
Croper
utils/video_preprocess/crop_videos_inference.py:18
↓ 1 callers
Class
DataLoaderX
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/dataset.py:41
↓ 1 callers
Class
Discriminator
models/stylegan2/model.py:653
↓ 1 callers
Class
Dist2LogitLayer
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 callers
Class
DistModel
third_part/PerceptualSimilarity/models/dist_model.py:25
↓ 1 callers
Class
DownBlock2d
models/styleheat/base_function.py:310
↓ 1 callers
Class
E4eEncoder
models/e4e/e4e.py:135
↓ 1 callers
Class
EmbeddingLossBuilder
loss/iterative_inversion_loss.py:11
↓ 1 callers
Class
Encoder4Editing
models/e4e/e4e.py:56
↓ 1 callers
Class
Extract3dmm
utils/video_preprocess/extract_3dmm.py:20
↓ 1 callers
Class
FLAME
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 callers
Class
FLAMETex
FLAME texture: https://github.com/TimoBolkart/TF_FLAME/blob/ade0ab152300ec5f0e8555d6765411555c5ed43d/sample_texture.py#L64 FLAME texture
third_part/decalib/models/FLAME.py:218
↓ 1 callers
Class
FineDecoderV2
docstring for FineDecoder
models/styleheat/base_function.py:253
↓ 1 callers
Class
FineEncoder
docstring for Encoder
models/styleheat/base_function.py:193
↓ 1 callers
Class
FirstBlock2d
Downsampling block for use in encoder.
models/styleheat/base_function.py:290
↓ 1 callers
Class
Flatten
r"""Flatten the tensor
loss/perceptual.py:367
↓ 1 callers
Class
Flatten
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/backbones/mobilefacenet.py:11
↓ 1 callers
Class
GANLoss
loss/gan_loss.py:11
↓ 1 callers
Class
GDC
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/backbones/mobilefacenet.py:75
↓ 1 callers
Class
Generator
models/stylegan2/model.py:364
↓ 1 callers
Class
Generator
third_part/decalib/models/decoders.py:19
↓ 1 callers
Class
HFGI
models/hfgi/hfgi.py:19
↓ 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
third_part/Deep3DFaceRecon_pytorch/util/html.py:6
↓ 1 callers
Class
HTML
third_part/PerceptualSimilarity/util/html.py:6
↓ 1 callers
Class
IResNet
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/backbones/iresnet.py:60
↓ 1 callers
Class
IResNet
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/backbones/iresnet2060.py:64
↓ 1 callers
Class
ImageDataset
data/inference_dataset.py:13
↓ 1 callers
Class
LPIPS
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 callers
Class
LatentEditor
models/hfgi/editing.py:25
↓ 1 callers
Class
LinLayers
loss/lpips.py:35
next →
1–100 of 196, ranked by callers