SYMBOL INDEX (223 symbols across 30 files)

FILE: fusion/affineFace.py
  function gammaTrans (line 13) | def gammaTrans(img, gamma):
  function erodeAndBlur (line 18) | def erodeAndBlur(img,kernelSize=21,blurSize=21):
  function affineface (line 25) | def affineface(img,src_pt,dst_pt,heatmapSize=256,needImg=True):
  function affineface_parts (line 41) | def affineface_parts(img,src_pt,dst_pt):
  function lightEye (line 73) | def lightEye(img_ref,lms_ref,img_gen,lms_gen,ratio=0.1):
  function multi (line 109) | def multi(img,mask):
  function fusion (line 117) | def fusion(img_ref,lms_ref,img_gen,lms_gen,ratio=0.2):
  function loaddata (line 150) | def loaddata(head,path_lms,flag=256,num = 50000):
  function gray2rgb (line 167) | def gray2rgb(img):
  function process (line 171) | def process(index, album_ref, album_gen, album_pose):

FILE: fusion/calcAffine.py
  function calAffine (line 14) | def calAffine(src_p, dst_p):
  function affinePts (line 34) | def affinePts(affine_mat,pt):
  function affineImg (line 44) | def affineImg(img,TransMat,dsize = 256):

FILE: fusion/parts2lms.py
  function parts2lms (line 3) | def parts2lms(parts):

FILE: fusion/points2heatmap.py
  function curve_interp (line 6) | def curve_interp(points, heatmapSize=256, sigma=3):
  function curve_fill (line 14) | def curve_fill(points, heatmapSize=256, sigma=3, erode=False):
  function curves2heatmap (line 30) | def curves2heatmap(curves,heatmapSize=256,sigma=3,flag='line'):
  function curves2segments (line 47) | def curves2segments(curves,heatmapSize=256,sigma=3):
  function curves2gaze (line 71) | def curves2gaze(curves,heatmapSize=256,sigma=3):
  function curves2parts (line 81) | def curves2parts(curves):
  function points2curves (line 96) | def points2curves(points, heatmapSize=256,  sigma=1, heatmap_num=17):
  function distance (line 212) | def distance(p1, p2):
  function curve_fitting (line 215) | def curve_fitting(points, heatmap_size, sigma):

FILE: fusion/test.py
  function func (line 6) | def func(i):

FILE: fusion/warper.py
  function warping (line 8) | def warping(img, src_bound, dst_bound, size=(256, 256)):
  function bilinear_interpolate (line 21) | def bilinear_interpolate(img, coords):
  function grid_coordinates (line 46) | def grid_coordinates(points):
  function process_warp (line 60) | def process_warp(src_img, result_img, tri_affines, dst_points, delaunay):
  function triangular_affine_matrices (line 80) | def triangular_affine_matrices(vertices, src_points, dest_points):
  function warp_image (line 98) | def warp_image(src_img, src_points, dest_points, dest_shape, dtype=np.ui...

FILE: loader/dataset_basic.py
  class DatasetBasic (line 13) | class DatasetBasic(torch.utils.data.Dataset):
    method __init__ (line 14) | def __init__(self, imgSize=256):
    method __len__ (line 21) | def __len__(self):
    method shape (line 24) | def shape(self):
    method loadtxt (line 27) | def loadtxt(self, path, head=''):
    method loadtxtList (line 42) | def loadtxtList(self, pathList, head):
    method warp (line 48) | def warp(self, img, srcPt, dstPt):
    method np2tensor (line 54) | def np2tensor(self, img, scale=1/255.0):
    method points2heatmap (line 64) | def points2heatmap(self, landmarks, mapSize, sigma, landmarkSize=255.0...
    method gammaTrans (line 134) | def gammaTrans(self, img, gamma):
    method __getitem__ (line 139) | def __getitem__(self, index):

FILE: loader/dataset_loader_demo.py
  class DatasetLoaderDemo (line 8) | class DatasetLoaderDemo(DatasetBasic):
    method __init__ (line 9) | def __init__(self, imgSize=256, gaze=True):
    method loadBounds (line 16) | def loadBounds(self, pathList, head):
    method loadAppears (line 19) | def loadAppears(self, pathList, head):
    method setAppearRule (line 22) | def setAppearRule(self, flag='random'):
    method findSimilar (line 36) | def findSimilar(self, pt_dst):
    method adjustPose (line 48) | def adjustPose(self, img_src, pt_src, pt_dst):
    method add_nose_bridge (line 52) | def add_nose_bridge(self, boundary, heatmap):
    method __getitem__ (line 60) | def __getitem__(self, index):
    method __len__ (line 101) | def __len__(self):

FILE: loader/dataset_loader_train.py
  class DatasetLoaderTrain (line 9) | class DatasetLoaderTrain(DatasetBasic):
    method __init__ (line 10) | def __init__(self, imgSize=256, gaze=True):
    method setSampleCurve (line 21) | def setSampleCurve(self, flag='Bound'):
    method transform (line 24) | def transform(self, img, pt):
    method loaddata (line 30) | def loaddata(self, pathList, head):
    method sampleCurve (line 35) | def sampleCurve(self, img, pt, flag='Bound'):
    method add_nose_bridge (line 51) | def add_nose_bridge(self, boundary, heatmap):
    method __getitem__ (line 59) | def __getitem__(self, index):
    method __len__ (line 87) | def __len__(self):

FILE: model/base_model.py
  class BaseModel (line 13) | class BaseModel():
    method modify_commandline_options (line 18) | def modify_commandline_options(parser, is_train):
    method name (line 21) | def name(self):
    method initialize (line 24) | def initialize(self, opt):
    method set_input (line 40) | def set_input(self, input):
    method forward (line 43) | def forward(self):
    method set_logger (line 46) | def set_logger(self, opt):
    method get_logger (line 59) | def get_logger(self, logdir):
    method save_config (line 71) | def save_config(self, config):
    method setup (line 80) | def setup(self, opt, parser=None):
    method load_networks_all (line 91) | def load_networks_all(self, prefix):
    method load_networks (line 101) | def load_networks(self, model, path):
    method eval (line 117) | def eval(self):
    method test (line 125) | def test(self):
    method get_image_paths (line 130) | def get_image_paths(self):
    method optimize_parameters (line 133) | def optimize_parameters(self):
    method update_learning_rate (line 137) | def update_learning_rate(self):
    method save_networks (line 161) | def save_networks(self, epoch):
    method print_networks (line 178) | def print_networks(self, verbose):
    method set_requires_grad (line 194) | def set_requires_grad(self, nets, requires_grad=False):

FILE: model/spade_model.py
  class SpadeModel (line 16) | class SpadeModel(BaseModel):
    method __init__ (line 17) | def __init__(self, opt):
    method set_input (line 86) | def set_input(self, input):
    method forward (line 99) | def forward(self):
    method backward_D_basic (line 111) | def backward_D_basic(self, netD, real, fake):
    method backward_D (line 122) | def backward_D(self):
    method backward_G (line 129) | def backward_G(self, epoch):
    method func_require_grad (line 172) | def func_require_grad(self, model_, flag_):
    method func_zero_grad (line 176) | def func_zero_grad(self, model_):
    method optimize_parameters (line 180) | def optimize_parameters(self, epoch):

FILE: net/ResNet.py
  function conv3x3 (line 23) | def conv3x3(in_planes, out_planes, stride=1):
  class BasicBlock (line 29) | class BasicBlock(nn.Module):
    method __init__ (line 32) | def __init__(self, inplanes, planes, stride=1, downsample=None):
    method forward (line 42) | def forward(self, x):
  class IRBlock (line 61) | class IRBlock(nn.Module):
    method __init__ (line 64) | def __init__(self, inplanes, planes, stride=1, downsample=None, use_se...
    method forward (line 78) | def forward(self, x):
  class Bottleneck (line 99) | class Bottleneck(nn.Module):
    method __init__ (line 102) | def __init__(self, inplanes, planes, stride=1, downsample=None):
    method forward (line 116) | def forward(self, x):
  class SEBlock (line 139) | class SEBlock(nn.Module):
    method __init__ (line 140) | def __init__(self, channel, reduction=16):
    method forward (line 150) | def forward(self, x):
  class ResNetFace (line 157) | class ResNetFace(nn.Module):
    method __init__ (line 158) | def __init__(self, block, layers, input_nc, use_se=True):
    method _make_layer (line 186) | def _make_layer(self, block, planes, blocks, stride=1):
    method forward (line 203) | def forward(self, x):
  class ResNet (line 222) | class ResNet(nn.Module):
    method __init__ (line 224) | def __init__(self, block, layers):
    method _make_layer (line 250) | def _make_layer(self, block, planes, blocks, stride=1):
    method forward (line 267) | def forward(self, x):
  function resnet18 (line 285) | def resnet18(pretrained=False, **kwargs):
  function resnet34 (line 296) | def resnet34(pretrained=False, **kwargs):
  function resnet50 (line 307) | def resnet50(pretrained=False, **kwargs):
  function resnet101 (line 318) | def resnet101(pretrained=False, **kwargs):
  function resnet152 (line 329) | def resnet152(pretrained=False, **kwargs):
  function resnet_face18 (line 340) | def resnet_face18(input_nc, use_se=True, **kwargs):

FILE: net/appear_decoder_net.py
  function defineAppDec (line 9) | def defineAppDec(input_nc, size_=256, norm='batch', init_type='normal', ...
  class appearDec (line 19) | class appearDec(nn.Module):
    method __init__ (line 20) | def __init__(self, input_c, norm_layer, size_=256):
    method forward (line 44) | def forward(self, input):
  class appearDec128 (line 52) | class appearDec128(nn.Module):
    method __init__ (line 53) | def __init__(self, input_c, norm_layer, size_=256):
    method forward (line 77) | def forward(self, input):

FILE: net/appear_encoder_net.py
  function defineAppEnc (line 9) | def defineAppEnc(input_nc, size_=256, norm='batch', init_type='normal', ...
  class appearEnc (line 60) | class appearEnc(nn.Module):
    method __init__ (line 61) | def __init__(self, input_c, norm_layer, size_=256, conv_k=4):
    method sample_z (line 86) | def sample_z(self, z_mu):
    method kl_loss (line 96) | def kl_loss(self, z_mu):
    method freeze (line 106) | def freeze(self):
    method forward (line 125) | def forward(self, input):

FILE: net/base_net.py
  function get_norm_layer (line 12) | def get_norm_layer(norm_type='instance'):
  function get_scheduler (line 26) | def get_scheduler(optimizer, opt):
  function init_weights (line 47) | def init_weights(net, init_type='normal', gain=0.02):
  function init_net (line 75) | def init_net(net, init_type='normal', init_gain=0.02, gpu_ids=[]):

FILE: net/discriminator_net.py
  function define_D (line 10) | def define_D(input_nc, ndf, netD,
  class GANLoss (line 39) | class GANLoss(nn.Module):
    method __init__ (line 40) | def __init__(self, use_lsgan=True, target_real_label=1.0, target_fake_...
    method get_target_tensor (line 49) | def get_target_tensor(self, input, target_is_real):
    method __call__ (line 56) | def __call__(self, input, target_is_real):
  class NLayerDiscriminator (line 62) | class NLayerDiscriminator(nn.Module):
    method __init__ (line 63) | def __init__(self, input_nc, ndf=64, n_layers=3, norm_layer=nn.BatchNo...
    method forward (line 106) | def forward(self, input):
  class PixelDiscriminator (line 110) | class PixelDiscriminator(nn.Module):
    method __init__ (line 111) | def __init__(self, input_nc, ndf=64, norm_layer=nn.BatchNorm2d, use_si...
    method forward (line 132) | def forward(self, input):

FILE: net/face_id_mlp_net.py
  class MLP (line 8) | class MLP(nn.Module):
    method __init__ (line 9) | def __init__(self, input_nc, output_nc):
    method forward (line 13) | def forward(self, input):
  class ArcMarginProduct (line 17) | class ArcMarginProduct(nn.Module):
    method __init__ (line 27) | def __init__(self, in_features, out_features, s=30.0, m=0.50, easy_mar...
    method forward (line 42) | def forward(self, input, label):
  class AddMarginProduct (line 64) | class AddMarginProduct(nn.Module):
    method __init__ (line 74) | def __init__(self, in_features, out_features, s=30.0, m=0.40):
    method forward (line 83) | def forward(self, input, label):
    method __repr__ (line 99) | def __repr__(self):
  class SphereProduct (line 107) | class SphereProduct(nn.Module):
    method __init__ (line 116) | def __init__(self, in_features, out_features, m=4):
    method forward (line 139) | def forward(self, input, label):
    method __repr__ (line 166) | def __repr__(self):

FILE: net/face_id_net.py
  function defineFaceID (line 11) | def defineFaceID(input_nc=3, class_num=10173, init_type='normal', init_g...
  class faceIDNet (line 17) | class faceIDNet(nn.Module):
    method __init__ (line 18) | def __init__(self, input_nc, class_num):
    method forward (line 24) | def forward(self, input):
    method face_id_loss (line 29) | def face_id_loss(self, x, target, loss_func):

FILE: net/generaotr_net.py
  function defineSPADEGenerator (line 14) | def defineSPADEGenerator(input_nc, output_nc, ngf, norm='instance', use_...
  class BasicSPADE (line 61) | class BasicSPADE(nn.Module):
    method __init__ (line 62) | def __init__(self, norm_layer, input_nc, planes):
    method forward (line 68) | def forward(self, x, bound):
  class ResBlkSPADE (line 76) | class ResBlkSPADE(nn.Module):
    method __init__ (line 77) | def __init__(self, norm_layer, input_nc, planes, conv_kernel_size=1, p...
    method forward (line 87) | def forward(self, x, bound):
  class SPADEGenerator (line 108) | class SPADEGenerator(nn.Module):
    method __init__ (line 109) | def __init__(self, input_nc, output_nc, ngf=64,
    method forward (line 169) | def forward(self, input, latent_z, decoder_result): # input: bound, ba...

FILE: net/generator_net_concat_1Layer.py
  function defineSPADEGenerator (line 14) | def defineSPADEGenerator(input_nc, output_nc, ngf, norm='instance', use_...
  class BasicSPADE (line 25) | class BasicSPADE(nn.Module):
    method __init__ (line 26) | def __init__(self, norm_layer, input_nc, planes):
    method forward (line 32) | def forward(self, x, bound):
  class ResBlkSPADE (line 40) | class ResBlkSPADE(nn.Module):
    method __init__ (line 41) | def __init__(self, norm_layer, input_nc, planes, conv_kernel_size=1, p...
    method forward (line 51) | def forward(self, x, bound):
  class SPADEGenerator (line 72) | class SPADEGenerator(nn.Module):
    method __init__ (line 73) | def __init__(self, input_nc, output_nc, ngf=64,
    method forward (line 161) | def forward(self, input, latent_z):

FILE: net/vgg_net.py
  function defineVGG (line 12) | def defineVGG(init_type='normal', init_gain=0.02, gpu_ids=[]):
  class VGGNet (line 17) | class VGGNet(nn.Module):
    method __init__ (line 18) | def __init__(self):
    method forward (line 24) | def forward(self, x):
    method perceptual_loss (line 41) | def perceptual_loss(self, x, target, loss_func):
    method style_loss (line 49) | def style_loss(self, x, target, loss_func):

FILE: opt/config.py
  class BaseOptions (line 6) | class BaseOptions():
    method __init__ (line 7) | def __init__(self):
    method initialize (line 11) | def initialize(self, misc_arg):
    method get_config (line 106) | def get_config(self):

FILE: opt/configTrain.py
  class TrainOptions (line 4) | class TrainOptions(BaseOptions):
    method initialize (line 9) | def initialize(self, misc_arg):

FILE: utils/affineFace.py
  function affineface (line 5) | def affineface(img, src_pt, dst_pt, heatmapSize=256):

FILE: utils/affine_util.py
  function th_affine2d (line 16) | def th_affine2d(x, matrix, output_img_width, output_img_height, center=T...
  function AffinePoint (line 42) | def AffinePoint(point, affine_mat):
  function exchange_landmarks (line 63) | def exchange_landmarks(input_tf, corr_list):

FILE: utils/calcAffine.py
  function calAffine (line 17) | def calAffine(src_p, dst_p):
  function affinePts (line 36) | def affinePts(affine_mat, pt):
  function affineImg (line 47) | def affineImg(img, TransMat, dsize=256):

FILE: utils/metric.py
  function gram_matrix (line 5) | def gram_matrix(feat):

FILE: utils/points2heatmap.py
  function curve_interp (line 6) | def curve_interp(points, heatmapSize=256, sigma=3):
  function curve_fill (line 14) | def curve_fill(points, heatmapSize=256, sigma=3, erode=False):
  function curves2heatmap (line 30) | def curves2heatmap(curves,heatmapSize=256,sigma=3,flag='line'):
  function curves2segments (line 47) | def curves2segments(curves,heatmapSize=256,sigma=3):
  function points2curves (line 72) | def points2curves(points, heatmapSize=256,  sigma=1, heatmap_num=17):
  function distance (line 201) | def distance(p1, p2):
  function curve_fitting (line 204) | def curve_fitting(points, heatmap_size, sigma):

FILE: utils/transforms.py
  function initAlignTransfer (line 16) | def initAlignTransfer(size, mirror=False, gaze=True):
  class AffineCompose (line 39) | class AffineCompose(object):
    method __init__ (line 41) | def __init__(self,
    method __call__ (line 57) | def __call__(self, *inputs):
  function dealcurve (line 122) | def dealcurve(curve):

FILE: utils/warper.py
  function warping (line 8) | def warping(img, src_bound, dst_bound, size=(256, 256)):
  function bilinear_interpolate (line 21) | def bilinear_interpolate(img, coords):
  function grid_coordinates (line 46) | def grid_coordinates(points):
  function process_warp (line 60) | def process_warp(src_img, result_img, tri_affines, dst_points, delaunay):
  function triangular_affine_matrices (line 80) | def triangular_affine_matrices(vertices, src_points, dest_points):
  function warp_image (line 98) | def warp_image(src_img, src_points, dest_points, dest_shape, dtype=np.ui...