SYMBOL INDEX (94 symbols across 5 files)

FILE: core/spectral_norm.py
  class SpectralNorm (line 8) | class SpectralNorm(object):
    method __init__ (line 19) | def __init__(self, name='weight', n_power_iterations=1, dim=0, eps=1e-...
    method reshape_weight_to_matrix (line 28) | def reshape_weight_to_matrix(self, weight):
    method compute_weight (line 37) | def compute_weight(self, module, do_power_iteration):
    method remove (line 89) | def remove(self, module):
    method __call__ (line 98) | def __call__(self, module, inputs):
    method _solve_v_and_rescale (line 101) | def _solve_v_and_rescale(self, weight_mat, u, target_sigma):
    method apply (line 109) | def apply(module, name, n_power_iterations, dim, eps):
  class SpectralNormLoadStateDictPreHook (line 146) | class SpectralNormLoadStateDictPreHook(object):
    method __init__ (line 148) | def __init__(self, fn):
    method __call__ (line 159) | def __call__(self, state_dict, prefix, local_metadata, strict,
  class SpectralNormStateDictHook (line 176) | class SpectralNormStateDictHook(object):
    method __init__ (line 178) | def __init__(self, fn):
    method __call__ (line 181) | def __call__(self, module, state_dict, prefix, local_metadata):
  function spectral_norm (line 190) | def spectral_norm(module, name='weight', n_power_iterations=1, eps=1e-12...
  function remove_spectral_norm (line 243) | def remove_spectral_norm(module, name='weight'):
  function use_spectral_norm (line 264) | def use_spectral_norm(module, use_sn=False):

FILE: core/utils.py
  class ZipReader (line 32) | class ZipReader(object):
    method __init__ (line 35) | def __init__(self):
    method build_file_dict (line 39) | def build_file_dict(path):
    method imread (line 49) | def imread(path, idx):
  class GroupRandomHorizontalFlip (line 61) | class GroupRandomHorizontalFlip(object):
    method __init__ (line 65) | def __init__(self, is_flow=False):
    method __call__ (line 68) | def __call__(self, img_group, is_flow=False):
  class Stack (line 81) | class Stack(object):
    method __init__ (line 82) | def __init__(self, roll=False):
    method __call__ (line 85) | def __call__(self, img_group):
  class ToTorchFormatTensor (line 107) | class ToTorchFormatTensor(object):
    method __init__ (line 111) | def __init__(self, div=True):
    method __call__ (line 114) | def __call__(self, pic):
  function create_random_shape_with_random_motion (line 133) | def create_random_shape_with_random_motion(video_length, imageHeight=240...
  function get_random_shape (line 163) | def get_random_shape(edge_num=9, ratio=0.7, width=432, height=240):
  function random_accelerate (line 204) | def random_accelerate(velocity, maxAcceleration, dist='uniform'):
  function get_random_velocity (line 219) | def get_random_velocity(max_speed=3, dist='uniform'):
  function random_move_control_points (line 231) | def random_move_control_points(X, Y, imageHeight, imageWidth, lineVeloci...

FILE: demo.py
  function read_frame_info_from_video (line 40) | def read_frame_info_from_video(vname):
  function read_mask (line 52) | def read_mask(path):
  function get_ref_index (line 59) | def get_ref_index(neighbor_ids, length):
  function pre_process (line 66) | def pre_process(task):
  function process (line 93) | def process(frames, model, device, w, h):
  function get_inpaint_mode_for_detext (line 127) | def get_inpaint_mode_for_detext(H, h, mask):  # get inpaint segment
  function main (line 148) | def main():  # detext

FILE: model/auto-sttn.py
  class BaseNetwork (line 11) | class BaseNetwork(nn.Module):
    method __init__ (line 12) | def __init__(self):
    method print_network (line 15) | def print_network(self):
    method init_weights (line 24) | def init_weights(self, init_type='normal', gain=0.02):
  class InpaintGenerator (line 64) | class InpaintGenerator(BaseNetwork):
    method __init__ (line 65) | def __init__(self, init_weights=True):
    method forward (line 100) | def forward(self, masked_frames):
    method infer (line 111) | def infer(self, feat):
  class deconv (line 118) | class deconv(nn.Module):
    method __init__ (line 119) | def __init__(self, input_channel, output_channel, kernel_size=3, paddi...
    method forward (line 124) | def forward(self, x):
  class Attention (line 135) | class Attention(nn.Module):
    method forward (line 140) | def forward(self, query, key, value):
  class MultiHeadedAttention (line 148) | class MultiHeadedAttention(nn.Module):
    method __init__ (line 153) | def __init__(self, patchsize, d_model):
    method forward (line 167) | def forward(self, x, b, c):
  class FeedForward (line 210) | class FeedForward(nn.Module):
    method __init__ (line 211) | def __init__(self, d_model):
    method forward (line 220) | def forward(self, x):
  class TransformerBlock (line 225) | class TransformerBlock(nn.Module):
    method __init__ (line 230) | def __init__(self, patchsize, hidden=128):
    method forward (line 235) | def forward(self, x):
  class Discriminator (line 246) | class Discriminator(BaseNetwork):
    method __init__ (line 247) | def __init__(self, in_channels=3, use_sigmoid=False, use_spectral_norm...
    method forward (line 280) | def forward(self, xs):
  function spectral_norm (line 291) | def spectral_norm(module, mode=True):

FILE: model/vis.py
  class BaseNetwork (line 12) | class BaseNetwork(nn.Module):
    method __init__ (line 13) | def __init__(self):
    method print_network (line 16) | def print_network(self):
    method init_weights (line 25) | def init_weights(self, init_type='normal', gain=0.02):
  class InpaintGenerator (line 65) | class InpaintGenerator(BaseNetwork):
    method __init__ (line 66) | def __init__(self, init_weights=True):  # 1046
    method forward (line 101) | def forward(self, masked_frames, masks):
    method infer (line 114) | def infer(self, feat, masks):
  class deconv (line 126) | class deconv(nn.Module):
    method __init__ (line 127) | def __init__(self, input_channel, output_channel, kernel_size=3, paddi...
    method forward (line 132) | def forward(self, x):
  class Attention (line 142) | class Attention(nn.Module):
    method forward (line 147) | def forward(self, query, key, value, m):
  class MultiHeadedAttention (line 156) | class MultiHeadedAttention(nn.Module):
    method __init__ (line 161) | def __init__(self, patchsize, d_model):
    method forward (line 175) | def forward(self, x, m, b, c):
  class FeedForward (line 229) | class FeedForward(nn.Module):
    method __init__ (line 230) | def __init__(self, d_model):
    method forward (line 239) | def forward(self, x):
  class TransformerBlock (line 244) | class TransformerBlock(nn.Module):
    method __init__ (line 249) | def __init__(self, patchsize, hidden=128):
    method forward (line 254) | def forward(self, x):
  class Discriminator (line 266) | class Discriminator(BaseNetwork):
    method __init__ (line 267) | def __init__(self, in_channels=3, use_sigmoid=False, use_spectral_norm...
    method forward (line 300) | def forward(self, xs):
  function spectral_norm (line 311) | def spectral_norm(module, mode=True):