SYMBOL INDEX (325 symbols across 54 files)

FILE: datasets/casia.py
  class CASIA (line 22) | class CASIA(Dataset):
    method __init__ (line 24) | def __init__(self, images_root_path, image_list_path, transform, use_l...
    method _read_samples_info (line 32) | def _read_samples_info(self):
    method get_num_classes (line 45) | def get_num_classes(self):
    method __len__ (line 49) | def __len__(self):
    method __getitem__ (line 53) | def __getitem__(self, idx):

FILE: datasets/celeba.py
  class CelebA (line 22) | class CelebA(Dataset):
    method __init__ (line 24) | def __init__(self, images_root_path, landmarks_folder_path, transform=...
    method _read_samples_info (line 35) | def _read_samples_info(self):
    method __len__ (line 65) | def __len__(self):
    method __getitem__ (line 69) | def __getitem__(self, idx):

FILE: datasets/imdbface.py
  class IMDBFace (line 23) | class IMDBFace(Dataset):
    method __init__ (line 25) | def __init__(self, images_root_path, image_list_path, transform=None):
    method _read_samples_info (line 37) | def _read_samples_info(self):
    method get_weights (line 65) | def get_weights(self):
    method get_num_classes (line 72) | def get_num_classes(self):
    method __len__ (line 76) | def __len__(self):
    method __getitem__ (line 80) | def __getitem__(self, idx):

FILE: datasets/lfw.py
  class LFW (line 22) | class LFW(Dataset):
    method __init__ (line 24) | def __init__(self, images_root_path, pairs_path, landmark_file_path=''...
    method _read_landmarks (line 34) | def _read_landmarks(self):
    method _read_pairs (line 45) | def _read_pairs(self):
    method _load_img (line 75) | def _load_img(self, img_path):
    method show_item (line 88) | def show_item(self, index):
    method __getitem__ (line 103) | def __getitem__(self, index):
    method __len__ (line 110) | def __len__(self):

FILE: datasets/megaface.py
  class MegaFace (line 21) | class MegaFace(Dataset):
    method __init__ (line 23) | def __init__(self, images_lsit, transform=None):
    method __len__ (line 27) | def __len__(self):
    method __getitem__ (line 31) | def __getitem__(self, idx):

FILE: datasets/ms_celeb1m.py
  class MSCeleb1M (line 23) | class MSCeleb1M(Dataset):
    method __init__ (line 25) | def __init__(self, images_root_path, image_list_path, transform=None):
    method _read_samples_info (line 37) | def _read_samples_info(self):
    method get_weights (line 68) | def get_weights(self):
    method get_num_classes (line 75) | def get_num_classes(self):
    method __len__ (line 79) | def __len__(self):
    method __getitem__ (line 83) | def __getitem__(self, idx):

FILE: datasets/ndg.py
  class NDG (line 22) | class NDG(Dataset):
    method __init__ (line 24) | def __init__(self, images_root_path, annotation_list, transform=None, ...
    method _read_samples_info (line 32) | def _read_samples_info(self):
    method __len__ (line 45) | def __len__(self):
    method __getitem__ (line 49) | def __getitem__(self, idx):

FILE: datasets/trillion_pairs.py
  class TrillionPairs (line 23) | class TrillionPairs(Dataset):
    method __init__ (line 25) | def __init__(self, images_root_path, image_list_path, test_mode=False,...
    method _read_samples_info (line 38) | def _read_samples_info(self):
    method get_weights (line 67) | def get_weights(self):
    method get_num_classes (line 74) | def get_num_classes(self):
    method __len__ (line 78) | def __len__(self):
    method __getitem__ (line 82) | def __getitem__(self, idx):

FILE: datasets/vggface2.py
  class VGGFace2 (line 23) | class VGGFace2(Dataset):
    method __init__ (line 25) | def __init__(self, images_root_path, image_list_path, landmarks_folder...
    method _read_samples_info (line 54) | def _read_samples_info(self):
    method get_weights (line 88) | def get_weights(self):
    method get_num_classes (line 96) | def get_num_classes(self):
    method __len__ (line 100) | def __len__(self):
    method __getitem__ (line 104) | def __getitem__(self, idx):

FILE: demo/run_demo.py
  class FaceDetector (line 27) | class FaceDetector:
    method __init__ (line 29) | def __init__(self, model_path, conf=.6, device='CPU', ext_path=''):
    method get_detections (line 34) | def get_detections(self, frame):
    method __decode_detections (line 41) | def __decode_detections(self, out, frame_shape):
  class VectorCNN (line 70) | class VectorCNN:
    method __init__ (line 72) | def __init__(self, model_path, device='CPU'):
    method forward (line 75) | def forward(self, batch):
  function get_embeddings (line 82) | def get_embeddings(frame, detections, face_reid, landmarks_predictor):
  function find_nearest (line 105) | def find_nearest(x, gallery, thr):
  function match_embeddings (line 116) | def match_embeddings(embeds, gallery, thr):
  function draw_detections (line 129) | def draw_detections(frame, detections, indexes):
  function load_gallery (line 144) | def load_gallery(path_to_gallery, face_det, landmarks_detector, face_rec...
  function run (line 162) | def run(params, capture, face_det, face_recognizer, landmarks_detector):
  function main (line 178) | def main():

FILE: dump_features.py
  function clean_megaface (line 35) | def clean_megaface(filenames, features, noises_list_path):
  function clean_facescrub (line 52) | def clean_facescrub(filenames, features, noises_list_path):
  function main (line 81) | def main(args):
  function parse_argument (line 175) | def parse_argument(argv):

FILE: evaluate_landmarks.py
  function evaluate (line 30) | def evaluate(val_loader, model):
  function start_evaluation (line 57) | def start_evaluation(args):
  function main (line 93) | def main():

FILE: evaluate_lfw.py
  function get_subset (line 37) | def get_subset(container, subset_bounds):
  function get_roc (line 45) | def get_roc(scores_with_gt, n_threshs=400):
  function get_auc (line 70) | def get_auc(fprs, tprs):
  function save_roc (line 81) | def save_roc(fp_rates, tp_rates, fname):
  function compute_embeddings_lfw (line 89) | def compute_embeddings_lfw(args, dataset, model, batch_size, dump_embedd...
  function compute_embeddings_lfw_ie (line 138) | def compute_embeddings_lfw_ie(args, dataset, model, batch_size=1, dump_e...
  function compute_optimal_thresh (line 164) | def compute_optimal_thresh(scores_with_gt):
  function evaluate (line 189) | def evaluate(args, dataset, model, compute_embeddings_fun, val_batch_siz...
  function load_test_dataset (line 272) | def load_test_dataset(arguments):
  function main (line 283) | def main():

FILE: losses/alignment.py
  function wing_core (line 21) | def wing_core(abs_x, w, eps):
  class AlignmentLoss (line 25) | class AlignmentLoss(nn.Module):
    method __init__ (line 27) | def __init__(self, loss_type='l2'):
    method set_weights (line 36) | def set_weights(self, weights):
    method forward (line 41) | def forward(self, input_values, target):

FILE: losses/am_softmax.py
  class AngleSimpleLinear (line 22) | class AngleSimpleLinear(nn.Module):
    method __init__ (line 24) | def __init__(self, in_features, out_features):
    method forward (line 31) | def forward(self, x):
  function focal_loss (line 36) | def focal_loss(input_values, gamma):
  class AMSoftmaxLoss (line 43) | class AMSoftmaxLoss(nn.Module):
    method __init__ (line 47) | def __init__(self, margin_type='cos', gamma=0., m=0.5, s=30, t=1.):
    method forward (line 63) | def forward(self, cos_theta, target):

FILE: losses/centroid_based.py
  class CenterLoss (line 20) | class CenterLoss(nn.Module):
    method __init__ (line 22) | def __init__(self, num_classes, embed_size, cos_dist=True):
    method get_centers (line 30) | def get_centers(self):
    method forward (line 34) | def forward(self, features, labels):
  class MinimumMargin (line 55) | class MinimumMargin(nn.Module):
    method __init__ (line 57) | def __init__(self, margin=.6):
    method forward (line 61) | def forward(self, centers, labels):
  class GlobalPushPlus (line 82) | class GlobalPushPlus(nn.Module):
    method __init__ (line 84) | def __init__(self, margin=.6):
    method forward (line 90) | def forward(self, features, centers, labels):
  class PushPlusLoss (line 120) | class PushPlusLoss(nn.Module):
    method __init__ (line 122) | def __init__(self, margin=.7):
    method forward (line 126) | def forward(self, features, centers, labels):
  class PushLoss (line 149) | class PushLoss(nn.Module):
    method __init__ (line 151) | def __init__(self, soft=True, margin=0.5):
    method forward (line 156) | def forward(self, features, labels):

FILE: losses/metric_losses.py
  class MetricLosses (line 18) | class MetricLosses:
    method __init__ (line 20) | def __init__(self, classes_num, embed_size, writer):
    method __call__ (line 38) | def __call__(self, features, labels, epoch_num, iteration):
    method init_iteration (line 80) | def init_iteration(self):
    method end_iteration (line 85) | def end_iteration(self):

FILE: losses/regularizer.py
  function l2_reg_ortho (line 18) | def l2_reg_ortho(mdl):
  class ODecayScheduler (line 47) | class ODecayScheduler():
    method __init__ (line 49) | def __init__(self, schedule, initial_decay, mult_factor):
    method step (line 56) | def step(self):
    method get_decay (line 64) | def get_decay(self):

FILE: model/backbones/resnet.py
  class ResNet (line 20) | class ResNet(nn.Module):
    method __init__ (line 21) | def __init__(self, block, layers, num_classes=1000, activation=nn.ReLU):
    method _make_layer (line 44) | def _make_layer(self, block, planes, blocks, stride=1, activation=nn.R...
    method forward (line 61) | def forward(self, x):
  function resnet50 (line 78) | def resnet50(**kwargs):
  function resnet34 (line 83) | def resnet34(**kwargs):

FILE: model/backbones/rmnet.py
  class RMNetBody (line 20) | class RMNetBody(nn.Module):
    method __init__ (line 21) | def __init__(self, block=RMBlock, blocks_per_stage=(None, 4, 8, 10, 11...
    method init_weights (line 49) | def init_weights(self):
    method forward (line 60) | def forward(self, x):

FILE: model/backbones/se_resnet.py
  class SEResNet (line 21) | class SEResNet(nn.Module):
    method __init__ (line 22) | def __init__(self, block, layers, num_classes=1000, activation=nn.ReLU):
    method _make_layer (line 46) | def _make_layer(self, block, planes, blocks, stride=1, activation=nn.R...
    method forward (line 63) | def forward(self, x):
  function se_resnet50 (line 80) | def se_resnet50(**kwargs):
  function se_resnet101 (line 85) | def se_resnet101(**kwargs):
  function se_resnet152 (line 90) | def se_resnet152(**kwargs):

FILE: model/backbones/se_resnext.py
  class SEResNeXt (line 20) | class SEResNeXt(nn.Module):
    method __init__ (line 22) | def __init__(self, block, layers, cardinality=32, num_classes=1000):
    method _make_layer (line 51) | def _make_layer(self, block, planes, blocks, stride=1):
    method forward (line 68) | def forward(self, x):
  function se_resnext50 (line 87) | def se_resnext50(**kwargs):
  function se_resnext101 (line 92) | def se_resnext101(**kwargs):
  function se_resnext152 (line 97) | def se_resnext152(**kwargs):

FILE: model/backbones/shufflenet_v2.py
  class ShuffleNetV2Body (line 19) | class ShuffleNetV2Body(nn.Module):
    method __init__ (line 20) | def __init__(self, input_size=224, width_mult=1.):
    method get_downscale_factor (line 62) | def get_downscale_factor():
    method init_weights (line 65) | def init_weights(self):
    method get_num_output_channels (line 70) | def get_num_output_channels(self):
    method forward (line 73) | def forward(self, x):

FILE: model/blocks/mobilenet_v2_blocks.py
  class InvertedResidual (line 19) | class InvertedResidual(nn.Module):
    method __init__ (line 21) | def __init__(self, in_channels, out_channels, stride, expand_ratio, ou...
    method forward (line 43) | def forward(self, x):

FILE: model/blocks/resnet_blocks.py
  class Bottleneck (line 19) | class Bottleneck(nn.Module):
    method __init__ (line 22) | def __init__(self, inplanes, planes, stride=1, downsample=None, activa...
    method forward (line 39) | def forward(self, x):
  class BasicBlock (line 62) | class BasicBlock(nn.Module):
    method __init__ (line 65) | def __init__(self, inplanes, planes, stride=1, downsample=None, activa...
    method forward (line 75) | def forward(self, x):

FILE: model/blocks/rmnet_blocks.py
  class RMBlock (line 20) | class RMBlock(nn.Module):
    method __init__ (line 21) | def __init__(self, input_planes, squeeze_planes, output_planes, downsa...
    method init_weights (line 48) | def init_weights(self):
    method forward (line 56) | def forward(self, x):

FILE: model/blocks/se_resnet_blocks.py
  class SEBottleneck (line 19) | class SEBottleneck(nn.Module):
    method __init__ (line 22) | def __init__(self, inplanes, planes, stride=1, downsample=None, activa...
    method forward (line 45) | def forward(self, x):

FILE: model/blocks/se_resnext_blocks.py
  class SEBottleneckX (line 19) | class SEBottleneckX(nn.Module):
    method __init__ (line 22) | def __init__(self, inplanes, planes, cardinality, stride=1, downsample...
    method forward (line 40) | def forward(self, x):

FILE: model/blocks/shared_blocks.py
  function make_activation (line 18) | def make_activation(activation):
  class SELayer (line 26) | class SELayer(nn.Module):
    method __init__ (line 28) | def __init__(self, inplanes, squeeze_ratio=8, activation=nn.PReLU, siz...
    method forward (line 41) | def forward(self, x):
  class ScaleFilter (line 50) | class ScaleFilter(nn.Module):
    method __init__ (line 52) | def __init__(self, q):
    method forward (line 57) | def forward(self, x):

FILE: model/blocks/shufflenet_v2_blocks.py
  function conv_bn (line 19) | def conv_bn(inp, oup, stride, activation=nn.ReLU):
  function conv_1x1_bn (line 29) | def conv_1x1_bn(inp, oup, activation=nn.ReLU):
  function channel_shuffle (line 39) | def channel_shuffle(x, groups):
  class ShuffleInvertedResidual (line 50) | class ShuffleInvertedResidual(nn.Module):
    method __init__ (line 51) | def __init__(self, inp, oup, stride, benchmodel, activation=nn.ReLU):
    method _concat (line 101) | def _concat(x, out):
    method init_weights (line 105) | def init_weights(self):
    method forward (line 113) | def forward(self, x):

FILE: model/common.py
  class ModelInterface (line 17) | class ModelInterface(nn.Module):
    method set_dropout_ratio (line 21) | def set_dropout_ratio(self, ratio):
    method get_input_res (line 25) | def get_input_res(self):

FILE: model/landnet.py
  class LandmarksNet (line 19) | class LandmarksNet(ModelInterface):
    method __init__ (line 21) | def __init__(self):
    method forward (line 63) | def forward(self, x):
    method get_input_res (line 69) | def get_input_res(self):
    method set_dropout_ratio (line 72) | def set_dropout_ratio(self, ratio):

FILE: model/mobilefacenet.py
  function init_block (line 23) | def init_block(in_channels, out_channels, stride, activation=nn.PReLU):
  class MobileFaceNet (line 33) | class MobileFaceNet(ModelInterface):
    method __init__ (line 35) | def __init__(self, embedding_size=128, num_classes=1, width_multiplier...
    method forward (line 95) | def forward(self, x):
    method get_input_res (line 109) | def get_input_res():
    method set_dropout_ratio (line 112) | def set_dropout_ratio(self, ratio):
    method init_weights (line 115) | def init_weights(self):

FILE: model/resnet_angular.py
  class ResNetAngular (line 20) | class ResNetAngular(ModelInterface):
    method __init__ (line 22) | def __init__(self, embedding_size=128, num_classes=0, feature=True):
    method forward (line 33) | def forward(self, x):
    method get_input_res (line 47) | def get_input_res():
    method set_dropout_ratio (line 50) | def set_dropout_ratio(self, ratio):

FILE: model/rmnet_angular.py
  class RMNetAngular (line 22) | class RMNetAngular(ModelInterface):
    method __init__ (line 25) | def __init__(self, embedding_size, num_classes=0, feature=True, body=R...
    method forward (line 34) | def forward(self, x):
    method set_dropout_ratio (line 47) | def set_dropout_ratio(self, ratio):
    method get_input_res (line 55) | def get_input_res():

FILE: model/se_resnet_angular.py
  class SEResNetAngular (line 21) | class SEResNetAngular(ModelInterface):
    method __init__ (line 23) | def __init__(self, embedding_size=128, num_classes=0, feature=True):
    method forward (line 34) | def forward(self, x):
    method get_input_res (line 47) | def get_input_res():
    method set_dropout_ratio (line 50) | def set_dropout_ratio(self, ratio):

FILE: model/shufflenet_v2_angular.py
  class ShuffleNetV2Angular (line 21) | class ShuffleNetV2Angular(ModelInterface):
    method __init__ (line 23) | def __init__(self, embedding_size, num_classes=0, feature=True, body=S...
    method forward (line 36) | def forward(self, x):
    method set_dropout_ratio (line 49) | def set_dropout_ratio(self, ratio):
    method get_input_res (line 53) | def get_input_res():

FILE: scripts/accuracy_check.py
  function main (line 26) | def main():

FILE: scripts/align_images.py
  class LandnetPT (line 29) | class LandnetPT:
    method __init__ (line 31) | def __init__(self, model):
    method get_landmarks (line 36) | def get_landmarks(self, batch):
  class FaceDetector (line 45) | class FaceDetector:
    method __init__ (line 47) | def __init__(self, proto, model, conf=.6, expand_ratio=(1.1, 1.05), si...
    method __decode_detections (line 59) | def __decode_detections(self, out, frame_shape):
    method get_detections (line 90) | def get_detections(self, frame):
  function draw_detections (line 99) | def draw_detections(frame, detections, landmarks):
  function run_dumping (line 111) | def run_dumping(images_list, face_det, landmarks_regressor, vis_flag):
  function create_images_list (line 139) | def create_images_list(images_root, imgs_list):
  function save_data (line 164) | def save_data(data, filename, root_dir):
  function main (line 178) | def main():

FILE: scripts/count_flops.py
  function main (line 21) | def main():

FILE: scripts/matio.py
  function write_mat (line 12) | def write_mat(f, m):
  function read_mat (line 24) | def read_mat(f):
  function read_mkl_vec (line 32) | def read_mkl_vec(f):
  function load_mkl_vec (line 43) | def load_mkl_vec(filename):
  function load_mat (line 49) | def load_mat(filename):
  function save_mat (line 55) | def save_mat(filename, m):
  function main (line 59) | def main():

FILE: scripts/plot_roc_curves_lfw.py
  function main (line 19) | def main():

FILE: scripts/pytorch2onnx.py
  function main (line 20) | def main():

FILE: tests/test_alignment.py
  class FaceAlignmentTests (line 22) | class FaceAlignmentTests(unittest.TestCase):
    method test_align_image (line 24) | def test_align_image(self):

FILE: tests/test_models.py
  class BackbonesTests (line 22) | class BackbonesTests(unittest.TestCase):
    method test_output_shape (line 24) | def test_output_shape(self):
    method test_save_load_snap (line 33) | def test_save_load_snap(self):
  class LandnetTests (line 51) | class LandnetTests(unittest.TestCase):
    method test_output_shape (line 53) | def test_output_shape(self):
    method test_save_load_snap (line 60) | def test_save_load_snap(self):

FILE: tests/test_utils.py
  class UtilsTests (line 20) | class UtilsTests(unittest.TestCase):
    method test_parameters_counter (line 22) | def test_parameters_counter(self):

FILE: train.py
  function train (line 39) | def train(args):
  function main (line 170) | def main():

FILE: train_landmarks.py
  function train (line 36) | def train(args):
  function main (line 120) | def main():

FILE: utils/augmentation.py
  class HorizontalFlipNumpy (line 25) | class HorizontalFlipNumpy:
    method __init__ (line 27) | def __init__(self, p=.5):
    method __call__ (line 31) | def __call__(self, img):
  class ShowTransform (line 37) | class ShowTransform:
    method __call__ (line 39) | def __call__(self, sample):
  class NumpyToTensor (line 46) | class NumpyToTensor:
    method __init__ (line 48) | def __init__(self, switch_rb=False):
    method __call__ (line 51) | def __call__(self, image):
  class RandomShiftNumpy (line 61) | class RandomShiftNumpy:
    method __init__ (line 63) | def __init__(self, max_rel_shift, p=.5):
    method __call__ (line 67) | def __call__(self, image):
  class RandomRotationNumpy (line 79) | class RandomRotationNumpy:
    method __init__ (line 81) | def __init__(self, max_angle, p=.5):
    method __call__ (line 85) | def __call__(self, image):
  class ResizeNumpy (line 95) | class ResizeNumpy:
    method __init__ (line 97) | def __init__(self, output_size):
    method __call__ (line 101) | def __call__(self, image):
  class CenterCropNumpy (line 116) | class CenterCropNumpy:
    method __init__ (line 118) | def __init__(self, output_size):
    method __call__ (line 122) | def __call__(self, image):
  class BlurNumpy (line 135) | class BlurNumpy:
    method __init__ (line 137) | def __init__(self, p, k):
    method __call__ (line 142) | def __call__(self, img):
  class CutOutWithPrior (line 148) | class CutOutWithPrior:
    method __init__ (line 150) | def __init__(self, p, max_area):
    method __call__ (line 155) | def __call__(self, img):

FILE: utils/face_align.py
  class FivePointsAligner (line 18) | class FivePointsAligner():
    method align (line 26) | def align(img, landmarks, d_size=(400, 400), normalized=False, show=Fa...
  function transformation_from_points (line 58) | def transformation_from_points(points1, points2):

FILE: utils/ie_tools.py
  class IEModel (line 21) | class IEModel:
    method __init__ (line 23) | def __init__(self, exec_net, inputs_info, input_key, output_key):
    method forward (line 29) | def forward(self, img):
    method get_input_shape (line 34) | def get_input_shape(self):
  function load_ie_model (line 39) | def load_ie_model(model_xml, device, plugin_dir, cpu_extension=''):

FILE: utils/landmarks_augmentation.py
  class Rescale (line 19) | class Rescale:
    method __init__ (line 21) | def __init__(self, output_size):
    method __call__ (line 25) | def __call__(self, sample):
  class RandomCrop (line 41) | class RandomCrop:
    method __init__ (line 43) | def __init__(self, output_size):
    method __call__ (line 51) | def __call__(self, sample):
  class HorizontalFlip (line 71) | class HorizontalFlip:
    method __init__ (line 73) | def __init__(self, p=.5):
    method __call__ (line 76) | def __call__(self, sample):
  class Blur (line 94) | class Blur:
    method __init__ (line 96) | def __init__(self, p, k):
    method __call__ (line 101) | def __call__(self, sample):
  class Show (line 110) | class Show:
    method __call__ (line 112) | def __call__(self, sample):
  class RandomRotate (line 122) | class RandomRotate:
    method __init__ (line 127) | def __init__(self, max_angle, p=.5):
    method __call__ (line 131) | def __call__(self, sample):
  class ToTensor (line 145) | class ToTensor:
    method __init__ (line 147) | def __init__(self, switch_rb=False):
    method __call__ (line 150) | def __call__(self, sample):
  class RandomScale (line 162) | class RandomScale:
    method __init__ (line 164) | def __init__(self, max_scale, min_scale, p=.5):
    method __call__ (line 169) | def __call__(self, sample):

FILE: utils/parser_yaml.py
  class ArgumentParserWithYaml (line 17) | class ArgumentParserWithYaml(ArgumentParser):
    method _check_arg_line_repr_None (line 22) | def _check_arg_line_repr_None(arg_line, k, v):
    method convert_arg_line_to_args (line 42) | def convert_arg_line_to_args(self, arg_line):

FILE: utils/utils.py
  function save_model_cpu (line 20) | def save_model_cpu(net, optim, ckpt_fname, epoch, write_solverstate=False):
  function get_model_parameters_number (line 35) | def get_model_parameters_number(model, as_string=True):
  function load_model_state (line 50) | def load_model_state(model, snap, device_id, eval_state=True):
  function flip_tensor (line 81) | def flip_tensor(x, dim):