Functions — github.com/Kevinfringe/MegaPortrait

CodeHub MCPcopy Create free account

hub / github.com/Kevinfringe/MegaPortrait / functions

Functions46 in github.com/Kevinfringe/MegaPortrait

⨍ Functions 46 ◇ Types & classes 17

↓ 11 callersMethod__init__

(self, input_channels)

↓ 4 callersFunctioncosine_distance

↓ 2 callersFunctioncapture_frames

:param n: :param path: :param flag: if flag = 1, then it generate source img, otherwise, generate driver img. :return:

data_generator.py:27

↓ 1 callersMethod__init__

VGGFace model. Face recognition network. It takes as input a Bx3x224x224 batch of face images and gives as output a BxC score vector

↓ 1 callersFunctioncosine_loss

( args, descriptor_driver, descriptor_source_rand, descriptor_driver_rand)

↓ 1 callersFunctionmain

↓ 1 callersFunctiontensor2image

(image, norm=255.0)

↓ 1 callersFunctiontrain

(args, models, device, driver_loader, source_loader, optimizers, schedulers, source_img_random, driver_img_ran

Method__getitem__

Method__getitem__

(self, in_channels=3)

(self, source_img_path, length, transform=None)

(self, driver_img_path, length, transform=None)

Create a block with len(units) - 1 convolutions. convolution number i transforms the number of channels from units[i - 1] to units[i]

(self, in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bia

(self, in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bia

(self, dimension, input_channels, output_channels)

(self, repeat, in_channels=3, outputs=256)

(self, in_channels, out_channels, downsample=False)

(self, in_channels, resblock=ResBlock, outputs=256)

(self, in_channels, resblock, outputs=256)

(self, input_channels, output_channels)

(self, input_channels)

(self, input_channels)

Methoddiscriminator_block

Returns downsampling layers of each discriminator block

Functionface_detect

data_generator.py:14

(self, img_A, img_B)

Functionhead_pose_estimation

(imgs, source=True)

HeadPoseEstimation.py:149

Functionvggface

VGGFace model. Args: pretrained (bool): If True, returns pre-trained model