MCPcopy Create free account

hub / github.com/MingtaoGuo/StyleSwap / functions

Functions255 in github.com/MingtaoGuo/StyleSwap

⨍ Functions 255◇ Types & classes 59↳ Endpoints 1
FunctionPYBIND11_MODULE
face_model/op/fused_bias_act.cpp:19
FunctionPYBIND11_MODULE
face_model/op/upfirdn2d.cpp:21
Method__getitem__
(self, index)
training/data_loader/dataset_face.py:27
Method__init__
(self, img_path, mask_path, resolution=512, same_prob=0.1)
training/data_loader/dataset_face.py:10
Method__init__
(self, requires_grad=False, pretrained=True)
training/lpips/pretrained_networks.py:6
Method__init__
(self, requires_grad=False, pretrained=True)
training/lpips/pretrained_networks.py:57
Method__init__
(self, requires_grad=False, pretrained=True, num=18)
training/lpips/pretrained_networks.py:139
Method__init__
(self)
training/lpips/lpips.py:126
Method__init__
(self, chn_in, chn_out=1, use_dropout=False)
training/lpips/lpips.py:137
Method__init__
(self, chn_mid=32, use_sigmoid=True)
training/lpips/lpips.py:149
Method__init__
(self, chn_mid=32)
training/lpips/lpips.py:165
Method__init__
(self, use_gpu=True, colorspace='Lab')
training/lpips/lpips.py:178
Method__init__
(self, input_size, num_layers, mode='ir', drop_ratio=0.4, affine=True)
training/loss/model_irse.py:11
Method__init__
(self, in_channel, depth, stride)
training/loss/helpers.py:76
Method__init__
(self, in_channel, depth, stride)
training/loss/helpers.py:98
Method__init__
(self, base_dir='./', device='cuda', ckpt_dict=None)
training/loss/id_loss.py:7
Method__init__
(self, model_path)
face_parsing/face_parsing.py:12
Method__init__
(self, in_chan, out_chan, ks=3, stride=1, padding=1, *args, **kwargs)
face_parsing/bisenet.py:14
Method__init__
(self, in_chan, mid_chan, n_classes, *args, **kwargs)
face_parsing/bisenet.py:37
Method__init__
(self, in_chan, out_chan, *args, **kwargs)
face_parsing/bisenet.py:67
Method__init__
(self, *args, **kwargs)
face_parsing/bisenet.py:92
Method__init__
(self, *args, **kwargs)
face_parsing/bisenet.py:146
Method__init__
(self, in_chan, out_chan, *args, **kwargs)
face_parsing/bisenet.py:180
Method__init__
(self, in_chan, out_chan, stride=1)
face_parsing/resnet.py:21
Method__init__
(self)
face_model/gpen_model.py:17
Method__init__
(self, kernel, factor=2, device='cpu')
face_model/gpen_model.py:36
Method__init__
(self, kernel, factor=2, device='cpu')
face_model/gpen_model.py:58
Method__init__
( self, in_channel, out_channel, kernel_size, stride=1, padding=0, bias=True )
face_model/gpen_model.py:100
Method__init__
( self, in_dim, out_dim, bias=True, bias_init=0, lr_mul=1, activation=None, device='cpu' )
face_model/gpen_model.py:138
Method__init__
(self, negative_slope=0.2)
face_model/gpen_model.py:174
Method__init__
( self, in_channel, out_channel, kernel_size, style_dim, demod
face_model/gpen_model.py:186
Method__init__
(self, isconcat=True)
face_model/gpen_model.py:286
Method__init__
(self, channel, size=4)
face_model/gpen_model.py:304
Method__init__
( self, in_channel, out_channel, kernel_size, style_dim, upsam
face_model/gpen_model.py:317
Method__init__
(self, in_channel, style_dim, upsample=True, blur_kernel=[1, 3, 3, 1], device='cpu')
face_model/gpen_model.py:358
Method__init__
(self, in_channel, style_dim, upsample=True, blur_kernel=[1, 3, 3, 1], device='cpu')
face_model/gpen_model.py:379
Method__init__
( self, size, style_dim, n_mlp, channel_multiplier=2, blur_ker
face_model/gpen_model.py:398
Method__init__
( self, in_channel, out_channel, kernel_size, downsample=False,
face_model/gpen_model.py:592
Method__init__
(self, in_channel, out_channel, blur_kernel=[1, 3, 3, 1], device='cpu')
face_model/gpen_model.py:642
Method__init__
( self, size, style_dim, n_mlp, channel_multiplier=2, blur_ker
face_model/gpen_model.py:662
Method__init__
(self, size, channel_multiplier=2, blur_kernel=[1, 3, 3, 1], narrow=1, device='cpu')
face_model/gpen_model.py:729
Method__init__
(self, requires_grad=False)
face_model/gpen_model.py:795
Method__init__
(self, inplanes, planes, stride=1, downsample=None, groups=1, base_width=64, dilation=1)
face_model/arcface.py:31
Method__init__
(self, channel, negative_slope=0.2, scale=2 ** 0.5, device='cpu')
face_model/op/fused_act.py:80
Method__len__
(self)
training/data_loader/dataset_face.py:24
Method__repr__
(self)
face_model/gpen_model.py:130
Method__repr__
(self)
face_model/gpen_model.py:167
Method__repr__
(self)
face_model/gpen_model.py:235
Method__str__
(self)
align_faces.py:97
Functionadd_gaussian_noise
Add Gaussian noise. Args: img (Numpy array): Input image, shape (h, w, c), range [0, 1], float32. sigma (float): Noise scale (mea
training/data_loader/degradations.py:439
Functionadd_gaussian_noise_pt
Add Gaussian noise (PyTorch version). Args: img (Tensor): Shape (b, c, h, w), range[0, 1], float32. scale (float | Tensor): Noise
training/data_loader/degradations.py:493
Functionadd_poisson_noise
Add poisson noise. Args: img (Numpy array): Input image, shape (h, w, c), range [0, 1], float32. scale (float): Noise scale. Defa
training/data_loader/degradations.py:587
Functionadd_poisson_noise_pt
Add poisson noise to a batch of images (PyTorch version). Args: img (Tensor): Input image, shape (b, c, h, w), range [0, 1], float32.
training/data_loader/degradations.py:658
Functionall_gather
(data)
distributed.py:69
Methodbackward
(ctx, gradgrad_input, gradgrad_bias)
face_model/op/fused_act.py:48
Methodbackward
(ctx, grad_output)
face_model/op/fused_act.py:69
Methodbackward
(ctx, gradgrad_input)
face_model/op/upfirdn2d.py:67
Functioncdf2
Calculate the CDF of the standard bivariate Gaussian distribution. Used in skewed Gaussian distribution. Args: d_matrix (ndarrasy
training/data_loader/degradations.py:66
Functioncircular_lowpass_kernel
2D sinc filter, ref: https://dsp.stackexchange.com/questions/58301/2-d-circularly-symmetric-low-pass-filter Args: cutoff (float): cutoff
training/data_loader/degradations.py:392
Functiondssim
(p0, p1, range=255.)
training/lpips/__init__.py:52
Methodforward
(self, X)
training/lpips/pretrained_networks.py:35
Methodforward
(self, X)
training/lpips/pretrained_networks.py:80
Methodforward
(self, X)
training/lpips/pretrained_networks.py:120
Methodforward
(self, X)
training/lpips/pretrained_networks.py:162
Methodforward
(self, in0, in1, retPerLayer=False, normalize=False)
training/lpips/lpips.py:80
Methodforward
(self, inp)
training/lpips/lpips.py:131
Methodforward
(self, x)
training/lpips/lpips.py:144
Methodforward
(self,d0,d1,eps=0.1)
training/lpips/lpips.py:161
Methodforward
(self, d0, d1, judge)
training/lpips/lpips.py:171
Methodforward
(self, in0, in1, retPerLayer=None)
training/lpips/lpips.py:201
Methodforward
(self, x)
training/loss/model_irse.py:45
Methodforward
(self, input)
training/loss/helpers.py:11
Methodforward
(self, x)
training/loss/helpers.py:65
Methodforward
(self, x)
training/loss/helpers.py:91
Methodforward
(self, x)
training/loss/helpers.py:116
Methodforward
(self, y_hat, y, x)
training/loss/id_loss.py:27
Methodforward
(self, x)
face_parsing/bisenet.py:25
Methodforward
(self, x)
face_parsing/bisenet.py:43
Methodforward
(self, x)
face_parsing/bisenet.py:75
Methodforward
(self, x)
face_parsing/bisenet.py:103
Methodforward
(self, x)
face_parsing/bisenet.py:154
Methodforward
(self, fsp, fcp)
face_parsing/bisenet.py:199
Methodforward
(self, x)
face_parsing/bisenet.py:240
Methodforward
(self, x)
face_parsing/resnet.py:36
Methodforward
(self, x)
face_parsing/resnet.py:71
Methodforward
(self, input)
face_model/gpen_model.py:20
Methodforward
(self, input)
face_model/gpen_model.py:51
Methodforward
(self, input)
face_model/gpen_model.py:73
Methodforward
(self, input)
face_model/gpen_model.py:93
Methodforward
(self, input)
face_model/gpen_model.py:119
Methodforward
(self, input)
face_model/gpen_model.py:157
Methodforward
(self, input)
face_model/gpen_model.py:179
Methodforward
(self, input, style)
face_model/gpen_model.py:241
Methodforward
(self, image, noise=None)
face_model/gpen_model.py:292
Methodforward
(self, input)
face_model/gpen_model.py:309
Methodforward
(self, input, style, noise=None)
face_model/gpen_model.py:348
Methodforward
(self, input, style, skip=None)
face_model/gpen_model.py:367
Methodforward
(self, input, style, skip=None)
face_model/gpen_model.py:387
Methodforward
( self, styles, return_latents=False, inject_index=None, truncation=1,
face_model/gpen_model.py:517
Methodforward
(self, input)
face_model/gpen_model.py:652
← previousnext →101–200 of 255, ranked by callers