SYMBOL INDEX (158 symbols across 20 files) FILE: data/kpuv_dataset.py class KpuvDataset (line 26) | class KpuvDataset(BaseDataset): method modify_commandline_options (line 32) | def modify_commandline_options(parser, is_train): method __init__ (line 36) | def __init__(self, opt): method __getitem__ (line 67) | def __getitem__(self, index): method __len__ (line 89) | def __len__(self): method crop_kps (line 93) | def crop_kps(self, kps, crop_size=256, inner_crop_size=192): method render_kps (line 124) | def render_kps(self, keypoints, draw, thresh=1., min_weight=0.25): FILE: data/layered_video_dataset.py class LayeredVideoDataset (line 27) | class LayeredVideoDataset(BaseDataset): method modify_commandline_options (line 34) | def modify_commandline_options(parser, is_train): method __init__ (line 42) | def __init__(self, opt): method __getitem__ (line 75) | def __getitem__(self, index): method __len__ (line 124) | def __len__(self): method get_params (line 128) | def get_params(self, do_jitter=False, jitter_rate=0.75): method apply_transform (line 146) | def apply_transform(self, data, params, interp_mode='bilinear'): method init_homographies (line 155) | def init_homographies(self, homography_path, n_images): method load_and_process_image (line 171) | def load_and_process_image(self, im_path): method load_and_process_iuv (line 178) | def load_and_process_iuv(self, iuv_path, i): method iuv2input (line 185) | def iuv2input(self, iuv, index): method get_background_inputs (line 234) | def get_background_inputs(self, index, w, h): method get_background_uv (line 246) | def get_background_uv(self, index, w, h): method transform2h (line 268) | def transform2h(self, x, y, m): method mask2trimap (line 275) | def mask2trimap(self, mask): FILE: datasets/iuv_crop2full.py function place_crop (line 23) | def place_crop(crop, image, center_x, center_y): function crop2full (line 48) | def crop2full(keypoints_path, metadata_path, uvdir, outdir): FILE: models/kp2uv_model.py class Kp2uvModel (line 20) | class Kp2uvModel(BaseModel): method modify_commandline_options (line 23) | def modify_commandline_options(parser, is_train=True): method __init__ (line 27) | def __init__(self, opt): method set_input (line 41) | def set_input(self, input): method forward (line 50) | def forward(self): method output2rgb (line 55) | def output2rgb(self, output): method optimize_parameters (line 70) | def optimize_parameters(self): FILE: models/lnr_model.py class LnrModel (line 22) | class LnrModel(BaseModel): method modify_commandline_options (line 25) | def modify_commandline_options(parser, is_train=True): method __init__ (line 46) | def __init__(self, opt): method setup_train (line 63) | def setup_train(self, opt): method set_input (line 83) | def set_input(self, input): method gen_crop_params (line 99) | def gen_crop_params(self, orig_h, orig_w, crop_size=256): method forward (line 107) | def forward(self): method backward (line 133) | def backward(self): method optimize_parameters (line 147) | def optimize_parameters(self): method update_lambdas (line 154) | def update_lambdas(self, epoch): method transfer_detail (line 168) | def transfer_detail(self): method get_results (line 181) | def get_results(self): method freeze_basenet (line 201) | def freeze_basenet(self): FILE: models/networks.py function define_LNR (line 25) | def define_LNR(nf=64, texture_channels=16, texture_res=16, n_textures=25... function define_kp2uv (line 41) | def define_kp2uv(nf=64, gpu_ids=[]): function cal_alpha_reg (line 53) | def cal_alpha_reg(prediction, lambda_alpha_l1, lambda_alpha_l0): class MaskLoss (line 77) | class MaskLoss(nn.Module): method __init__ (line 80) | def __init__(self): method __call__ (line 84) | def __call__(self, prediction, target): class ConvBlock (line 104) | class ConvBlock(nn.Module): method __init__ (line 107) | def __init__(self, conv, in_channels, out_channels, ksize=4, stride=1,... method forward (line 140) | def forward(self, x): class ResBlock (line 169) | class ResBlock(nn.Module): method __init__ (line 172) | def __init__(self, channels, ksize=4, stride=1, dil=1, norm=None, acti... method forward (line 180) | def forward(self, x): class kp2uv (line 188) | class kp2uv(nn.Module): method __init__ (line 194) | def __init__(self, nf=64): method forward (line 217) | def forward(self, x): class LayeredNeuralRenderer (line 240) | class LayeredNeuralRenderer(nn.Module): method __init__ (line 246) | def __init__(self, nf=64, texture_channels=16, texture_res=16, n_textu... method render (line 285) | def render(self, x): method forward (line 304) | def forward(self, uv_map, id_layers, uv_map_upsampled=None, crop_param... FILE: options/base_options.py class BaseOptions (line 24) | class BaseOptions(): method __init__ (line 31) | def __init__(self): method initialize (line 35) | def initialize(self, parser): method gather_options (line 60) | def gather_options(self): method print_options (line 88) | def print_options(self, opt): method parse (line 113) | def parse(self): method parse_dataset_meta (line 138) | def parse_dataset_meta(self): FILE: options/test_options.py class TestOptions (line 18) | class TestOptions(BaseOptions): method initialize (line 24) | def initialize(self, parser): FILE: options/train_options.py class TrainOptions (line 18) | class TrainOptions(BaseOptions): method initialize (line 24) | def initialize(self, parser): FILE: third_party/data/__init__.py function find_dataset_using_name (line 19) | def find_dataset_using_name(dataset_name): function get_option_setter (line 42) | def get_option_setter(dataset_name): function create_dataset (line 48) | def create_dataset(opt, use_fast_loader=False): class CustomDatasetDataLoader (line 65) | class CustomDatasetDataLoader(): method __init__ (line 68) | def __init__(self, opt, use_fast_loader=False): method load_data (line 89) | def load_data(self): method __len__ (line 92) | def __len__(self): method __iter__ (line 96) | def __iter__(self): FILE: third_party/data/base_dataset.py class BaseDataset (line 13) | class BaseDataset(data.Dataset, ABC): method __init__ (line 23) | def __init__(self, opt): method modify_commandline_options (line 33) | def modify_commandline_options(parser, is_train): method __len__ (line 46) | def __len__(self): method __getitem__ (line 51) | def __getitem__(self, index): function get_params (line 63) | def get_params(opt, size): function get_transform (line 81) | def get_transform(opt, params=None, grayscale=False, method=Image.BICUBI... function __make_power_2 (line 115) | def __make_power_2(img, base, method=Image.BICUBIC): function __scale_width (line 126) | def __scale_width(img, target_size, crop_size, method=Image.BICUBIC): function __crop (line 135) | def __crop(img, pos, size): function __flip (line 144) | def __flip(img, flip): function __print_size_warning (line 150) | def __print_size_warning(ow, oh, w, h): FILE: third_party/data/fast_data_loader.py class _RepeatSampler (line 8) | class _RepeatSampler(object): method __init__ (line 15) | def __init__(self, sampler): method __iter__ (line 18) | def __iter__(self): class FastDataLoader (line 23) | class FastDataLoader(torch.utils.data.dataloader.DataLoader): method __init__ (line 25) | def __init__(self, *args, **kwargs): method __len__ (line 30) | def __len__(self): method __iter__ (line 33) | def __iter__(self): FILE: third_party/data/image_folder.py function is_image_file (line 19) | def is_image_file(filename): function make_dataset (line 23) | def make_dataset(dir, max_dataset_size=float("inf")): function default_loader (line 36) | def default_loader(path): class ImageFolder (line 40) | class ImageFolder(data.Dataset): method __init__ (line 42) | def __init__(self, root, transform=None, return_paths=False, method __getitem__ (line 55) | def __getitem__(self, index): method __len__ (line 65) | def __len__(self): FILE: third_party/models/__init__.py function find_model_using_name (line 25) | def find_model_using_name(model_name): function get_option_setter (line 48) | def get_option_setter(model_name): function create_model (line 54) | def create_model(opt): FILE: third_party/models/base_model.py class BaseModel (line 8) | class BaseModel(ABC): method __init__ (line 18) | def __init__(self, opt): method modify_commandline_options (line 45) | def modify_commandline_options(parser, is_train): method set_input (line 58) | def set_input(self, input): method forward (line 67) | def forward(self): method optimize_parameters (line 72) | def optimize_parameters(self): method setup (line 76) | def setup(self, opt): method eval (line 89) | def eval(self): method test (line 96) | def test(self): method compute_visuals (line 106) | def compute_visuals(self): method get_image_paths (line 110) | def get_image_paths(self): method update_learning_rate (line 114) | def update_learning_rate(self): method get_current_visuals (line 127) | def get_current_visuals(self): method get_current_losses (line 135) | def get_current_losses(self): method save_networks (line 143) | def save_networks(self, epoch): method __patch_instance_norm_state_dict (line 161) | def __patch_instance_norm_state_dict(self, state_dict, module, keys, i... method load_networks (line 175) | def load_networks(self, epoch): method print_networks (line 200) | def print_networks(self, verbose): method set_requires_grad (line 222) | def set_requires_grad(self, nets, requires_grad=False): FILE: third_party/models/networks.py function get_scheduler (line 9) | def get_scheduler(optimizer, opt): function init_net (line 39) | def init_net(net, gpu_ids=[]): FILE: third_party/util/html.py class HTML (line 6) | class HTML: method __init__ (line 14) | def __init__(self, web_dir, title, refresh=0): method get_image_dir (line 38) | def get_image_dir(self): method get_video_dir (line 42) | def get_video_dir(self): method add_header (line 46) | def add_header(self, text): method add_images (line 55) | def add_images(self, ims, txts, links, width=400): method add_videos (line 75) | def add_videos(self, vids, txts, links, width=400): method save (line 96) | def save(self): FILE: third_party/util/util.py function tensor2im (line 9) | def tensor2im(input_image, imtype=np.uint8): function render_png (line 30) | def render_png(image, background='checker'): function diagnose_network (line 46) | def diagnose_network(net, name='network'): function save_image (line 65) | def save_image(image_numpy, image_path, aspect_ratio=1.0): function print_numpy (line 83) | def print_numpy(x, val=True, shp=False): function mkdirs (line 99) | def mkdirs(paths): function mkdir (line 112) | def mkdir(path): FILE: third_party/util/visualizer.py function save_images (line 17) | def save_images(webpage, visuals, image_path, aspect_ratio=1.0, width=256): function save_videos (line 47) | def save_videos(webpage, visuals, width=256): class Visualizer (line 84) | class Visualizer(): method __init__ (line 90) | def __init__(self, opt): method reset (line 125) | def reset(self): method create_visdom_connections (line 129) | def create_visdom_connections(self): method display_current_results (line 136) | def display_current_results(self, visuals, epoch, save_result): method plot_current_losses (line 220) | def plot_current_losses(self, epoch, counter_ratio, losses): method print_current_losses (line 246) | def print_current_losses(self, epoch, iters, losses, t_comp, t_data): FILE: train.py function main (line 42) | def main(): function train (line 91) | def train(model, dataset, visualizer, opt):