SYMBOL INDEX (171 symbols across 19 files)

FILE: animate.py
  function normalize_kp (line 16) | def normalize_kp(kp_source, kp_driving, kp_driving_initial, adapt_moveme...
  function animate (line 39) | def animate(config, generator, kp_detector, checkpoint, log_dir, dataset):

FILE: augmentation.py
  function crop_clip (line 20) | def crop_clip(clip, min_h, min_w, h, w):
  function pad_clip (line 34) | def pad_clip(clip, h, w):
  function resize_clip (line 42) | def resize_clip(clip, size, interpolation='bilinear'):
  function get_resize_sizes (line 81) | def get_resize_sizes(im_h, im_w, size):
  class RandomFlip (line 91) | class RandomFlip(object):
    method __init__ (line 92) | def __init__(self, time_flip=False, horizontal_flip=False):
    method __call__ (line 96) | def __call__(self, clip):
  class RandomResize (line 105) | class RandomResize(object):
    method __init__ (line 115) | def __init__(self, ratio=(3. / 4., 4. / 3.), interpolation='nearest'):
    method __call__ (line 119) | def __call__(self, clip):
  class RandomCrop (line 136) | class RandomCrop(object):
    method __init__ (line 143) | def __init__(self, size):
    method __call__ (line 149) | def __call__(self, clip):
  class RandomRotation (line 175) | class RandomRotation(object):
    method __init__ (line 184) | def __init__(self, degrees):
    method __call__ (line 197) | def __call__(self, clip):
  class ColorJitter (line 217) | class ColorJitter(object):
    method __init__ (line 230) | def __init__(self, brightness=0, contrast=0, saturation=0, hue=0):
    method get_params (line 236) | def get_params(self, brightness, contrast, saturation, hue):
    method __call__ (line 261) | def __call__(self, clip):
  class AllAugmentationTransform (line 323) | class AllAugmentationTransform:
    method __init__ (line 324) | def __init__(self, resize_param=None, rotation_param=None, flip_param=...
    method __call__ (line 342) | def __call__(self, clip):

FILE: crop-video.py
  function extract_bbox (line 14) | def extract_bbox(frame, fa):
  function bb_intersection_over_union (line 29) | def bb_intersection_over_union(boxA, boxB):
  function join (line 41) | def join(tube_bbox, bbox):
  function compute_bbox (line 49) | def compute_bbox(start, end, fps, tube_bbox, frame_shape, inp, image_sha...
  function compute_bbox_trajectories (line 75) | def compute_bbox_trajectories(trajectories, fps, frame_shape, args):
  function process_video (line 84) | def process_video(args):

FILE: data/taichi-loading/load_videos.py
  function save (line 20) | def save(path, frames, format):
  function download (line 36) | def download(video_id, args):
  function run (line 45) | def run(data):

FILE: demo.py
  function load_checkpoints (line 25) | def load_checkpoints(config_path, checkpoint_path, cpu=False):
  function make_animation (line 58) | def make_animation(source_image, driving_video, generator, kp_detector, ...
  function find_best_frame (line 81) | def find_best_frame(source, driving, cpu=False):

FILE: frames_dataset.py
  function read_video (line 14) | def read_video(name, frame_shape):
  class FramesDataset (line 55) | class FramesDataset(Dataset):
    method __init__ (line 63) | def __init__(self, root_dir, frame_shape=(256, 256, 3), id_sampling=Fa...
    method __len__ (line 97) | def __len__(self):
    method __getitem__ (line 100) | def __getitem__(self, idx):
  class DatasetRepeater (line 141) | class DatasetRepeater(Dataset):
    method __init__ (line 146) | def __init__(self, dataset, num_repeats=100):
    method __len__ (line 150) | def __len__(self):
    method __getitem__ (line 153) | def __getitem__(self, idx):
  class PairedDataset (line 157) | class PairedDataset(Dataset):
    method __init__ (line 162) | def __init__(self, initial_dataset, number_of_pairs, seed=0):
    method __len__ (line 187) | def __len__(self):
    method __getitem__ (line 190) | def __getitem__(self, idx):

FILE: logger.py
  class Logger (line 13) | class Logger:
    method __init__ (line 14) | def __init__(self, log_dir, checkpoint_freq=100, visualizer_params=Non...
    method log_scores (line 29) | def log_scores(self, loss_names):
    method visualize_rec (line 39) | def visualize_rec(self, inp, out):
    method save_cpk (line 43) | def save_cpk(self, emergent=False):
    method load_cpk (line 51) | def load_cpk(checkpoint_path, generator=None, discriminator=None, kp_d...
    method __enter__ (line 79) | def __enter__(self):
    method __exit__ (line 82) | def __exit__(self, exc_type, exc_val, exc_tb):
    method log_iter (line 87) | def log_iter(self, losses):
    method log_epoch (line 93) | def log_epoch(self, epoch, models, inp, out):
  class Visualizer (line 102) | class Visualizer:
    method __init__ (line 103) | def __init__(self, kp_size=5, draw_border=False, colormap='gist_rainbo...
    method draw_image_with_kp (line 108) | def draw_image_with_kp(self, image, kp_array):
    method create_image_column_with_kp (line 118) | def create_image_column_with_kp(self, images, kp):
    method create_image_column (line 122) | def create_image_column(self, images):
    method create_image_grid (line 128) | def create_image_grid(self, *args):
    method visualize (line 137) | def visualize(self, driving, source, out):

FILE: modules/dense_motion.py
  class DenseMotionNetwork (line 7) | class DenseMotionNetwork(nn.Module):
    method __init__ (line 12) | def __init__(self, block_expansion, num_blocks, max_features, num_kp, ...
    method create_heatmap_representations (line 32) | def create_heatmap_representations(self, source_image, kp_driving, kp_...
    method create_sparse_motions (line 47) | def create_sparse_motions(self, source_image, kp_driving, kp_source):
    method create_deformed_source_image (line 69) | def create_deformed_source_image(self, source_image, sparse_motions):
    method forward (line 81) | def forward(self, source_image, kp_driving, kp_source):

FILE: modules/discriminator.py
  class DownBlock2d (line 7) | class DownBlock2d(nn.Module):
    method __init__ (line 12) | def __init__(self, in_features, out_features, norm=False, kernel_size=...
    method forward (line 25) | def forward(self, x):
  class Discriminator (line 36) | class Discriminator(nn.Module):
    method __init__ (line 41) | def __init__(self, num_channels=3, block_expansion=64, num_blocks=4, m...
    method forward (line 59) | def forward(self, x, kp=None):
  class MultiScaleDiscriminator (line 74) | class MultiScaleDiscriminator(nn.Module):
    method __init__ (line 79) | def __init__(self, scales=(), **kwargs):
    method forward (line 87) | def forward(self, x, kp=None):

FILE: modules/generator.py
  class OcclusionAwareGenerator (line 8) | class OcclusionAwareGenerator(nn.Module):
    method __init__ (line 14) | def __init__(self, num_channels, num_kp, block_expansion, max_features...
    method deform_input (line 50) | def deform_input(self, inp, deformation):
    method forward (line 59) | def forward(self, source_image, kp_driving, kp_source):

FILE: modules/keypoint_detector.py
  class KPDetector (line 7) | class KPDetector(nn.Module):
    method __init__ (line 12) | def __init__(self, block_expansion, num_kp, num_channels, max_features,
    method gaussian2kp (line 37) | def gaussian2kp(self, heatmap):
    method forward (line 49) | def forward(self, x):

FILE: modules/model.py
  class Vgg19 (line 10) | class Vgg19(torch.nn.Module):
    method __init__ (line 14) | def __init__(self, requires_grad=False):
    method forward (line 42) | def forward(self, X):
  class ImagePyramide (line 53) | class ImagePyramide(torch.nn.Module):
    method __init__ (line 57) | def __init__(self, scales, num_channels):
    method forward (line 64) | def forward(self, x):
  class Transform (line 71) | class Transform:
    method __init__ (line 75) | def __init__(self, bs, **kwargs):
    method transform_frame (line 89) | def transform_frame(self, frame):
    method warp_coordinates (line 95) | def warp_coordinates(self, coordinates):
    method jacobian (line 115) | def jacobian(self, coordinates):
  function detach_kp (line 123) | def detach_kp(kp):
  class GeneratorFullModel (line 127) | class GeneratorFullModel(torch.nn.Module):
    method __init__ (line 132) | def __init__(self, kp_extractor, generator, discriminator, train_params):
    method forward (line 151) | def forward(self, x):
  class DiscriminatorFullModel (line 225) | class DiscriminatorFullModel(torch.nn.Module):
    method __init__ (line 230) | def __init__(self, kp_extractor, generator, discriminator, train_params):
    method forward (line 243) | def forward(self, x, generated):

FILE: modules/util.py
  function kp2gaussian (line 9) | def kp2gaussian(kp, spatial_size, kp_variance):
  function make_coordinate_grid (line 33) | def make_coordinate_grid(spatial_size, type):
  class ResBlock2d (line 52) | class ResBlock2d(nn.Module):
    method __init__ (line 57) | def __init__(self, in_features, kernel_size, padding):
    method forward (line 66) | def forward(self, x):
  class UpBlock2d (line 77) | class UpBlock2d(nn.Module):
    method __init__ (line 82) | def __init__(self, in_features, out_features, kernel_size=3, padding=1...
    method forward (line 89) | def forward(self, x):
  class DownBlock2d (line 97) | class DownBlock2d(nn.Module):
    method __init__ (line 102) | def __init__(self, in_features, out_features, kernel_size=3, padding=1...
    method forward (line 109) | def forward(self, x):
  class SameBlock2d (line 117) | class SameBlock2d(nn.Module):
    method __init__ (line 122) | def __init__(self, in_features, out_features, groups=1, kernel_size=3,...
    method forward (line 128) | def forward(self, x):
  class Encoder (line 135) | class Encoder(nn.Module):
    method __init__ (line 140) | def __init__(self, block_expansion, in_features, num_blocks=3, max_fea...
    method forward (line 150) | def forward(self, x):
  class Decoder (line 157) | class Decoder(nn.Module):
    method __init__ (line 162) | def __init__(self, block_expansion, in_features, num_blocks=3, max_fea...
    method forward (line 175) | def forward(self, x):
  class Hourglass (line 184) | class Hourglass(nn.Module):
    method __init__ (line 189) | def __init__(self, block_expansion, in_features, num_blocks=3, max_fea...
    method forward (line 195) | def forward(self, x):
  class AntiAliasInterpolation2d (line 199) | class AntiAliasInterpolation2d(nn.Module):
    method __init__ (line 203) | def __init__(self, channels, scale):
    method forward (line 237) | def forward(self, input):

FILE: reconstruction.py
  function reconstruction (line 11) | def reconstruction(config, generator, kp_detector, checkpoint, log_dir, ...

FILE: sync_batchnorm/batchnorm.py
  function _sum_ft (line 24) | def _sum_ft(tensor):
  function _unsqueeze_ft (line 29) | def _unsqueeze_ft(tensor):
  class _SynchronizedBatchNorm (line 38) | class _SynchronizedBatchNorm(_BatchNorm):
    method __init__ (line 39) | def __init__(self, num_features, eps=1e-5, momentum=0.1, affine=True):
    method forward (line 48) | def forward(self, input):
    method __data_parallel_replicate__ (line 80) | def __data_parallel_replicate__(self, ctx, copy_id):
    method _data_parallel_master (line 90) | def _data_parallel_master(self, intermediates):
    method _compute_mean_std (line 113) | def _compute_mean_std(self, sum_, ssum, size):
  class SynchronizedBatchNorm1d (line 128) | class SynchronizedBatchNorm1d(_SynchronizedBatchNorm):
    method _check_input_dim (line 184) | def _check_input_dim(self, input):
  class SynchronizedBatchNorm2d (line 191) | class SynchronizedBatchNorm2d(_SynchronizedBatchNorm):
    method _check_input_dim (line 247) | def _check_input_dim(self, input):
  class SynchronizedBatchNorm3d (line 254) | class SynchronizedBatchNorm3d(_SynchronizedBatchNorm):
    method _check_input_dim (line 311) | def _check_input_dim(self, input):

FILE: sync_batchnorm/comm.py
  class FutureResult (line 18) | class FutureResult(object):
    method __init__ (line 21) | def __init__(self):
    method put (line 26) | def put(self, result):
    method get (line 32) | def get(self):
  class SlavePipe (line 46) | class SlavePipe(_SlavePipeBase):
    method run_slave (line 49) | def run_slave(self, msg):
  class SyncMaster (line 56) | class SyncMaster(object):
    method __init__ (line 67) | def __init__(self, master_callback):
    method __getstate__ (line 78) | def __getstate__(self):
    method __setstate__ (line 81) | def __setstate__(self, state):
    method register_slave (line 84) | def register_slave(self, identifier):
    method run_master (line 102) | def run_master(self, master_msg):
    method nr_slaves (line 136) | def nr_slaves(self):

FILE: sync_batchnorm/replicate.py
  class CallbackContext (line 23) | class CallbackContext(object):
  function execute_replication_callbacks (line 27) | def execute_replication_callbacks(modules):
  class DataParallelWithCallback (line 50) | class DataParallelWithCallback(DataParallel):
    method replicate (line 64) | def replicate(self, module, device_ids):
  function patch_replication_callback (line 70) | def patch_replication_callback(data_parallel):

FILE: sync_batchnorm/unittest.py
  function as_numpy (line 17) | def as_numpy(v):
  class TorchTestCase (line 23) | class TorchTestCase(unittest.TestCase):
    method assertTensorClose (line 24) | def assertTensorClose(self, a, b, atol=1e-3, rtol=1e-3):

FILE: train.py
  function train (line 16) | def train(config, generator, discriminator, kp_detector, checkpoint, log...