SYMBOL INDEX (1277 symbols across 125 files) FILE: docs/static/js/bulma-carousel.js function __webpack_require__ (line 16) | function __webpack_require__(moduleId) { function detectSupportsPassive (line 189) | function detectSupportsPassive() { function defineProperties (line 273) | function defineProperties(target, props) { for (var i = 0; i < props.len... function _toConsumableArray (line 275) | function _toConsumableArray(arr) { if (Array.isArray(arr)) { for (var i ... function _classCallCheck (line 277) | function _classCallCheck(instance, Constructor) { if (!(instance instanc... function EventEmitter (line 280) | function EventEmitter() { function defineProperties (line 324) | function defineProperties(target, props) { for (var i = 0; i < props.len... function _classCallCheck (line 326) | function _classCallCheck(instance, Constructor) { if (!(instance instanc... function Coordinate (line 329) | function Coordinate() { function defineProperties (line 473) | function defineProperties(target, props) { for (var i = 0; i < props.len... function _defineProperty (line 475) | function _defineProperty(obj, key, value) { if (key in obj) { Object.def... function _classCallCheck (line 477) | function _classCallCheck(instance, Constructor) { if (!(instance instanc... function _possibleConstructorReturn (line 479) | function _possibleConstructorReturn(self, call) { if (!self) { throw new... function _inherits (line 481) | function _inherits(subClass, superClass) { if (typeof superClass !== "fu... function bulmaCarousel (line 504) | function bulmaCarousel(selector) { function _toConsumableArray (line 983) | function _toConsumableArray(arr) { if (Array.isArray(arr)) { for (var i ... function defineProperties (line 1019) | function defineProperties(target, props) { for (var i = 0; i < props.len... function _classCallCheck (line 1021) | function _classCallCheck(instance, Constructor) { if (!(instance instanc... function _possibleConstructorReturn (line 1023) | function _possibleConstructorReturn(self, call) { if (!self) { throw new... function _inherits (line 1025) | function _inherits(subClass, superClass) { if (typeof superClass !== "fu... function Autoplay (line 1042) | function Autoplay(slider) { function defineProperties (line 1209) | function defineProperties(target, props) { for (var i = 0; i < props.len... function _classCallCheck (line 1211) | function _classCallCheck(instance, Constructor) { if (!(instance instanc... function Breakpoints (line 1216) | function Breakpoints(slider) { function defineProperties (line 1347) | function defineProperties(target, props) { for (var i = 0; i < props.len... function _toConsumableArray (line 1349) | function _toConsumableArray(arr) { if (Array.isArray(arr)) { for (var i ... function _classCallCheck (line 1351) | function _classCallCheck(instance, Constructor) { if (!(instance instanc... function Infinite (line 1354) | function Infinite(slider) { function defineProperties (line 1427) | function defineProperties(target, props) { for (var i = 0; i < props.len... function _classCallCheck (line 1429) | function _classCallCheck(instance, Constructor) { if (!(instance instanc... function Loop (line 1434) | function Loop(slider) { function defineProperties (line 1494) | function defineProperties(target, props) { for (var i = 0; i < props.len... function _classCallCheck (line 1496) | function _classCallCheck(instance, Constructor) { if (!(instance instanc... function Navigation (line 1502) | function Navigation(slider) { function defineProperties (line 1644) | function defineProperties(target, props) { for (var i = 0; i < props.len... function _classCallCheck (line 1646) | function _classCallCheck(instance, Constructor) { if (!(instance instanc... function Pagination (line 1653) | function Pagination(slider) { function defineProperties (line 1804) | function defineProperties(target, props) { for (var i = 0; i < props.len... function _classCallCheck (line 1806) | function _classCallCheck(instance, Constructor) { if (!(instance instanc... function Swipe (line 1812) | function Swipe(slider) { function defineProperties (line 1946) | function defineProperties(target, props) { for (var i = 0; i < props.len... function _classCallCheck (line 1948) | function _classCallCheck(instance, Constructor) { if (!(instance instanc... function Transitioner (line 1954) | function Transitioner(slider) { function defineProperties (line 2047) | function defineProperties(target, props) { for (var i = 0; i < props.len... function _classCallCheck (line 2049) | function _classCallCheck(instance, Constructor) { if (!(instance instanc... function Fade (line 2054) | function Fade(transitioner, slider) { function defineProperties (line 2182) | function defineProperties(target, props) { for (var i = 0; i < props.len... function _classCallCheck (line 2184) | function _classCallCheck(instance, Constructor) { if (!(instance instanc... function Translate (line 2190) | function Translate(transitioner, slider) { FILE: docs/static/js/bulma-slider.js function __webpack_require__ (line 16) | function __webpack_require__(moduleId) { function defineProperties (line 86) | function defineProperties(target, props) { for (var i = 0; i < props.len... function _classCallCheck (line 90) | function _classCallCheck(instance, Constructor) { if (!(instance instanc... function _possibleConstructorReturn (line 92) | function _possibleConstructorReturn(self, call) { if (!self) { throw new... function _inherits (line 94) | function _inherits(subClass, superClass) { if (typeof superClass !== "fu... function bulmaSlider (line 105) | function bulmaSlider(selector) { function defineProperties (line 281) | function defineProperties(target, props) { for (var i = 0; i < props.len... function _classCallCheck (line 283) | function _classCallCheck(instance, Constructor) { if (!(instance instanc... function EventEmitter (line 286) | function EventEmitter() { FILE: inference.py function main (line 31) | def main(): function datagen (line 279) | def datagen(frames, mels, full_frames, frames_pil, cox): FILE: models/DNet.py class DNet (line 13) | class DNet(nn.Module): method __init__ (line 14) | def __init__(self): method forward (line 20) | def forward(self, input_image, driving_source, stage=None): class MappingNet (line 30) | class MappingNet(nn.Module): method __init__ (line 31) | def __init__(self, coeff_nc=73, descriptor_nc=256, layer=3): method forward (line 48) | def forward(self, input_3dmm): class WarpingNet (line 56) | class WarpingNet(nn.Module): method __init__ (line 57) | def __init__( method forward (line 83) | def forward(self, input_image, descriptor): class EditingNet (line 93) | class EditingNet(nn.Module): method __init__ (line 94) | def __init__( method forward (line 114) | def forward(self, input_image, warp_image, descriptor): FILE: models/ENet.py class ENet (line 8) | class ENet(nn.Module): method __init__ (line 9) | def __init__( method forward (line 82) | def forward(self, audio_sequences, face_sequences, gt_sequences): FILE: models/LNet.py class Visual_Encoder (line 10) | class Visual_Encoder(nn.Module): method __init__ (line 11) | def __init__(self, image_nc, ngf, img_f, layers, norm_layer=nn.BatchNo... method forward (line 30) | def forward(self, maskGT, ref): class Decoder (line 46) | class Decoder(nn.Module): method __init__ (line 47) | def __init__(self, image_nc, feature_nc, ngf, img_f, layers, num_block... method forward (line 67) | def forward(self, x, z): class LNet (line 80) | class LNet(nn.Module): method __init__ (line 81) | def __init__( method forward (line 122) | def forward(self, audio_sequences, face_sequences): FILE: models/__init__.py function _load (line 7) | def _load(checkpoint_path): function load_checkpoint (line 12) | def load_checkpoint(path, model): function load_network (line 25) | def load_network(args): function load_DNet (line 32) | def load_DNet(args): FILE: models/base_blocks.py class Conv2d (line 12) | class Conv2d(nn.Module): method __init__ (line 13) | def __init__(self, cin, cout, kernel_size, stride, padding, residual=F... method forward (line 22) | def forward(self, x): class ResBlock (line 29) | class ResBlock(nn.Module): method __init__ (line 30) | def __init__(self, in_channels, out_channels, mode='down'): method forward (line 40) | def forward(self, x): class LayerNorm2d (line 52) | class LayerNorm2d(nn.Module): method __init__ (line 53) | def __init__(self, n_out, affine=True): method forward (line 62) | def forward(self, x): function spectral_norm (line 72) | def spectral_norm(module, use_spect=True): class FirstBlock2d (line 79) | class FirstBlock2d(nn.Module): method __init__ (line 80) | def __init__(self, input_nc, output_nc, norm_layer=nn.BatchNorm2d, non... method forward (line 90) | def forward(self, x): class DownBlock2d (line 95) | class DownBlock2d(nn.Module): method __init__ (line 96) | def __init__(self, input_nc, output_nc, norm_layer=nn.BatchNorm2d, non... method forward (line 107) | def forward(self, x): class UpBlock2d (line 112) | class UpBlock2d(nn.Module): method __init__ (line 113) | def __init__(self, input_nc, output_nc, norm_layer=nn.BatchNorm2d, non... method forward (line 122) | def forward(self, x): class ADAIN (line 127) | class ADAIN(nn.Module): method __init__ (line 128) | def __init__(self, norm_nc, feature_nc): method forward (line 143) | def forward(self, x, feature): class FineADAINResBlock2d (line 160) | class FineADAINResBlock2d(nn.Module): method __init__ (line 164) | def __init__(self, input_nc, feature_nc, norm_layer=nn.BatchNorm2d, no... method forward (line 173) | def forward(self, x, z): class FineADAINResBlocks (line 180) | class FineADAINResBlocks(nn.Module): method __init__ (line 181) | def __init__(self, num_block, input_nc, feature_nc, norm_layer=nn.Batc... method forward (line 188) | def forward(self, x, z): class ADAINEncoderBlock (line 195) | class ADAINEncoderBlock(nn.Module): method __init__ (line 196) | def __init__(self, input_nc, output_nc, feature_nc, nonlinearity=nn.Le... method forward (line 209) | def forward(self, x, z): class ADAINDecoderBlock (line 215) | class ADAINDecoderBlock(nn.Module): method __init__ (line 216) | def __init__(self, input_nc, output_nc, hidden_nc, feature_nc, use_tra... method forward (line 243) | def forward(self, x, z): method shortcut (line 250) | def shortcut(self, x, z): class FineEncoder (line 255) | class FineEncoder(nn.Module): method __init__ (line 257) | def __init__(self, image_nc, ngf, img_f, layers, norm_layer=nn.BatchNo... method forward (line 268) | def forward(self, x): class FineDecoder (line 278) | class FineDecoder(nn.Module): method __init__ (line 280) | def __init__(self, image_nc, feature_nc, ngf, img_f, layers, num_block... method forward (line 295) | def forward(self, x, z): class ADAINEncoder (line 308) | class ADAINEncoder(nn.Module): method __init__ (line 309) | def __init__(self, image_nc, pose_nc, ngf, img_f, layers, nonlinearity... method forward (line 320) | def forward(self, x, z): class ADAINDecoder (line 330) | class ADAINDecoder(nn.Module): method __init__ (line 332) | def __init__(self, pose_nc, ngf, img_f, encoder_layers, decoder_layers... method forward (line 348) | def forward(self, x, z): class ADAINHourglass (line 357) | class ADAINHourglass(nn.Module): method __init__ (line 358) | def __init__(self, image_nc, pose_nc, ngf, img_f, encoder_layers, deco... method forward (line 364) | def forward(self, x, z): class FineADAINLama (line 368) | class FineADAINLama(nn.Module): method __init__ (line 369) | def __init__(self, input_nc, feature_nc, norm_layer=nn.BatchNorm2d, no... method forward (line 382) | def forward(self, x, z): class FFCResnetBlock (line 389) | class FFCResnetBlock(nn.Module): method __init__ (line 390) | def __init__(self, dim, feature_dim, padding_type='reflect', norm_laye... method forward (line 397) | def forward(self, x, z): class FFCADAINResBlocks (line 414) | class FFCADAINResBlocks(nn.Module): method __init__ (line 415) | def __init__(self, num_block, input_nc, feature_nc, norm_layer=nn.Batc... method forward (line 422) | def forward(self, x, z): class Jump (line 429) | class Jump(nn.Module): method __init__ (line 430) | def __init__(self, input_nc, norm_layer=nn.BatchNorm2d, nonlinearity=n... method forward (line 439) | def forward(self, x): class FinalBlock2d (line 444) | class FinalBlock2d(nn.Module): method __init__ (line 445) | def __init__(self, input_nc, output_nc, use_spect=False, tanh_or_sigmo... method forward (line 455) | def forward(self, x): class ModulatedConv2d (line 460) | class ModulatedConv2d(nn.Module): method __init__ (line 461) | def __init__(self, method forward (line 487) | def forward(self, x, style): method __repr__ (line 510) | def __repr__(self): class StyleConv (line 515) | class StyleConv(nn.Module): method __init__ (line 516) | def __init__(self, in_channels, out_channels, kernel_size, num_style_f... method forward (line 524) | def forward(self, x, style, noise=None): class ToRGB (line 539) | class ToRGB(nn.Module): method __init__ (line 540) | def __init__(self, in_channels, num_style_feat, upsample=True): method forward (line 547) | def forward(self, x, style, skip=None): FILE: models/ffc.py class SELayer (line 11) | class SELayer(nn.Module): method __init__ (line 12) | def __init__(self, channel, reduction=16): method forward (line 22) | def forward(self, x): class FFCSE_block (line 30) | class FFCSE_block(nn.Module): method __init__ (line 31) | def __init__(self, channels, ratio_g): method forward (line 47) | def forward(self, x): class FourierUnit (line 62) | class FourierUnit(nn.Module): method __init__ (line 64) | def __init__(self, in_channels, out_channels, groups=1, spatial_scale_... method forward (line 89) | def forward(self, x): class SpectralTransform (line 129) | class SpectralTransform(nn.Module): method __init__ (line 130) | def __init__(self, in_channels, out_channels, stride=1, groups=1, enab... method forward (line 154) | def forward(self, x): class FFC (line 176) | class FFC(nn.Module): method __init__ (line 178) | def __init__(self, in_channels, out_channels, kernel_size, method forward (line 213) | def forward(self, x): FILE: models/transformer.py class GELU (line 11) | class GELU(nn.Module): method __init__ (line 12) | def __init__(self): method forward (line 14) | def forward(self, x): function pair (line 19) | def pair(t): class PreNorm (line 24) | class PreNorm(nn.Module): method __init__ (line 25) | def __init__(self, dim, fn): method forward (line 29) | def forward(self, x, **kwargs): class DualPreNorm (line 32) | class DualPreNorm(nn.Module): method __init__ (line 33) | def __init__(self, dim, fn): method forward (line 38) | def forward(self, x, y, **kwargs): class FeedForward (line 41) | class FeedForward(nn.Module): method __init__ (line 42) | def __init__(self, dim, hidden_dim, dropout = 0.): method forward (line 51) | def forward(self, x): class Attention (line 54) | class Attention(nn.Module): method __init__ (line 55) | def __init__(self, dim, heads = 8, dim_head = 64, dropout = 0.): method forward (line 75) | def forward(self, x, y): class Transformer (line 89) | class Transformer(nn.Module): method __init__ (line 90) | def __init__(self, dim, depth, heads, dim_head, mlp_dim, dropout = 0.): method forward (line 100) | def forward(self, x, y): # x is the cropped, y is the foreign reference class RETURNX (line 114) | class RETURNX(nn.Module): method __init__ (line 115) | def __init__(self,): method forward (line 118) | def forward(self, x, y): # x is the cropped, y is the foreign reference FILE: predict.py class Predictor (line 49) | class Predictor(BasePredictor): method setup (line 50) | def setup(self) -> None: method predict (line 77) | def predict( function datagen (line 440) | def datagen(frames, mels, full_frames, args, cox): FILE: third_part/GFPGAN/gfpgan/archs/arcface_arch.py function conv3x3 (line 5) | def conv3x3(inplanes, outplanes, stride=1): class BasicBlock (line 16) | class BasicBlock(nn.Module): method __init__ (line 27) | def __init__(self, inplanes, planes, stride=1, downsample=None): method forward (line 37) | def forward(self, x): class IRBlock (line 56) | class IRBlock(nn.Module): method __init__ (line 68) | def __init__(self, inplanes, planes, stride=1, downsample=None, use_se... method forward (line 82) | def forward(self, x): class Bottleneck (line 103) | class Bottleneck(nn.Module): method __init__ (line 114) | def __init__(self, inplanes, planes, stride=1, downsample=None): method forward (line 126) | def forward(self, x): class SEBlock (line 149) | class SEBlock(nn.Module): method __init__ (line 157) | def __init__(self, channel, reduction=16): method forward (line 164) | def forward(self, x): class ResNetArcFace (line 172) | class ResNetArcFace(nn.Module): method __init__ (line 183) | def __init__(self, block, layers, use_se=True): method _make_layer (line 214) | def _make_layer(self, block, planes, num_blocks, stride=1): method forward (line 229) | def forward(self, x): FILE: third_part/GFPGAN/gfpgan/archs/gfpgan_bilinear_arch.py class StyleGAN2GeneratorBilinearSFT (line 12) | class StyleGAN2GeneratorBilinearSFT(StyleGAN2GeneratorBilinear): method __init__ (line 28) | def __init__(self, method forward (line 45) | def forward(self, class GFPGANBilinear (line 132) | class GFPGANBilinear(nn.Module): method __init__ (line 156) | def __init__( method forward (line 264) | def forward(self, x, return_latents=False, return_rgb=True, randomize_... FILE: third_part/GFPGAN/gfpgan/archs/gfpganv1_arch.py class StyleGAN2GeneratorSFT (line 12) | class StyleGAN2GeneratorSFT(StyleGAN2Generator): method __init__ (line 27) | def __init__(self, method forward (line 46) | def forward(self, class ConvUpLayer (line 132) | class ConvUpLayer(nn.Module): method __init__ (line 146) | def __init__(self, method forward (line 180) | def forward(self, x): class ResUpBlock (line 197) | class ResUpBlock(nn.Module): method __init__ (line 205) | def __init__(self, in_channels, out_channels): method forward (line 212) | def forward(self, x): class GFPGANv1 (line 221) | class GFPGANv1(nn.Module): method __init__ (line 243) | def __init__( method forward (line 353) | def forward(self, x, return_latents=False, return_rgb=True, randomize_... class FacialComponentDiscriminator (line 405) | class FacialComponentDiscriminator(nn.Module): method __init__ (line 409) | def __init__(self): method forward (line 419) | def forward(self, x, return_feats=False): FILE: third_part/GFPGAN/gfpgan/archs/gfpganv1_clean_arch.py class StyleGAN2GeneratorCSFT (line 11) | class StyleGAN2GeneratorCSFT(StyleGAN2GeneratorClean): method __init__ (line 25) | def __init__(self, out_size, num_style_feat=512, num_mlp=8, channel_mu... method forward (line 34) | def forward(self, class ResBlock (line 120) | class ResBlock(nn.Module): method __init__ (line 129) | def __init__(self, in_channels, out_channels, mode='down'): method forward (line 140) | def forward(self, x): class GFPGANv1Clean (line 153) | class GFPGANv1Clean(nn.Module): method __init__ (line 174) | def __init__( method forward (line 277) | def forward(self, x, return_latents=False, return_rgb=True, randomize_... FILE: third_part/GFPGAN/gfpgan/archs/stylegan2_bilinear_arch.py class NormStyleCode (line 10) | class NormStyleCode(nn.Module): method forward (line 12) | def forward(self, x): class EqualLinear (line 24) | class EqualLinear(nn.Module): method __init__ (line 38) | def __init__(self, in_channels, out_channels, bias=True, bias_init_val... method forward (line 55) | def forward(self, x): method __repr__ (line 67) | def __repr__(self): class ModulatedConv2d (line 72) | class ModulatedConv2d(nn.Module): method __init__ (line 90) | def __init__(self, method forward (line 120) | def forward(self, x, style): method __repr__ (line 155) | def __repr__(self): class StyleConv (line 162) | class StyleConv(nn.Module): method __init__ (line 175) | def __init__(self, method forward (line 195) | def forward(self, x, style, noise=None): class ToRGB (line 208) | class ToRGB(nn.Module): method __init__ (line 217) | def __init__(self, in_channels, num_style_feat, upsample=True, interpo... method forward (line 235) | def forward(self, x, style, skip=None): class ConstantInput (line 256) | class ConstantInput(nn.Module): method __init__ (line 264) | def __init__(self, num_channel, size): method forward (line 268) | def forward(self, batch): class StyleGAN2GeneratorBilinear (line 274) | class StyleGAN2GeneratorBilinear(nn.Module): method __init__ (line 287) | def __init__(self, method make_noise (line 369) | def make_noise(self): method get_latent (line 380) | def get_latent(self, x): method mean_latent (line 383) | def mean_latent(self, num_latent): method forward (line 388) | def forward(self, class ScaledLeakyReLU (line 465) | class ScaledLeakyReLU(nn.Module): method __init__ (line 472) | def __init__(self, negative_slope=0.2): method forward (line 476) | def forward(self, x): class EqualConv2d (line 481) | class EqualConv2d(nn.Module): method __init__ (line 496) | def __init__(self, in_channels, out_channels, kernel_size, stride=1, p... method forward (line 511) | def forward(self, x): method __repr__ (line 522) | def __repr__(self): class ConvLayer (line 530) | class ConvLayer(nn.Sequential): method __init__ (line 543) | def __init__(self, class ResBlock (line 579) | class ResBlock(nn.Module): method __init__ (line 587) | def __init__(self, in_channels, out_channels, interpolation_mode='bili... method forward (line 608) | def forward(self, x): FILE: third_part/GFPGAN/gfpgan/archs/stylegan2_clean_arch.py class NormStyleCode (line 10) | class NormStyleCode(nn.Module): method forward (line 12) | def forward(self, x): class ModulatedConv2d (line 24) | class ModulatedConv2d(nn.Module): method __init__ (line 39) | def __init__(self, method forward (line 65) | def forward(self, x, style): method __repr__ (line 101) | def __repr__(self): class StyleConv (line 106) | class StyleConv(nn.Module): method __init__ (line 118) | def __init__(self, in_channels, out_channels, kernel_size, num_style_f... method forward (line 126) | def forward(self, x, style, noise=None): class ToRGB (line 141) | class ToRGB(nn.Module): method __init__ (line 150) | def __init__(self, in_channels, num_style_feat, upsample=True): method forward (line 157) | def forward(self, x, style, skip=None): class ConstantInput (line 177) | class ConstantInput(nn.Module): method __init__ (line 185) | def __init__(self, num_channel, size): method forward (line 189) | def forward(self, batch): class StyleGAN2GeneratorClean (line 195) | class StyleGAN2GeneratorClean(nn.Module): method __init__ (line 206) | def __init__(self, out_size, num_style_feat=512, num_mlp=8, channel_mu... method make_noise (line 279) | def make_noise(self): method get_latent (line 290) | def get_latent(self, x): method mean_latent (line 293) | def mean_latent(self, num_latent): method forward (line 298) | def forward(self, FILE: third_part/GFPGAN/gfpgan/data/ffhq_degradation_dataset.py class FFHQDegradationDataset (line 17) | class FFHQDegradationDataset(data.Dataset): method __init__ (line 32) | def __init__(self, opt): method color_jitter (line 91) | def color_jitter(img, shift): method color_jitter_pt (line 99) | def color_jitter_pt(img, brightness, contrast, saturation, hue): method get_component_coordinates (line 120) | def get_component_coordinates(self, index, status): method __getitem__ (line 145) | def __getitem__(self, index): method __len__ (line 229) | def __len__(self): FILE: third_part/GFPGAN/gfpgan/models/gfpgan_model.py class GFPGANModel (line 19) | class GFPGANModel(BaseModel): method __init__ (line 22) | def __init__(self, opt): method init_training_settings (line 42) | def init_training_settings(self): method setup_optimizers (line 151) | def setup_optimizers(self): method feed_data (line 202) | def feed_data(self, data): method construct_img_pyramid (line 226) | def construct_img_pyramid(self): method get_roi_regions (line 235) | def get_roi_regions(self, eye_out_size=80, mouth_out_size=120): method _gram_mat (line 267) | def _gram_mat(self, x): method gray_resize_for_identity (line 282) | def gray_resize_for_identity(self, out, size=128): method optimize_parameters (line 288) | def optimize_parameters(self, current_iter): method test (line 474) | def test(self): method dist_validation (line 486) | def dist_validation(self, dataloader, current_iter, tb_logger, save_img): method nondist_validation (line 490) | def nondist_validation(self, dataloader, current_iter, tb_logger, save... method _log_validation_metric_values (line 555) | def _log_validation_metric_values(self, current_iter, dataset_name, tb... method save (line 570) | def save(self, epoch, current_iter): FILE: third_part/GFPGAN/gfpgan/utils.py class GFPGANer (line 16) | class GFPGANer(): method __init__ (line 32) | def __init__(self, model_path, upscale=2, arch='clean', channel_multip... method enhance (line 97) | def enhance(self, img, has_aligned=False, only_center_face=False, past... FILE: third_part/GPEN/align_faces.py function _umeyama (line 25) | def _umeyama(src, dst, estimate_scale=True, scale=1.0): class FaceWarpException (line 96) | class FaceWarpException(Exception): method __str__ (line 97) | def __str__(self): function get_reference_facial_points (line 102) | def get_reference_facial_points(output_size=None, function get_affine_transform_matrix (line 187) | def get_affine_transform_matrix(src_pts, dst_pts): function warp_and_crop_face (line 210) | def warp_and_crop_face(src_img, FILE: third_part/GPEN/face_detect/data/data_augment.py function _crop (line 7) | def _crop(image, boxes, labels, landm, img_dim): function _distort (line 80) | def _distort(image): function _expand (line 141) | def _expand(image, boxes, fill, p): function _mirror (line 167) | def _mirror(image, boxes, landms): function _pad_to_square (line 189) | def _pad_to_square(image, rgb_mean, pad_image_flag): function _resize_subtract_mean (line 200) | def _resize_subtract_mean(image, insize, rgb_mean): class preproc (line 209) | class preproc(object): method __init__ (line 211) | def __init__(self, img_dim, rgb_means): method __call__ (line 215) | def __call__(self, image, targets): FILE: third_part/GPEN/face_detect/data/wider_face.py class WiderFaceDetection (line 9) | class WiderFaceDetection(data.Dataset): method __init__ (line 10) | def __init__(self, txt_path, preproc=None): method __len__ (line 37) | def __len__(self): method __getitem__ (line 40) | def __getitem__(self, index): function detection_collate (line 79) | def detection_collate(batch): FILE: third_part/GPEN/face_detect/facemodels/net.py function conv_bn (line 9) | def conv_bn(inp, oup, stride = 1, leaky = 0): function conv_bn_no_relu (line 16) | def conv_bn_no_relu(inp, oup, stride): function conv_bn1X1 (line 22) | def conv_bn1X1(inp, oup, stride, leaky=0): function conv_dw (line 29) | def conv_dw(inp, oup, stride, leaky=0.1): class SSH (line 40) | class SSH(nn.Module): method __init__ (line 41) | def __init__(self, in_channel, out_channel): method forward (line 55) | def forward(self, input): class FPN (line 68) | class FPN(nn.Module): method __init__ (line 69) | def __init__(self,in_channels_list,out_channels): method forward (line 81) | def forward(self, input): class MobileNetV1 (line 102) | class MobileNetV1(nn.Module): method __init__ (line 103) | def __init__(self): method forward (line 128) | def forward(self, x): FILE: third_part/GPEN/face_detect/facemodels/retinaface.py class ClassHead (line 14) | class ClassHead(nn.Module): method __init__ (line 15) | def __init__(self,inchannels=512,num_anchors=3): method forward (line 20) | def forward(self,x): class BboxHead (line 26) | class BboxHead(nn.Module): method __init__ (line 27) | def __init__(self,inchannels=512,num_anchors=3): method forward (line 31) | def forward(self,x): class LandmarkHead (line 37) | class LandmarkHead(nn.Module): method __init__ (line 38) | def __init__(self,inchannels=512,num_anchors=3): method forward (line 42) | def forward(self,x): class RetinaFace (line 48) | class RetinaFace(nn.Module): method __init__ (line 49) | def __init__(self, cfg = None, phase = 'train'): method _make_class_head (line 89) | def _make_class_head(self,fpn_num=3,inchannels=64,anchor_num=2): method _make_bbox_head (line 95) | def _make_bbox_head(self,fpn_num=3,inchannels=64,anchor_num=2): method _make_landmark_head (line 101) | def _make_landmark_head(self,fpn_num=3,inchannels=64,anchor_num=2): method forward (line 107) | def forward(self,inputs): FILE: third_part/GPEN/face_detect/layers/functions/prior_box.py class PriorBox (line 7) | class PriorBox(object): method __init__ (line 8) | def __init__(self, cfg, image_size=None, phase='train'): method forward (line 17) | def forward(self): FILE: third_part/GPEN/face_detect/layers/modules/multibox_loss.py class MultiBoxLoss (line 9) | class MultiBoxLoss(nn.Module): method __init__ (line 32) | def __init__(self, num_classes, overlap_thresh, prior_for_matching, bk... method forward (line 44) | def forward(self, predictions, priors, targets): FILE: third_part/GPEN/face_detect/retinaface_detection.py class RetinaFaceDetection (line 19) | class RetinaFaceDetection(object): method __init__ (line 20) | def __init__(self, base_dir, device='cuda', network='RetinaFace-R50'): method check_keys (line 32) | def check_keys(self, pretrained_state_dict): method remove_prefix (line 41) | def remove_prefix(self, state_dict, prefix): method load_model (line 46) | def load_model(self, load_to_cpu=False): method detect (line 60) | def detect(self, img_raw, resize=1, confidence_threshold=0.9, nms_thre... method detect_tensor (line 133) | def detect_tensor(self, img, resize=1, confidence_threshold=0.9, nms_t... FILE: third_part/GPEN/face_detect/utils/box_utils.py function point_form (line 5) | def point_form(boxes): function center_size (line 17) | def center_size(boxes): function intersect (line 29) | def intersect(box_a, box_b): function jaccard (line 50) | def jaccard(box_a, box_b): function matrix_iou (line 71) | def matrix_iou(a, b): function matrix_iof (line 84) | def matrix_iof(a, b): function match (line 96) | def match(threshold, truths, priors, variances, labels, landms, loc_t, c... function encode (line 156) | def encode(matched, priors, variances): function encode_landm (line 179) | def encode_landm(matched, priors, variances): function decode (line 209) | def decode(loc, priors, variances): function decode_landm (line 229) | def decode_landm(pre, priors, variances): function log_sum_exp (line 250) | def log_sum_exp(x): function nms (line 264) | def nms(boxes, scores, overlap=0.5, top_k=200): FILE: third_part/GPEN/face_detect/utils/nms/py_cpu_nms.py function py_cpu_nms (line 10) | def py_cpu_nms(dets, thresh): FILE: third_part/GPEN/face_detect/utils/timer.py class Timer (line 11) | class Timer(object): method __init__ (line 13) | def __init__(self): method tic (line 20) | def tic(self): method toc (line 25) | def toc(self, average=True): method clear (line 35) | def clear(self): FILE: third_part/GPEN/face_model/face_gan.py class FaceGAN (line 15) | class FaceGAN(object): method __init__ (line 16) | def __init__(self, base_dir='./', size=512, model=None, channel_multip... method load_model (line 24) | def load_model(self, channel_multiplier=2, narrow=1): method process (line 31) | def process(self, img): method img2tensor (line 42) | def img2tensor(self, img): method tensor2img (line 49) | def tensor2img(self, img_t, pmax=255.0, imtype=np.uint8): FILE: third_part/GPEN/face_model/gpen_model.py class PixelNorm (line 18) | class PixelNorm(nn.Module): method __init__ (line 19) | def __init__(self): method forward (line 22) | def forward(self, input): function make_kernel (line 26) | def make_kernel(k): class Upsample (line 37) | class Upsample(nn.Module): method __init__ (line 38) | def __init__(self, kernel, factor=2, device='cpu'): method forward (line 53) | def forward(self, input): class Downsample (line 59) | class Downsample(nn.Module): method __init__ (line 60) | def __init__(self, kernel, factor=2, device='cpu'): method forward (line 75) | def forward(self, input): class Blur (line 81) | class Blur(nn.Module): method __init__ (line 82) | def __init__(self, kernel, pad, upsample_factor=1, device='cpu'): method forward (line 95) | def forward(self, input): class EqualConv2d (line 101) | class EqualConv2d(nn.Module): method __init__ (line 102) | def __init__( method forward (line 121) | def forward(self, input): method __repr__ (line 132) | def __repr__(self): class EqualLinear (line 139) | class EqualLinear(nn.Module): method __init__ (line 140) | def __init__( method forward (line 159) | def forward(self, input): method __repr__ (line 169) | def __repr__(self): class ScaledLeakyReLU (line 175) | class ScaledLeakyReLU(nn.Module): method __init__ (line 176) | def __init__(self, negative_slope=0.2): method forward (line 181) | def forward(self, input): class ModulatedConv2d (line 187) | class ModulatedConv2d(nn.Module): method __init__ (line 188) | def __init__( method __repr__ (line 237) | def __repr__(self): method forward (line 243) | def forward(self, input, style): class NoiseInjection (line 287) | class NoiseInjection(nn.Module): method __init__ (line 288) | def __init__(self, isconcat=True): method forward (line 294) | def forward(self, image, noise=None): class ConstantInput (line 305) | class ConstantInput(nn.Module): method __init__ (line 306) | def __init__(self, channel, size=4): method forward (line 311) | def forward(self, input): class StyledConv (line 318) | class StyledConv(nn.Module): method __init__ (line 319) | def __init__( method forward (line 350) | def forward(self, input, style, noise=None): class ToRGB (line 359) | class ToRGB(nn.Module): method __init__ (line 360) | def __init__(self, in_channel, style_dim, upsample=True, blur_kernel=[... method forward (line 369) | def forward(self, input, style, skip=None): class Generator (line 380) | class Generator(nn.Module): method __init__ (line 381) | def __init__( method make_noise (line 465) | def make_noise(self): method mean_latent (line 476) | def mean_latent(self, n_latent): method get_latent (line 484) | def get_latent(self, input): method forward (line 487) | def forward( class ConvLayer (line 557) | class ConvLayer(nn.Sequential): method __init__ (line 558) | def __init__( class ResBlock (line 607) | class ResBlock(nn.Module): method __init__ (line 608) | def __init__(self, in_channel, out_channel, blur_kernel=[1, 3, 3, 1], ... method forward (line 618) | def forward(self, input): class FullGenerator (line 627) | class FullGenerator(nn.Module): method __init__ (line 628) | def __init__( method forward (line 669) | def forward(self, class Discriminator (line 689) | class Discriminator(nn.Module): method __init__ (line 690) | def __init__(self, size, channel_multiplier=2, blur_kernel=[1, 3, 3, 1... method forward (line 729) | def forward(self, input): FILE: third_part/GPEN/face_model/op/fused_act.py class FusedLeakyReLUFunctionBackward (line 25) | class FusedLeakyReLUFunctionBackward(Function): method forward (line 27) | def forward(ctx, grad_output, out, negative_slope, scale): method backward (line 48) | def backward(ctx, gradgrad_input, gradgrad_bias): class FusedLeakyReLUFunction (line 57) | class FusedLeakyReLUFunction(Function): method forward (line 59) | def forward(ctx, input, bias, negative_slope, scale): method backward (line 69) | def backward(ctx, grad_output): class FusedLeakyReLU (line 79) | class FusedLeakyReLU(nn.Module): method __init__ (line 80) | def __init__(self, channel, negative_slope=0.2, scale=2 ** 0.5, device... method forward (line 88) | def forward(self, input): function fused_leaky_relu (line 92) | def fused_leaky_relu(input, bias, negative_slope=0.2, scale=2 ** 0.5, de... FILE: third_part/GPEN/face_model/op/fused_bias_act.cpp function fused_bias_act (line 11) | torch::Tensor fused_bias_act(const torch::Tensor& input, const torch::Te... function PYBIND11_MODULE (line 19) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) { FILE: third_part/GPEN/face_model/op/upfirdn2d.cpp function upfirdn2d (line 12) | torch::Tensor upfirdn2d(const torch::Tensor& input, const torch::Tensor&... function PYBIND11_MODULE (line 21) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) { FILE: third_part/GPEN/face_model/op/upfirdn2d.py class UpFirDn2dBackward (line 23) | class UpFirDn2dBackward(Function): method forward (line 25) | def forward( method backward (line 67) | def backward(ctx, gradgrad_input): class UpFirDn2d (line 92) | class UpFirDn2d(Function): method forward (line 94) | def forward(ctx, input, kernel, up, down, pad): method backward (line 131) | def backward(ctx, grad_output): function upfirdn2d (line 149) | def upfirdn2d(input, kernel, up=1, down=1, pad=(0, 0), device='cpu'): function upfirdn2d_native (line 160) | def upfirdn2d_native( FILE: third_part/GPEN/face_morpher/facemorpher/aligner.py function positive_cap (line 7) | def positive_cap(num): function roi_coordinates (line 18) | def roi_coordinates(rect, size, scale): function scaling_factor (line 39) | def scaling_factor(rect, size): function resize_image (line 60) | def resize_image(img, scale): function resize_align (line 72) | def resize_align(img, points, size): FILE: third_part/GPEN/face_morpher/facemorpher/averager.py function list_imgpaths (line 38) | def list_imgpaths(imgfolder): function sharpen (line 45) | def sharpen(img): function load_image_points (line 49) | def load_image_points(path, size): function averager (line 59) | def averager(imgpaths, dest_filename=None, width=500, height=600, backgr... function main (line 112) | def main(): FILE: third_part/GPEN/face_morpher/facemorpher/blender.py function mask_from_points (line 5) | def mask_from_points(size, points): function overlay_image (line 21) | def overlay_image(foreground_image, mask, background_image): function apply_mask (line 32) | def apply_mask(img, mask): function weighted_average (line 45) | def weighted_average(img1, img2, percent=0.5): function alpha_feathering (line 53) | def alpha_feathering(src_img, dest_img, img_mask, blur_radius=15): function poisson_blend (line 63) | def poisson_blend(img_source, dest_img, img_mask, offset=(0, 0)): FILE: third_part/GPEN/face_morpher/facemorpher/locator.py function boundary_points (line 19) | def boundary_points(points, width_percent=0.1, height_percent=0.1): function face_points (line 33) | def face_points(img, add_boundary_points=True): function face_points_dlib (line 36) | def face_points_dlib(img, add_boundary_points=True): function face_points_stasm (line 68) | def face_points_stasm(img, add_boundary_points=True): function average_points (line 91) | def average_points(point_set): function weighted_average_points (line 99) | def weighted_average_points(start_points, end_points, percent=0.5): FILE: third_part/GPEN/face_morpher/facemorpher/morpher.py function verify_args (line 41) | def verify_args(args): function load_image_points (line 54) | def load_image_points(path, size): function load_valid_image_points (line 64) | def load_valid_image_points(imgpaths, size): function list_imgpaths (line 71) | def list_imgpaths(images_folder=None, src_image=None, dest_image=None): function morph (line 82) | def morph(src_img, src_points, dest_img, dest_points, function morpher (line 125) | def morpher(imgpaths, width=500, height=600, num_frames=20, fps=10, function main (line 141) | def main(): FILE: third_part/GPEN/face_morpher/facemorpher/plotter.py function bgr2rgb (line 11) | def bgr2rgb(img): function check_do_plot (line 17) | def check_do_plot(func): function check_do_save (line 24) | def check_do_save(func): class Plotter (line 31) | class Plotter(object): method __init__ (line 32) | def __init__(self, plot=True, rows=0, cols=0, num_images=0, out_folder... method set_filepath (line 48) | def set_filepath(self, folder): method save (line 59) | def save(self, img, filename=None): method plot_one (line 70) | def plot_one(self, img): method show (line 78) | def show(self): method plot_mesh (line 86) | def plot_mesh(self, points, tri, color='k'): FILE: third_part/GPEN/face_morpher/facemorpher/videoer.py function check_write_video (line 9) | def check_write_video(func): class Video (line 18) | class Video(object): method __init__ (line 19) | def __init__(self, filename, fps, w, h): method write (line 29) | def write(self, img, num_times=1): method end (line 34) | def end(self): FILE: third_part/GPEN/face_morpher/facemorpher/warper.py function bilinear_interpolate (line 4) | def bilinear_interpolate(img, coords): function grid_coordinates (line 28) | def grid_coordinates(points): function process_warp (line 41) | def process_warp(src_img, result_img, tri_affines, dst_points, delaunay): function triangular_affine_matrices (line 60) | def triangular_affine_matrices(vertices, src_points, dest_points): function warp_image (line 77) | def warp_image(src_img, src_points, dest_points, dest_shape, dtype=np.ui... function test_local (line 93) | def test_local(): FILE: third_part/GPEN/face_parse/blocks.py class NormLayer (line 8) | class NormLayer(nn.Module): method __init__ (line 15) | def __init__(self, channels, normalize_shape=None, norm_type='bn', ref... method forward (line 34) | def forward(self, x, ref=None): class ReluLayer (line 41) | class ReluLayer(nn.Module): method __init__ (line 52) | def __init__(self, channels, relu_type='relu'): method forward (line 68) | def forward(self, x): class ConvLayer (line 72) | class ConvLayer(nn.Module): method __init__ (line 73) | def __init__(self, in_channels, out_channels, kernel_size=3, scale='no... method forward (line 92) | def forward(self, x): class ResidualBlock (line 102) | class ResidualBlock(nn.Module): method __init__ (line 106) | def __init__(self, c_in, c_out, relu_type='prelu', norm_type='bn', sca... method forward (line 120) | def forward(self, x): FILE: third_part/GPEN/face_parse/face_parsing.py class FaceParse (line 15) | class FaceParse(object): method __init__ (line 16) | def __init__(self, base_dir='./', model='ParseNet-latest', device='cud... method load_model (line 37) | def load_model(self): method process (line 43) | def process(self, im, masks=[0, 255, 255, 255, 255, 255, 255, 255, 255... method process_tensor (line 52) | def process_tensor(self, imt): method img2tensor (line 64) | def img2tensor(self, img): method tenor2mask (line 70) | def tenor2mask(self, tensor, masks): class FaceParse_v2 (line 88) | class FaceParse_v2(object): method __init__ (line 89) | def __init__(self, device='cuda', mask_map = [0, 255, 255, 255, 255, 2... method load_model (line 113) | def load_model(self): method process (line 119) | def process(self, im, masks=[0, 255, 255, 255, 255, 255, 255, 255, 255... method tenor2mask (line 134) | def tenor2mask(self, tensor, masks): FILE: third_part/GPEN/face_parse/model.py class ConvBNReLU (line 14) | class ConvBNReLU(nn.Module): method __init__ (line 15) | def __init__(self, in_chan, out_chan, ks=3, stride=1, padding=1, *args... method forward (line 26) | def forward(self, x): method init_weight (line 31) | def init_weight(self): class BiSeNetOutput (line 37) | class BiSeNetOutput(nn.Module): method __init__ (line 38) | def __init__(self, in_chan, mid_chan, n_classes, *args, **kwargs): method forward (line 44) | def forward(self, x): method init_weight (line 49) | def init_weight(self): method get_params (line 55) | def get_params(self): class AttentionRefinementModule (line 67) | class AttentionRefinementModule(nn.Module): method __init__ (line 68) | def __init__(self, in_chan, out_chan, *args, **kwargs): method forward (line 76) | def forward(self, x): method init_weight (line 85) | def init_weight(self): class ContextPath (line 92) | class ContextPath(nn.Module): method __init__ (line 93) | def __init__(self, *args, **kwargs): method forward (line 104) | def forward(self, x): method init_weight (line 127) | def init_weight(self): method get_params (line 133) | def get_params(self): class SpatialPath (line 146) | class SpatialPath(nn.Module): method __init__ (line 147) | def __init__(self, *args, **kwargs): method forward (line 155) | def forward(self, x): method init_weight (line 162) | def init_weight(self): method get_params (line 168) | def get_params(self): class FeatureFusionModule (line 180) | class FeatureFusionModule(nn.Module): method __init__ (line 181) | def __init__(self, in_chan, out_chan, *args, **kwargs): method forward (line 200) | def forward(self, fsp, fcp): method init_weight (line 212) | def init_weight(self): method get_params (line 218) | def get_params(self): class BiSeNet (line 230) | class BiSeNet(nn.Module): method __init__ (line 231) | def __init__(self, n_classes, *args, **kwargs): method forward (line 241) | def forward(self, x): method init_weight (line 256) | def init_weight(self): method get_params (line 262) | def get_params(self): FILE: third_part/GPEN/face_parse/parse_model.py function define_P (line 12) | def define_P(in_size=512, out_size=512, min_feat_size=32, relu_type='Lea... class ParseNet (line 21) | class ParseNet(nn.Module): method __init__ (line 22) | def __init__(self, method forward (line 69) | def forward(self, x): FILE: third_part/GPEN/face_parse/resnet.py function conv3x3 (line 14) | def conv3x3(in_planes, out_planes, stride=1): class BasicBlock (line 20) | class BasicBlock(nn.Module): method __init__ (line 21) | def __init__(self, in_chan, out_chan, stride=1): method forward (line 36) | def forward(self, x): function create_layer_basic (line 51) | def create_layer_basic(in_chan, out_chan, bnum, stride=1): class Resnet18 (line 58) | class Resnet18(nn.Module): method __init__ (line 59) | def __init__(self): method forward (line 71) | def forward(self, x): method init_weight (line 82) | def init_weight(self): method get_params (line 90) | def get_params(self): FILE: third_part/GPEN/gpen_face_enhancer.py class FaceEnhancement (line 13) | class FaceEnhancement(object): method __init__ (line 14) | def __init__(self, base_dir='./', size=512, model=None, use_sr=True, s... method mask_postprocess (line 42) | def mask_postprocess(self, mask, thres=20): method process (line 49) | def process(self, img, ori_img, bbox=None, face_enhance=True, possion_... FILE: third_part/face3d/coeff_detector.py class CoeffDetector (line 18) | class CoeffDetector(nn.Module): method __init__ (line 19) | def __init__(self, opt): method forward (line 30) | def forward(self, img, lm): method image_transform (line 53) | def image_transform(self, images, lm): function get_data_path (line 74) | def get_data_path(root, keypoint_root): FILE: third_part/face3d/data/__init__.py function find_dataset_using_name (line 19) | def find_dataset_using_name(dataset_name): function get_option_setter (line 42) | def get_option_setter(dataset_name): function create_dataset (line 48) | def create_dataset(opt, rank=0): class CustomDatasetDataLoader (line 62) | class CustomDatasetDataLoader(): method __init__ (line 65) | def __init__(self, opt, rank=0): method set_epoch (line 99) | def set_epoch(self, epoch): method load_data (line 104) | def load_data(self): method __len__ (line 107) | def __len__(self): method __iter__ (line 111) | def __iter__(self): FILE: third_part/face3d/data/base_dataset.py class BaseDataset (line 13) | class BaseDataset(data.Dataset, ABC): method __init__ (line 23) | def __init__(self, opt): method modify_commandline_options (line 34) | def modify_commandline_options(parser, is_train): method __len__ (line 47) | def __len__(self): method __getitem__ (line 52) | def __getitem__(self, index): function get_transform (line 64) | def get_transform(grayscale=False): function get_affine_mat (line 71) | def get_affine_mat(opt, size): function apply_img_affine (line 98) | def apply_img_affine(img, affine_inv, method=Image.BICUBIC): function apply_lm_affine (line 101) | def apply_lm_affine(landmark, affine, flip, size): FILE: third_part/face3d/data/flist_dataset.py function default_flist_reader (line 19) | def default_flist_reader(flist): function jason_flist_reader (line 31) | def jason_flist_reader(flist): function parse_label (line 36) | def parse_label(label): class FlistDataset (line 40) | class FlistDataset(BaseDataset): method __init__ (line 46) | def __init__(self, opt): method __getitem__ (line 67) | def __getitem__(self, index): method _augmentation (line 111) | def _augmentation(self, img, lm, opt, msk=None): method __len__ (line 122) | def __len__(self): FILE: third_part/face3d/data/image_folder.py function is_image_file (line 20) | def is_image_file(filename): function make_dataset (line 24) | def make_dataset(dir, max_dataset_size=float("inf")): function default_loader (line 36) | def default_loader(path): class ImageFolder (line 40) | class ImageFolder(data.Dataset): method __init__ (line 42) | def __init__(self, root, transform=None, return_paths=False, method __getitem__ (line 55) | def __getitem__(self, index): method __len__ (line 65) | def __len__(self): FILE: third_part/face3d/data/template_dataset.py class TemplateDataset (line 19) | class TemplateDataset(BaseDataset): method modify_commandline_options (line 22) | def modify_commandline_options(parser, is_train): method __init__ (line 36) | def __init__(self, opt): method __getitem__ (line 54) | def __getitem__(self, index): method __len__ (line 73) | def __len__(self): FILE: third_part/face3d/data_preparation.py function data_prepare (line 21) | def data_prepare(folder_list,mode): FILE: third_part/face3d/extract_kp_videos.py class KeypointExtractor (line 15) | class KeypointExtractor(): method __init__ (line 16) | def __init__(self): method extract_keypoint (line 20) | def extract_keypoint(self, images, name=None, info=True): function read_video (line 59) | def read_video(filename): function run (line 73) | def run(data): FILE: third_part/face3d/face_recon_videos.py function get_data_path (line 18) | def get_data_path(root, keypoint_root): class VideoPathDataset (line 34) | class VideoPathDataset(torch.utils.data.Dataset): method __init__ (line 35) | def __init__(self, filenames, txt_filenames, bfm_folder): method __len__ (line 40) | def __len__(self): method __getitem__ (line 43) | def __getitem__(self, index): method read_video (line 61) | def read_video(self, filename): method image_transform (line 75) | def image_transform(self, images, lm): function main (line 92) | def main(opt, model): FILE: third_part/face3d/models/__init__.py function find_model_using_name (line 25) | def find_model_using_name(model_name): function get_option_setter (line 48) | def get_option_setter(model_name): function create_model (line 54) | def create_model(opt): FILE: third_part/face3d/models/arcface_torch/backbones/__init__.py function get_model (line 5) | def get_model(name, **kwargs): FILE: third_part/face3d/models/arcface_torch/backbones/iresnet.py function conv3x3 (line 7) | def conv3x3(in_planes, out_planes, stride=1, groups=1, dilation=1): function conv1x1 (line 19) | def conv1x1(in_planes, out_planes, stride=1): class IBasicBlock (line 28) | class IBasicBlock(nn.Module): method __init__ (line 30) | def __init__(self, inplanes, planes, stride=1, downsample=None, method forward (line 46) | def forward(self, x): class IResNet (line 60) | class IResNet(nn.Module): method __init__ (line 62) | def __init__(self, method _make_layer (line 114) | def _make_layer(self, block, planes, blocks, stride=1, dilate=False): method forward (line 140) | def forward(self, x): function _iresnet (line 157) | def _iresnet(arch, block, layers, pretrained, progress, **kwargs): function iresnet18 (line 164) | def iresnet18(pretrained=False, progress=True, **kwargs): function iresnet34 (line 169) | def iresnet34(pretrained=False, progress=True, **kwargs): function iresnet50 (line 174) | def iresnet50(pretrained=False, progress=True, **kwargs): function iresnet100 (line 179) | def iresnet100(pretrained=False, progress=True, **kwargs): function iresnet200 (line 184) | def iresnet200(pretrained=False, progress=True, **kwargs): FILE: third_part/face3d/models/arcface_torch/backbones/iresnet2060.py function conv3x3 (line 10) | def conv3x3(in_planes, out_planes, stride=1, groups=1, dilation=1): function conv1x1 (line 22) | def conv1x1(in_planes, out_planes, stride=1): class IBasicBlock (line 31) | class IBasicBlock(nn.Module): method __init__ (line 34) | def __init__(self, inplanes, planes, stride=1, downsample=None, method forward (line 50) | def forward(self, x): class IResNet (line 64) | class IResNet(nn.Module): method __init__ (line 67) | def __init__(self, method _make_layer (line 119) | def _make_layer(self, block, planes, blocks, stride=1, dilate=False): method checkpoint (line 145) | def checkpoint(self, func, num_seg, x): method forward (line 151) | def forward(self, x): function _iresnet (line 168) | def _iresnet(arch, block, layers, pretrained, progress, **kwargs): function iresnet2060 (line 175) | def iresnet2060(pretrained=False, progress=True, **kwargs): FILE: third_part/face3d/models/arcface_torch/backbones/mobilefacenet.py class Flatten (line 11) | class Flatten(Module): method forward (line 12) | def forward(self, x): class ConvBlock (line 16) | class ConvBlock(Module): method __init__ (line 17) | def __init__(self, in_c, out_c, kernel=(1, 1), stride=(1, 1), padding=... method forward (line 25) | def forward(self, x): class LinearBlock (line 29) | class LinearBlock(Module): method __init__ (line 30) | def __init__(self, in_c, out_c, kernel=(1, 1), stride=(1, 1), padding=... method forward (line 37) | def forward(self, x): class DepthWise (line 41) | class DepthWise(Module): method __init__ (line 42) | def __init__(self, in_c, out_c, residual=False, kernel=(3, 3), stride=... method forward (line 51) | def forward(self, x): class Residual (line 63) | class Residual(Module): method __init__ (line 64) | def __init__(self, c, num_block, groups, kernel=(3, 3), stride=(1, 1),... method forward (line 71) | def forward(self, x): class GDC (line 75) | class GDC(Module): method __init__ (line 76) | def __init__(self, embedding_size): method forward (line 84) | def forward(self, x): class MobileFaceNet (line 88) | class MobileFaceNet(Module): method __init__ (line 89) | def __init__(self, fp16=False, num_features=512): method _initialize_weights (line 107) | def _initialize_weights(self): method forward (line 121) | def forward(self, x): function get_mbf (line 129) | def get_mbf(fp16, num_features): FILE: third_part/face3d/models/arcface_torch/dataset.py class BackgroundGenerator (line 13) | class BackgroundGenerator(threading.Thread): method __init__ (line 14) | def __init__(self, generator, local_rank, max_prefetch=6): method run (line 22) | def run(self): method next (line 28) | def next(self): method __next__ (line 34) | def __next__(self): method __iter__ (line 37) | def __iter__(self): class DataLoaderX (line 41) | class DataLoaderX(DataLoader): method __init__ (line 43) | def __init__(self, local_rank, **kwargs): method __iter__ (line 48) | def __iter__(self): method preload (line 54) | def preload(self): method __next__ (line 62) | def __next__(self): class MXFaceDataset (line 71) | class MXFaceDataset(Dataset): method __init__ (line 72) | def __init__(self, root_dir, local_rank): method __getitem__ (line 93) | def __getitem__(self, index): method __len__ (line 106) | def __len__(self): class SyntheticDataset (line 110) | class SyntheticDataset(Dataset): method __init__ (line 111) | def __init__(self, local_rank): method __getitem__ (line 120) | def __getitem__(self, index): method __len__ (line 123) | def __len__(self): FILE: third_part/face3d/models/arcface_torch/eval/verification.py class LFold (line 41) | class LFold: method __init__ (line 42) | def __init__(self, n_splits=2, shuffle=False): method split (line 47) | def split(self, indices): function calculate_roc (line 54) | def calculate_roc(thresholds, function calculate_accuracy (line 109) | def calculate_accuracy(threshold, dist, actual_issame): function calculate_val (line 124) | def calculate_val(thresholds, function calculate_val_far (line 165) | def calculate_val_far(threshold, dist, actual_issame): function evaluate (line 179) | def evaluate(embeddings, actual_issame, nrof_folds=10, pca=0): function load_bin (line 200) | def load_bin(path, image_size): function test (line 227) | def test(data_set, backbone, batch_size, nfolds=10): function dumpR (line 275) | def dumpR(data_set, FILE: third_part/face3d/models/arcface_torch/eval_ijbc.py class Embedding (line 54) | class Embedding(object): method __init__ (line 55) | def __init__(self, prefix, data_shape, batch_size=1): method get (line 75) | def get(self, rimg, landmark): method forward_db (line 104) | def forward_db(self, batch_data): function divideIntoNstrand (line 113) | def divideIntoNstrand(listTemp, n): function read_template_media_list (line 120) | def read_template_media_list(path): function read_template_pair_list (line 131) | def read_template_pair_list(path): function read_image_feature (line 145) | def read_image_feature(path): function get_image_feature (line 154) | def get_image_feature(img_path, files_list, model_path, epoch, gpu_id): function image2template_feature (line 212) | def image2template_feature(img_feats=None, templates=None, medias=None): function verification (line 252) | def verification(template_norm_feats=None, function verification2 (line 282) | def verification2(template_norm_feats=None, function read_score (line 306) | def read_score(path): FILE: third_part/face3d/models/arcface_torch/inference.py function inference (line 11) | def inference(weight, name, img): FILE: third_part/face3d/models/arcface_torch/losses.py function get_loss (line 5) | def get_loss(name): class CosFace (line 14) | class CosFace(nn.Module): method __init__ (line 15) | def __init__(self, s=64.0, m=0.40): method forward (line 20) | def forward(self, cosine, label): class ArcFace (line 29) | class ArcFace(nn.Module): method __init__ (line 30) | def __init__(self, s=64.0, m=0.5): method forward (line 35) | def forward(self, cosine: torch.Tensor, label): FILE: third_part/face3d/models/arcface_torch/onnx_helper.py class ArcFaceORT (line 15) | class ArcFaceORT: method __init__ (line 16) | def __init__(self, model_path, cpu=False): method check (line 22) | def check(self, track='cfat', test_img = None): method check_batch (line 184) | def check_batch(self, img): method meta_info (line 202) | def meta_info(self): method forward (line 206) | def forward(self, imgs): method benchmark (line 222) | def benchmark(self, img): FILE: third_part/face3d/models/arcface_torch/onnx_ijbc.py class AlignedDataSet (line 28) | class AlignedDataSet(mx.gluon.data.Dataset): method __init__ (line 29) | def __init__(self, root, lines, align=True): method __len__ (line 34) | def __len__(self): method __getitem__ (line 37) | def __getitem__(self, idx): function extract (line 54) | def extract(model_root, dataset): function read_template_media_list (line 78) | def read_template_media_list(path): function read_template_pair_list (line 85) | def read_template_pair_list(path): function read_image_feature (line 93) | def read_image_feature(path): function image2template_feature (line 99) | def image2template_feature(img_feats=None, function verification (line 125) | def verification(template_norm_feats=None, function verification2 (line 147) | def verification2(template_norm_feats=None, function main (line 169) | def main(args): FILE: third_part/face3d/models/arcface_torch/partial_fc.py class PartialFC (line 11) | class PartialFC(Module): method __init__ (line 20) | def __init__(self, rank, local_rank, world_size, batch_size, resume, method save_params (line 93) | def save_params(self): method sample (line 100) | def sample(self, total_label): method forward (line 125) | def forward(self, total_features, norm_weight): method update (line 133) | def update(self): method prepare (line 139) | def prepare(self, label, optimizer): method forward_backward (line 159) | def forward_backward(self, label, features, optimizer): FILE: third_part/face3d/models/arcface_torch/torch2onnx.py function convert_onnx (line 6) | def convert_onnx(net, path_module, output, opset=11, simplify=False): FILE: third_part/face3d/models/arcface_torch/train.py function main (line 21) | def main(args): FILE: third_part/face3d/models/arcface_torch/utils/plot.py function read_template_pair_list (line 19) | def read_template_pair_list(path): FILE: third_part/face3d/models/arcface_torch/utils/utils_amp.py class _MultiDeviceReplicator (line 14) | class _MultiDeviceReplicator(object): method __init__ (line 19) | def __init__(self, master_tensor: torch.Tensor) -> None: method get (line 24) | def get(self, device) -> torch.Tensor: class MaxClipGradScaler (line 32) | class MaxClipGradScaler(GradScaler): method __init__ (line 33) | def __init__(self, init_scale, max_scale: float, growth_interval=100): method scale_clip (line 37) | def scale_clip(self): method scale (line 46) | def scale(self, outputs): FILE: third_part/face3d/models/arcface_torch/utils/utils_callbacks.py class CallBackVerification (line 12) | class CallBackVerification(object): method __init__ (line 13) | def __init__(self, frequent, rank, val_targets, rec_prefix, image_size... method ver_test (line 23) | def ver_test(self, backbone: torch.nn.Module, global_step: int): method init_dataset (line 36) | def init_dataset(self, val_targets, data_dir, image_size): method __call__ (line 44) | def __call__(self, num_update, backbone: torch.nn.Module): class CallBackLogging (line 51) | class CallBackLogging(object): method __init__ (line 52) | def __init__(self, frequent, rank, total_step, batch_size, world_size,... method __call__ (line 64) | def __call__(self, class CallBackModelCheckpoint (line 105) | class CallBackModelCheckpoint(object): method __init__ (line 106) | def __init__(self, rank, output="./"): method __call__ (line 110) | def __call__(self, global_step, backbone, partial_fc, ): FILE: third_part/face3d/models/arcface_torch/utils/utils_config.py function get_config (line 5) | def get_config(config_file): FILE: third_part/face3d/models/arcface_torch/utils/utils_logging.py class AverageMeter (line 6) | class AverageMeter(object): method __init__ (line 10) | def __init__(self): method reset (line 17) | def reset(self): method update (line 23) | def update(self, val, n=1): function init_logging (line 30) | def init_logging(rank, models_root): FILE: third_part/face3d/models/base_model.py class BaseModel (line 12) | class BaseModel(ABC): method __init__ (line 22) | def __init__(self, opt): method dict_grad_hook_factory (line 49) | def dict_grad_hook_factory(add_func=lambda x: x): method modify_commandline_options (line 60) | def modify_commandline_options(parser, is_train): method set_input (line 73) | def set_input(self, input): method forward (line 82) | def forward(self): method optimize_parameters (line 87) | def optimize_parameters(self): method setup (line 91) | def setup(self, opt): method parallelize (line 107) | def parallelize(self, convert_sync_batchnorm=True): method data_dependent_initialize (line 138) | def data_dependent_initialize(self, data): method train (line 141) | def train(self): method eval (line 148) | def eval(self): method test (line 155) | def test(self): method compute_visuals (line 165) | def compute_visuals(self): method get_image_paths (line 169) | def get_image_paths(self, name='A'): method update_learning_rate (line 173) | def update_learning_rate(self): method get_current_visuals (line 184) | def get_current_visuals(self): method get_current_losses (line 192) | def get_current_losses(self): method save_networks (line 200) | def save_networks(self, epoch): method __patch_instance_norm_state_dict (line 230) | def __patch_instance_norm_state_dict(self, state_dict, module, keys, i... method load_networks (line 244) | def load_networks(self, epoch): method print_networks (line 284) | def print_networks(self, verbose): method set_requires_grad (line 302) | def set_requires_grad(self, nets, requires_grad=False): method generate_visuals_for_evaluation (line 315) | def generate_visuals_for_evaluation(self, data, mode): FILE: third_part/face3d/models/bfm.py function perspective_projection (line 11) | def perspective_projection(focal, center): class SH (line 19) | class SH: method __init__ (line 20) | def __init__(self): class ParametricFaceModel (line 26) | class ParametricFaceModel: method __init__ (line 27) | def __init__(self, method to (line 79) | def to(self, device): method compute_shape (line 86) | def compute_shape(self, id_coeff, exp_coeff): method compute_texture (line 102) | def compute_texture(self, tex_coeff, normalize=True): method compute_norm (line 117) | def compute_norm(self, face_shape): method compute_color (line 140) | def compute_color(self, face_texture, face_norm, gamma): method compute_rotation (line 174) | def compute_rotation(self, angles): method to_camera (line 210) | def to_camera(self, face_shape): method to_image (line 214) | def to_image(self, face_shape): method transform (line 229) | def transform(self, face_shape, rot, trans): method get_landmarks (line 242) | def get_landmarks(self, face_proj): method split_coeff (line 252) | def split_coeff(self, coeffs): method compute_for_render (line 274) | def compute_for_render(self, coeffs): FILE: third_part/face3d/models/facerecon_model.py class FaceReconModel (line 17) | class FaceReconModel(BaseModel): method modify_commandline_options (line 20) | def modify_commandline_options(parser, is_train=True): method __init__ (line 73) | def __init__(self, opt): method set_input (line 121) | def set_input(self, input): method forward (line 133) | def forward(self): method compute_losses (line 144) | def compute_losses(self): method optimize_parameters (line 176) | def optimize_parameters(self, isTrain=True): method compute_visuals (line 185) | def compute_visuals(self): method save_mesh (line 207) | def save_mesh(self, name): method save_coeff (line 218) | def save_coeff(self,name): FILE: third_part/face3d/models/losses.py function resize_n_crop (line 7) | def resize_n_crop(image, M, dsize=112): class PerceptualLoss (line 13) | class PerceptualLoss(nn.Module): method __init__ (line 14) | def __init__(self, recog_net, input_size=112): method forward (line 19) | def forward(imageA, imageB, M): function perceptual_loss (line 39) | def perceptual_loss(id_featureA, id_featureB): function photo_loss (line 45) | def photo_loss(imageA, imageB, mask, eps=1e-6): function landmark_loss (line 56) | def landmark_loss(predict_lm, gt_lm, weight=None): function reg_loss (line 76) | def reg_loss(coeffs_dict, opt=None): function reflectance_loss (line 101) | def reflectance_loss(texture, mask): FILE: third_part/face3d/models/networks.py function resize_n_crop (line 21) | def resize_n_crop(image, M, dsize=112): function filter_state_dict (line 26) | def filter_state_dict(state_dict, remove_name='fc'): function get_scheduler (line 34) | def get_scheduler(optimizer, opt): function define_net_recon (line 61) | def define_net_recon(net_recon, use_last_fc=False, init_path=None): function define_net_recog (line 64) | def define_net_recog(net_recog, pretrained_path=None): class ReconNetWrapper (line 69) | class ReconNetWrapper(nn.Module): method __init__ (line 71) | def __init__(self, net_recon, use_last_fc=False, init_path=None): method forward (line 97) | def forward(self, x): class RecogNetWrapper (line 107) | class RecogNetWrapper(nn.Module): method __init__ (line 108) | def __init__(self, net_recog, pretrained_path=None, input_size=112): method forward (line 121) | def forward(self, image, M): function conv3x3 (line 146) | def conv3x3(in_planes: int, out_planes: int, stride: int = 1, groups: in... function conv1x1 (line 152) | def conv1x1(in_planes: int, out_planes: int, stride: int = 1, bias: bool... class BasicBlock (line 157) | class BasicBlock(nn.Module): method __init__ (line 160) | def __init__( method forward (line 187) | def forward(self, x: Tensor) -> Tensor: class Bottleneck (line 206) | class Bottleneck(nn.Module): method __init__ (line 215) | def __init__( method forward (line 241) | def forward(self, x: Tensor) -> Tensor: class ResNet (line 264) | class ResNet(nn.Module): method __init__ (line 266) | def __init__( method _make_layer (line 331) | def _make_layer(self, block: Type[Union[BasicBlock, Bottleneck]], plan... method _forward_impl (line 356) | def _forward_impl(self, x: Tensor) -> Tensor: method forward (line 374) | def forward(self, x: Tensor) -> Tensor: function _resnet (line 378) | def _resnet( function resnet18 (line 394) | def resnet18(pretrained: bool = False, progress: bool = True, **kwargs: ... function resnet34 (line 406) | def resnet34(pretrained: bool = False, progress: bool = True, **kwargs: ... function resnet50 (line 418) | def resnet50(pretrained: bool = False, progress: bool = True, **kwargs: ... function resnet101 (line 430) | def resnet101(pretrained: bool = False, progress: bool = True, **kwargs:... function resnet152 (line 442) | def resnet152(pretrained: bool = False, progress: bool = True, **kwargs:... function resnext50_32x4d (line 454) | def resnext50_32x4d(pretrained: bool = False, progress: bool = True, **k... function resnext101_32x8d (line 468) | def resnext101_32x8d(pretrained: bool = False, progress: bool = True, **... function wide_resnet50_2 (line 482) | def wide_resnet50_2(pretrained: bool = False, progress: bool = True, **k... function wide_resnet101_2 (line 500) | def wide_resnet101_2(pretrained: bool = False, progress: bool = True, **... FILE: third_part/face3d/models/template_model.py class TemplateModel (line 24) | class TemplateModel(BaseModel): method modify_commandline_options (line 26) | def modify_commandline_options(parser, is_train=True): method __init__ (line 42) | def __init__(self, opt): method set_input (line 73) | def set_input(self, input): method forward (line 84) | def forward(self): method backward (line 88) | def backward(self): method optimize_parameters (line 95) | def optimize_parameters(self): FILE: third_part/face3d/options/base_options.py class BaseOptions (line 13) | class BaseOptions(): method __init__ (line 20) | def __init__(self, cmd_line=None): method initialize (line 27) | def initialize(self, parser): method gather_options (line 52) | def gather_options(self): method print_options (line 93) | def print_options(self, opt): method parse (line 122) | def parse(self): FILE: third_part/face3d/options/inference_options.py class InferenceOptions (line 4) | class InferenceOptions(BaseOptions): method initialize (line 10) | def initialize(self, parser): FILE: third_part/face3d/options/test_options.py class TestOptions (line 7) | class TestOptions(BaseOptions): method initialize (line 13) | def initialize(self, parser): FILE: third_part/face3d/options/train_options.py class TrainOptions (line 7) | class TrainOptions(BaseOptions): method initialize (line 13) | def initialize(self, parser): FILE: third_part/face3d/util/detect_lm68.py function save_label (line 12) | def save_label(labels, save_path): function draw_landmarks (line 15) | def draw_landmarks(img, landmark, save_name): function load_data (line 35) | def load_data(img_name, txt_name): function load_lm_graph (line 39) | def load_lm_graph(graph_filename): function detect_68p (line 53) | def detect_68p(img_path,sess,input_op,output_op): FILE: third_part/face3d/util/generate_list.py function write_list (line 7) | def write_list(lms_list, imgs_list, msks_list, mode='train',save_folder=... function check_list (line 21) | def check_list(rlms_list, rimgs_list, rmsks_list): FILE: third_part/face3d/util/html.py class HTML (line 6) | class HTML: method __init__ (line 14) | def __init__(self, web_dir, title, refresh=0): method get_image_dir (line 35) | def get_image_dir(self): method add_header (line 39) | def add_header(self, text): method add_images (line 48) | def add_images(self, ims, txts, links, width=400): method save (line 68) | def save(self): FILE: third_part/face3d/util/load_mats.py function LoadExpBasis (line 11) | def LoadExpBasis(bfm_folder='BFM'): function transferBFM09 (line 32) | def transferBFM09(bfm_folder='BFM'): function load_lm3d (line 105) | def load_lm3d(bfm_folder): FILE: third_part/face3d/util/nvdiffrast.py function ndc_projection (line 15) | def ndc_projection(x=0.1, n=1.0, f=50.0): class MeshRenderer (line 21) | class MeshRenderer(nn.Module): method __init__ (line 22) | def __init__(self, method forward (line 35) | def forward(self, vertex, tri, feat=None): FILE: third_part/face3d/util/preprocess.py function POS (line 18) | def POS(xp, x): function BBRegression (line 43) | def BBRegression(points, params): function img_padding (line 72) | def img_padding(img, box): function crop (line 86) | def crop(img, bbox): function scale_trans (line 99) | def scale_trans(img, lm, t, s): function align_for_lm (line 126) | def align_for_lm(img, five_points): function resize_n_crop_img (line 137) | def resize_n_crop_img(img, lm, t, s, target_size=224., mask=None): function extract_5p (line 161) | def extract_5p(lm): function align_img (line 169) | def align_img(img, lm, lm3D, mask=None, target_size=224., rescale_factor... function estimate_norm (line 201) | def estimate_norm(lm_68p, H): function estimate_norm_torch (line 224) | def estimate_norm_torch(lm_68p, H): FILE: third_part/face3d/util/skin_mask.py class GMM (line 9) | class GMM: method __init__ (line 10) | def __init__(self, dim, num, w, mu, cov, cov_det, cov_inv): method likelihood (line 23) | def likelihood(self, data): function _rgb2ycbcr (line 42) | def _rgb2ycbcr(rgb): function _bgr2ycbcr (line 54) | def _bgr2ycbcr(bgr): function skinmask (line 90) | def skinmask(imbgr): function get_skin_mask (line 111) | def get_skin_mask(img_path): FILE: third_part/face3d/util/util.py function str2bool (line 14) | def str2bool(v): function copyconf (line 25) | def copyconf(default_opt, **kwargs): function genvalconf (line 31) | def genvalconf(train_opt, **kwargs): function find_class_in_module (line 43) | def find_class_in_module(target_cls_name, module): function tensor2im (line 56) | def tensor2im(input_image, imtype=np.uint8): function diagnose_network (line 77) | def diagnose_network(net, name='network'): function save_image (line 96) | def save_image(image_numpy, image_path, aspect_ratio=1.0): function print_numpy (line 116) | def print_numpy(x, val=True, shp=False): function mkdirs (line 132) | def mkdirs(paths): function mkdir (line 145) | def mkdir(path): function correct_resize_label (line 155) | def correct_resize_label(t, size): function correct_resize (line 169) | def correct_resize(t, size, mode=Image.BICUBIC): function draw_landmarks (line 180) | def draw_landmarks(img, landmark, color='r', step=2): FILE: third_part/face3d/util/visualizer.py function save_images (line 13) | def save_images(webpage, visuals, image_path, aspect_ratio=1.0, width=256): class Visualizer (line 44) | class Visualizer(): method __init__ (line 50) | def __init__(self, opt): method reset (line 77) | def reset(self): method display_current_results (line 82) | def display_current_results(self, visuals, total_iters, epoch, save_re... method plot_current_losses (line 117) | def plot_current_losses(self, total_iters, losses): method print_current_losses (line 131) | def print_current_losses(self, epoch, iters, losses, t_comp, t_data): class MyVisualizer (line 150) | class MyVisualizer: method __init__ (line 151) | def __init__(self, opt): method display_current_results (line 174) | def display_current_results(self, visuals, total_iters, epoch, dataset... method plot_current_losses (line 205) | def plot_current_losses(self, total_iters, losses, dataset='train'): method print_current_losses (line 210) | def print_current_losses(self, epoch, iters, losses, t_comp, t_data, d... FILE: third_part/face_detection/api.py class LandmarksType (line 17) | class LandmarksType(Enum): class NetworkSize (line 30) | class NetworkSize(Enum): method __new__ (line 36) | def __new__(cls, value): method __int__ (line 41) | def __int__(self): class FaceAlignment (line 46) | class FaceAlignment: method __init__ (line 47) | def __init__(self, landmarks_type, network_size=NetworkSize.LARGE, method get_detections_for_batch (line 64) | def get_detections_for_batch(self, images): FILE: third_part/face_detection/detection/core.py class FaceDetector (line 9) | class FaceDetector(object): method __init__ (line 18) | def __init__(self, device, verbose): method detect_from_image (line 32) | def detect_from_image(self, tensor_or_path): method detect_from_directory (line 54) | def detect_from_directory(self, path, extensions=['.jpg', '.png'], rec... method reference_scale (line 104) | def reference_scale(self): method reference_x_shift (line 108) | def reference_x_shift(self): method reference_y_shift (line 112) | def reference_y_shift(self): method tensor_or_path_to_ndarray (line 116) | def tensor_or_path_to_ndarray(tensor_or_path, rgb=True): FILE: third_part/face_detection/detection/sfd/bbox.py function IOU (line 17) | def IOU(ax1, ay1, ax2, ay2, bx1, by1, bx2, by2): function bboxlog (line 30) | def bboxlog(x1, y1, x2, y2, axc, ayc, aww, ahh): function bboxloginv (line 37) | def bboxloginv(dx, dy, dw, dh, axc, ayc, aww, ahh): function nms (line 44) | def nms(dets, thresh): function encode (line 67) | def encode(matched, priors, variances): function decode (line 91) | def decode(loc, priors, variances): function batch_decode (line 111) | def batch_decode(loc, priors, variances): FILE: third_part/face_detection/detection/sfd/detect.py function detect (line 19) | def detect(net, img, device): function batch_detect (line 58) | def batch_detect(net, imgs, device): function flip_detect (line 99) | def flip_detect(net, img, device): function pts_to_bb (line 112) | def pts_to_bb(pts): FILE: third_part/face_detection/detection/sfd/net_s3fd.py class L2Norm (line 6) | class L2Norm(nn.Module): method __init__ (line 7) | def __init__(self, n_channels, scale=1.0): method forward (line 16) | def forward(self, x): class s3fd (line 22) | class s3fd(nn.Module): method __init__ (line 23) | def __init__(self): method forward (line 70) | def forward(self, x): FILE: third_part/face_detection/detection/sfd/sfd_detector.py class SFDDetector (line 16) | class SFDDetector(FaceDetector): method __init__ (line 17) | def __init__(self, device, path_to_detector='/apdcephfs/share_1290939/... method detect_from_image (line 31) | def detect_from_image(self, tensor_or_path): method detect_from_batch (line 41) | def detect_from_batch(self, images): method reference_scale (line 50) | def reference_scale(self): method reference_x_shift (line 54) | def reference_x_shift(self): method reference_y_shift (line 58) | def reference_y_shift(self): FILE: third_part/face_detection/models.py function conv3x3 (line 7) | def conv3x3(in_planes, out_planes, strd=1, padding=1, bias=False): class ConvBlock (line 13) | class ConvBlock(nn.Module): method __init__ (line 14) | def __init__(self, in_planes, out_planes): method forward (line 33) | def forward(self, x): class Bottleneck (line 58) | class Bottleneck(nn.Module): method __init__ (line 62) | def __init__(self, inplanes, planes, stride=1, downsample=None): method forward (line 75) | def forward(self, x): class HourGlass (line 98) | class HourGlass(nn.Module): method __init__ (line 99) | def __init__(self, num_modules, depth, num_features): method _generate_network (line 107) | def _generate_network(self, level): method _forward (line 119) | def _forward(self, level, inp): method forward (line 141) | def forward(self, x): class FAN (line 145) | class FAN(nn.Module): method __init__ (line 147) | def __init__(self, num_modules=1): method forward (line 174) | def forward(self, x): class ResNetDepth (line 204) | class ResNetDepth(nn.Module): method __init__ (line 206) | def __init__(self, block=Bottleneck, layers=[3, 8, 36, 3], num_classes... method _make_layer (line 229) | def _make_layer(self, block, planes, blocks, stride=1): method forward (line 246) | def forward(self, x): FILE: third_part/face_detection/utils.py function _gaussian (line 11) | def _gaussian( function draw_gaussian (line 37) | def draw_gaussian(image, point, sigma): function transform (line 56) | def transform(point, center, scale, resolution, invert=False): function crop (line 92) | def crop(image, center, scale, resolution=256.0): function get_preds_fromhm (line 132) | def get_preds_fromhm(hm, center=None, scale=None): function get_preds_fromhm_batch (line 172) | def get_preds_fromhm_batch(hm, centers=None, scales=None): function shuffle_lr (line 212) | def shuffle_lr(parts, pairs=None): function flip (line 237) | def flip(tensor, is_label=False): function appdata_dir (line 259) | def appdata_dir(appname=None, roaming=False): FILE: third_part/ganimation_replicate/data/base_dataset.py class BaseDataset (line 11) | class BaseDataset(torch.utils.data.Dataset): method __init__ (line 13) | def __init__(self): method name (line 16) | def name(self): method initialize (line 19) | def initialize(self, opt): method make_dataset (line 36) | def make_dataset(self): method load_dict (line 39) | def load_dict(self, pkl_path): method get_img_by_path (line 45) | def get_img_by_path(self, img_path): method get_aus_by_path (line 50) | def get_aus_by_path(self, img_path): method img_transformer (line 53) | def img_transformer(self): method __len__ (line 75) | def __len__(self): FILE: third_part/ganimation_replicate/data/celeba.py class CelebADataset (line 7) | class CelebADataset(BaseDataset): method __init__ (line 9) | def __init__(self): method initialize (line 12) | def initialize(self, opt): method get_aus_by_path (line 15) | def get_aus_by_path(self, img_path): method make_dataset (line 20) | def make_dataset(self): method __getitem__ (line 30) | def __getitem__(self, index): FILE: third_part/ganimation_replicate/data/data_loader.py function create_dataloader (line 12) | def create_dataloader(opt): class DataLoader (line 18) | class DataLoader: method name (line 19) | def name(self): method create_datase (line 22) | def create_datase(self): method initialize (line 32) | def initialize(self, opt): method __len__ (line 42) | def __len__(self): method __iter__ (line 45) | def __iter__(self): FILE: third_part/ganimation_replicate/model/__init__.py function create_model (line 7) | def create_model(opt): FILE: third_part/ganimation_replicate/model/base_model.py class BaseModel (line 8) | class BaseModel: method __init__ (line 10) | def __init__(self): method initialize (line 14) | def initialize(self, opt): method setup (line 22) | def setup(self): method set_eval (line 42) | def set_eval(self): method set_train (line 54) | def set_train(self): method set_requires_grad (line 63) | def set_requires_grad(self, parameters, requires_grad=False): method get_latest_visuals (line 70) | def get_latest_visuals(self, visuals_name): method get_latest_losses (line 77) | def get_latest_losses(self, losses_name): method feed_batch (line 87) | def feed_batch(self, batch): method forward (line 90) | def forward(self): method optimize_paras (line 93) | def optimize_paras(self): method update_learning_rate (line 96) | def update_learning_rate(self): method save_ckpt (line 102) | def save_ckpt(self, epoch, models_name): method load_ckpt (line 116) | def load_ckpt(self, epoch, models_name): method clean_ckpt (line 140) | def clean_ckpt(self, epoch, models_name): method gradient_penalty (line 148) | def gradient_penalty(self, input_img, generate_img): FILE: third_part/ganimation_replicate/model/ganimation.py class GANimationModel (line 6) | class GANimationModel(BaseModel): method __init__ (line 8) | def __init__(self): method initialize (line 12) | def initialize(self): method setup (line 29) | def setup(self): method feed_batch (line 43) | def feed_batch(self, batch): method forward (line 50) | def forward(self): method backward_dis (line 60) | def backward_dis(self): method backward_gen (line 80) | def backward_gen(self): method optimize_paras (line 104) | def optimize_paras(self, train_gen): method save_ckpt (line 119) | def save_ckpt(self, epoch): method load_ckpt (line 124) | def load_ckpt(self, epoch): method clean_ckpt (line 131) | def clean_ckpt(self, epoch): method get_latest_losses (line 136) | def get_latest_losses(self): method get_latest_visuals (line 140) | def get_latest_visuals(self): FILE: third_part/ganimation_replicate/model/model_utils.py function get_norm_layer (line 14) | def get_norm_layer(norm_type='instance'): function get_scheduler (line 34) | def get_scheduler(optimizer, opt): function init_weights (line 49) | def init_weights(net, init_type='normal', gain=0.02): function init_net (line 73) | def init_net(net, init_type='normal', init_gain=0.02, gpu_ids=[]): function define_G (line 83) | def define_G(input_nc, output_nc, ngf, which_model_netG, norm='batch', u... function define_D (line 100) | def define_D(input_nc, ndf, which_model_netD, class GANLoss (line 126) | class GANLoss(nn.Module): method __init__ (line 127) | def __init__(self, gan_type='wgan-gp', target_real_label=1.0, target_f... method get_target_tensor (line 141) | def get_target_tensor(self, input, target_is_real): method __call__ (line 148) | def __call__(self, input, target_is_real): class ResnetGenerator (line 160) | class ResnetGenerator(nn.Module): method __init__ (line 161) | def __init__(self, input_nc, output_nc, ngf=64, norm_layer=nn.BatchNor... method forward (line 204) | def forward(self, input): class ResnetBlock (line 209) | class ResnetBlock(nn.Module): method __init__ (line 210) | def __init__(self, dim, padding_type, norm_layer, use_dropout, use_bias): method build_conv_block (line 214) | def build_conv_block(self, dim, padding_type, norm_layer, use_dropout,... method forward (line 246) | def forward(self, x): class UnetGenerator (line 255) | class UnetGenerator(nn.Module): method __init__ (line 256) | def __init__(self, input_nc, output_nc, num_downs, ngf=64, method forward (line 271) | def forward(self, input): class UnetSkipConnectionBlock (line 278) | class UnetSkipConnectionBlock(nn.Module): method __init__ (line 279) | def __init__(self, outer_nc, inner_nc, input_nc=None, method forward (line 324) | def forward(self, x): class NLayerDiscriminator (line 332) | class NLayerDiscriminator(nn.Module): method __init__ (line 333) | def __init__(self, input_nc, ndf=64, n_layers=3, norm_layer=nn.BatchNo... method forward (line 375) | def forward(self, input): class PixelDiscriminator (line 379) | class PixelDiscriminator(nn.Module): method __init__ (line 380) | def __init__(self, input_nc, ndf=64, norm_layer=nn.BatchNorm2d, use_si... method forward (line 400) | def forward(self, input): function define_splitG (line 407) | def define_splitG(img_nc, aus_nc, ngf, use_dropout=False, norm='instance... function define_splitD (line 413) | def define_splitD(input_nc, aus_nc, image_size, ndf, norm='instance', in... class SplitGenerator (line 419) | class SplitGenerator(nn.Module): method __init__ (line 420) | def __init__(self, img_nc, aus_nc, ngf=64, norm_layer=nn.BatchNorm2d, ... method forward (line 474) | def forward(self, img, au): class SplitDiscriminator (line 485) | class SplitDiscriminator(nn.Module): method __init__ (line 486) | def __init__(self, input_nc, aus_nc, image_size=128, ndf=64, n_layers=... method forward (line 519) | def forward(self, img): class TVLoss (line 527) | class TVLoss(nn.Module): method __init__ (line 528) | def __init__(self, TVLoss_weight=1): method forward (line 532) | def forward(self,x): method _tensor_size (line 542) | def _tensor_size(self,t): FILE: third_part/ganimation_replicate/model/stargan.py class StarGANModel (line 7) | class StarGANModel(BaseModel): method __init__ (line 9) | def __init__(self): method initialize (line 13) | def initialize(self, opt): method setup (line 28) | def setup(self): method feed_batch (line 42) | def feed_batch(self, batch): method forward (line 49) | def forward(self): method backward_dis (line 57) | def backward_dis(self): method backward_gen (line 77) | def backward_gen(self): method optimize_paras (line 93) | def optimize_paras(self, train_gen): method save_ckpt (line 108) | def save_ckpt(self, epoch): method load_ckpt (line 113) | def load_ckpt(self, epoch): method clean_ckpt (line 120) | def clean_ckpt(self, epoch): method get_latest_losses (line 125) | def get_latest_losses(self): method get_latest_visuals (line 129) | def get_latest_visuals(self): FILE: third_part/ganimation_replicate/options.py class Options (line 13) | class Options(object): method __init__ (line 15) | def __init__(self): method initialize (line 18) | def initialize(self): method parse (line 92) | def parse(self): FILE: third_part/ganimation_replicate/solvers.py function create_solver (line 17) | def create_solver(opt): class Solver (line 24) | class Solver(object): method __init__ (line 26) | def __init__(self): method initialize (line 29) | def initialize(self, opt): method run_solver (line 34) | def run_solver(self): method train_networks (line 40) | def train_networks(self): method init_train_setting (line 57) | def init_train_setting(self): method train_epoch (line 65) | def train_epoch(self, epoch): method test_networks (line 104) | def test_networks(self, opt): method init_test_setting (line 108) | def init_test_setting(self, opt): method test_ops (line 112) | def test_ops(self): method test_save_imgs (line 133) | def test_save_imgs(self, faces_list, paths_list): FILE: third_part/ganimation_replicate/visualizer.py class Visualizer (line 10) | class Visualizer(object): method __init__ (line 12) | def __init__(self): method initialize (line 15) | def initialize(self, opt): method throw_visdom_connection_error (line 28) | def throw_visdom_connection_error(self): method print_losses_info (line 32) | def print_losses_info(self, info_dict): method display_current_losses (line 44) | def display_current_losses(self, epoch, counter_ratio, losses_dict): method display_online_results (line 62) | def display_online_results(self, visuals, epoch): method tensor2im (line 80) | def tensor2im(self, input_image, imtype=np.uint8): method numpy2im (line 89) | def numpy2im(self, image_numpy, imtype=np.uint8): FILE: utils/alignment_stit.py function paste_image (line 15) | def paste_image(inverse_transform, img, orig_image): function get_landmark (line 21) | def get_landmark(filepath, predictor, detector=None, fa=None): function align_face (line 53) | def align_face(filepath_or_image, predictor, output_size, detector=None, function crop_image (line 69) | def crop_image(filepath, output_size, quad, enable_padding=False): function compute_transform (line 117) | def compute_transform(lm, predictor, detector=None, scale=1.0, fa=None): function crop_faces (line 150) | def crop_faces(IMAGE_SIZE, files, scale, center_sigma=0.0, xy_sigma=0.0,... function crop_faces_by_quads (line 190) | def crop_faces_by_quads(IMAGE_SIZE, files, quads): function calc_alignment_coefficients (line 201) | def calc_alignment_coefficients(pa, pb): FILE: utils/audio.py function load_wav (line 9) | def load_wav(path, sr): function save_wav (line 12) | def save_wav(wav, path, sr): function save_wavenet_wav (line 17) | def save_wavenet_wav(wav, path, sr): function preemphasis (line 20) | def preemphasis(wav, k, preemphasize=True): function inv_preemphasis (line 25) | def inv_preemphasis(wav, k, inv_preemphasize=True): function get_hop_size (line 30) | def get_hop_size(): function linearspectrogram (line 37) | def linearspectrogram(wav): function melspectrogram (line 45) | def melspectrogram(wav): function _lws_processor (line 53) | def _lws_processor(): function _stft (line 57) | def _stft(y): function num_frames (line 65) | def num_frames(length, fsize, fshift): function pad_lr (line 76) | def pad_lr(x, fsize, fshift): function librosa_pad_lr (line 86) | def librosa_pad_lr(x, fsize, fshift): function _linear_to_mel (line 92) | def _linear_to_mel(spectogram): function _build_mel_basis (line 98) | def _build_mel_basis(): function _amp_to_db (line 103) | def _amp_to_db(x): function _db_to_amp (line 107) | def _db_to_amp(x): function _normalize (line 110) | def _normalize(S): function _denormalize (line 124) | def _denormalize(D): FILE: utils/ffhq_preprocess.py class Croper (line 34) | class Croper: method __init__ (line 35) | def __init__(self, path_of_lm): method get_landmark (line 39) | def get_landmark(self, img_np): method align_face (line 57) | def align_face(self, img, lm, output_size=1024): method crop (line 118) | def crop(self, img_np_list, xsize=512): # first frame for all video FILE: utils/flow_util.py function convert_flow_to_deformation (line 3) | def convert_flow_to_deformation(flow): function make_coordinate_grid (line 17) | def make_coordinate_grid(flow): function warp_image (line 41) | def warp_image(source_image, deformation): FILE: utils/hparams.py class HParams (line 3) | class HParams: method __init__ (line 4) | def __init__(self, **kwargs): method __getattr__ (line 10) | def __getattr__(self, key): method set_hparam (line 15) | def set_hparam(self, key, value): function hparams_debug_string (line 134) | def hparams_debug_string(): FILE: utils/inference_utils.py function options (line 15) | def options(): function mask_postprocess (line 57) | def mask_postprocess(mask, thres=20): function trans_image (line 64) | def trans_image(image): function obtain_seq_index (line 71) | def obtain_seq_index(index, num_frames): function transform_semantic (line 76) | def transform_semantic(semantic, frame_index, crop_norm_ratio=None): function find_crop_norm_ratio (line 91) | def find_crop_norm_ratio(source_coeff, target_coeffs): function get_smoothened_boxes (line 99) | def get_smoothened_boxes(boxes, T): function face_detect (line 108) | def face_detect(images, args, jaw_correction=False, detector=None): function _load (line 149) | def _load(checkpoint_path, device): function split_coeff (line 157) | def split_coeff(coeffs): function Laplacian_Pyramid_Blending_with_mask (line 180) | def Laplacian_Pyramid_Blending_with_mask(A, B, m, num_levels = 6): function load_model (line 223) | def load_model(args, device): function normalize_kp (line 228) | def normalize_kp(kp_source, kp_driving, kp_driving_initial, adapt_moveme... function load_face3d_net (line 248) | def load_face3d_net(ckpt_path, device): FILE: webUI.py function convert (line 11) | def convert(segment_length, video, audio, progress=gradio.Progress()): function cleanup_temp_files (line 42) | def cleanup_temp_files(file_list): function cut_video_segments (line 48) | def cut_video_segments(video_file, segment_length): function cut_audio_segments (line 62) | def cut_audio_segments(audio_file, segment_length): function process_segment (line 76) | def process_segment(video_seg, audio_seg, i): function concatenate_videos (line 85) | def concatenate_videos(video_segments, output_file):