SYMBOL INDEX (113 symbols across 21 files)

FILE: callbacks.py
  class EpochBasedLearningRateSchedule (line 10) | class EpochBasedLearningRateSchedule(keras.callbacks.Callback):
    method __init__ (line 13) | def __init__(self, schedule):
    method on_epoch_begin (line 22) | def on_epoch_begin(self, epoch, logs=None):
  class LogImages (line 48) | class LogImages(keras.callbacks.Callback):
    method __init__ (line 49) | def __init__(self, logdir, sample_image):
    method on_epoch_end (line 54) | def on_epoch_end(self, epoch, logs={}):

FILE: dataset.py
  function data_generator (line 11) | def data_generator(data_dir, name, image_size, number_marks, training):
  class WFLWSequence (line 63) | class WFLWSequence(tf.keras.utils.Sequence):
    method __init__ (line 70) | def __init__(self, data_dir, name, training, batch_size):
    method __len__ (line 85) | def __len__(self):
    method __getitem__ (line 88) | def __getitem__(self, index):
  function build_dataset (line 131) | def build_dataset(data_dir,
  function top_k_indices (line 177) | def top_k_indices(x, k):
  function get_peak_location (line 186) | def get_peak_location(heatmap, image_size=(256, 256)):
  function _parse_heatmaps (line 195) | def _parse_heatmaps(img, heatmaps):

FILE: evaluate.py
  function compute_nme (line 14) | def compute_nme(prediction, ground_truth):
  function evaluate (line 24) | def evaluate(dataset: fmd.mark_dataset.dataset, model):
  function make_dataset (line 96) | def make_dataset():

FILE: fmd/aflw2000_3d.py
  class AFLW2000_3D (line 9) | class AFLW2000_3D(MarkDataset):
    method populate_dataset (line 12) | def populate_dataset(self, image_dir):
    method get_marks_from_file (line 44) | def get_marks_from_file(self, mark_file):

FILE: fmd/afw.py
  class AFW (line 13) | class AFW(MarkDataset):
    method populate_dataset (line 16) | def populate_dataset(self, image_dir):
    method get_marks_from_file (line 48) | def get_marks_from_file(self, mark_file):

FILE: fmd/ds300vw.py
  class DS300VW (line 13) | class DS300VW(MarkDataset):
    method populate_dataset (line 16) | def populate_dataset(self, image_dir):
    method get_marks_from_file (line 48) | def get_marks_from_file(self, mark_file):

FILE: fmd/ds300w.py
  class DS300W (line 13) | class DS300W(MarkDataset):
    method populate_dataset (line 16) | def populate_dataset(self, image_dir):
    method get_marks_from_file (line 48) | def get_marks_from_file(self, mark_file):

FILE: fmd/ds300w_lp.py
  class DS300W_LP (line 9) | class DS300W_LP(MarkDataset):
    method populate_dataset (line 12) | def populate_dataset(self, image_dir):
    method get_marks_from_file (line 44) | def get_marks_from_file(self, mark_file):

FILE: fmd/helen.py
  class HELEN (line 13) | class HELEN(MarkDataset):
    method populate_dataset (line 16) | def populate_dataset(self, image_dir):
    method get_marks_from_file (line 48) | def get_marks_from_file(self, mark_file):

FILE: fmd/ibug.py
  class IBUG (line 13) | class IBUG(MarkDataset):
    method populate_dataset (line 16) | def populate_dataset(self, image_dir):
    method get_marks_from_file (line 48) | def get_marks_from_file(self, mark_file):

FILE: fmd/lfpw.py
  class LFPW (line 13) | class LFPW(MarkDataset):
    method populate_dataset (line 16) | def populate_dataset(self, image_dir):
    method get_marks_from_file (line 48) | def get_marks_from_file(self, mark_file):

FILE: fmd/mark_dataset/data_pair.py
  class DataPair (line 8) | class DataPair(object):
    method __init__ (line 11) | def __init__(self, image_file, marks, key_marks_indices):
    method read_image (line 29) | def read_image(self, format="BGR"):
    method get_marks (line 44) | def get_marks(self):
    method get_key_marks (line 55) | def get_key_marks(self):
    method save_mark_to_json (line 70) | def save_mark_to_json(self, file_name):

FILE: fmd/mark_dataset/dataset.py
  class MarkDataset (line 6) | class MarkDataset(ABC):
    method __init__ (line 8) | def __init__(self, dataset_name):
    method __str__ (line 20) | def __str__(self):
    method __len__ (line 26) | def __len__(self):
    method __iter__ (line 30) | def __iter__(self):
    method __next__ (line 33) | def __next__(self):
    method _make_datapair (line 40) | def _make_datapair(self, data_index):
    method populate_dataset (line 49) | def populate_dataset(self):
    method get_marks_from_file (line 72) | def get_marks_from_file(self, mark_file):
    method pick_one (line 77) | def pick_one(self):
    method export (line 84) | def export(self, export_dir):

FILE: fmd/mark_dataset/util.py
  class FileListGenerator (line 9) | class FileListGenerator:
    method __init__ (line 12) | def __init__(self):
    method generate_list (line 17) | def generate_list(self, target_dir, format_list=["jpg", "png"]):
    method save_file_path_list (line 41) | def save_file_path_list(self, list_name='list.csv'):
    method save_basename_list (line 58) | def save_basename_list(self, list_name='basename.csv'):
  function draw_marks (line 76) | def draw_marks(image, marks, mark_size=3, color=(0, 255, 0), line_width=...

FILE: fmd/universal.py
  class Universal (line 20) | class Universal(MarkDataset):
    method populate_dataset (line 23) | def populate_dataset(self, image_dir, key_marks_indices):
    method get_marks_from_file (line 55) | def get_marks_from_file(self, mark_file):

FILE: fmd/wflw.py
  class WFLW (line 15) | class WFLW(MarkDataset):
    method __init__ (line 23) | def __init__(self, is_train, name):
    method populate_dataset (line 33) | def populate_dataset(self, image_dir):
    method get_marks_from_file (line 90) | def get_marks_from_file(self, mark_file):

FILE: mark_operator.py
  class MarkOperator (line 21) | class MarkOperator(object):
    method __init__ (line 24) | def __init__(self):
    method get_distance (line 27) | def get_distance(self, mark1, mark2):
    method get_angle (line 31) | def get_angle(self, vector1, vector2, in_radian=False):
    method pad_to_3d (line 48) | def pad_to_3d(self, marks_2d, pad_value=-1):
    method get_center (line 55) | def get_center(self, marks):
    method get_height_width_depth (line 61) | def get_height_width_depth(self, marks):
    method rotate (line 67) | def rotate(self, marks, radian, center=(0, 0)):
    method flip_lr (line 78) | def flip_lr(self, marks, width):
    method _generate_heatmap (line 113) | def _generate_heatmap(self, heatmap_size, center_point, sigma):
    method generate_heatmaps (line 162) | def generate_heatmaps(self, norm_marks, map_size=(64, 64), sigma=3):

FILE: network.py
  function hrnet_stem (line 9) | def hrnet_stem(filters=64):
  function hrnet_heads (line 24) | def hrnet_heads(input_channels=64, output_channels=17):
  class HRNetStem (line 46) | class HRNetStem(layers.Layer, tfmot.sparsity.keras.PrunableLayer):
    method __init__ (line 48) | def __init__(self, filters=64, **kwargs):
    method build (line 53) | def build(self, input_shape):
    method call (line 63) | def call(self, inputs):
    method get_config (line 72) | def get_config(self):
    method get_prunable_weights (line 78) | def get_prunable_weights(self):
  class HRNetHeads (line 85) | class HRNetHeads(layers.Layer):
    method __init__ (line 87) | def __init__(self, input_channels=64, output_channels=17, **kwargs):
    method build (line 93) | def build(self, input_shape):
    method call (line 107) | def call(self, inputs):
    method get_config (line 117) | def get_config(self):
    method get_prunable_weights (line 124) | def get_prunable_weights(self):
  function hrnet_v2 (line 131) | def hrnet_v2(input_shape, output_channels, width=18, name="hrnetv2"):

FILE: postprocessing.py
  function top_k_indices (line 6) | def top_k_indices(x, k):
  function get_peak_location (line 16) | def get_peak_location(heatmap, image_size=(256, 256)):
  function parse_heatmaps (line 26) | def parse_heatmaps(heatmaps, image_size):
  function draw_marks (line 42) | def draw_marks(image, marks):

FILE: preprocessing.py
  function crop_face (line 10) | def crop_face(image, marks, scale=1.8, shift_ratios=(0, 0)):
  function normalize (line 64) | def normalize(inputs):
  function rotate_randomly (line 81) | def rotate_randomly(image, marks, degrees=(-30, 30)):
  function scale_randomly (line 105) | def scale_randomly(image, marks, output_size=(256, 256), scale_range=(0,...
  function flip_randomly (line 147) | def flip_randomly(image, marks, probability=0.5):
  function generate_heatmaps (line 164) | def generate_heatmaps(marks, img_size, map_size):

FILE: quantization.py
  function representative_dataset_gen (line 17) | def representative_dataset_gen():
  function quantize (line 40) | def quantize(saved_model, mode=None, representative_dataset=None):
  class TFLiteModelPredictor (line 83) | class TFLiteModelPredictor(object):
    method __init__ (line 86) | def __init__(self, model_path):
    method predict (line 92) | def predict(self, inputs):