SYMBOL INDEX (173 symbols across 29 files)

FILE: common/base.py
  function worker_init_fn (line 28) | def worker_init_fn(worder_id):
  class Base (line 31) | class Base(object):
    method __init__ (line 34) | def __init__(self, log_name='logs.txt'):
    method _make_batch_generator (line 46) | def _make_batch_generator(self):
    method _make_model (line 50) | def _make_model(self):
  class Trainer (line 53) | class Trainer(Base):
    method __init__ (line 54) | def __init__(self):
    method get_optimizer (line 57) | def get_optimizer(self, model):
    method save_model (line 64) | def save_model(self, state, epoch):
    method load_model (line 80) | def load_model(self, model, optimizer):
    method set_lr (line 92) | def set_lr(self, epoch):
    method get_lr (line 104) | def get_lr(self):
    method _make_batch_generator (line 109) | def _make_batch_generator(self):
    method _make_model (line 139) | def _make_model(self):
  class Tester (line 155) | class Tester(Base):
    method __init__ (line 156) | def __init__(self, test_epoch):
    method _make_batch_generator (line 160) | def _make_batch_generator(self):
    method _make_model (line 169) | def _make_model(self):
    method _evaluate (line 183) | def _evaluate(self, outs, cur_sample_idx):
    method _print_eval_result (line 187) | def _print_eval_result(self, eval_result):
  function check_data_parallel (line 190) | def check_data_parallel(train_weight):

FILE: common/logger.py
  class colorlogger (line 16) | class colorlogger():
    method __init__ (line 17) | def __init__(self, log_dir, log_name='train_logs.txt'):
    method debug (line 36) | def debug(self, msg):
    method info (line 39) | def info(self, msg):
    method warning (line 42) | def warning(self, msg):
    method critical (line 45) | def critical(self, msg):
    method error (line 48) | def error(self, msg):

FILE: common/nets/layer.py
  function make_linear_layers (line 7) | def make_linear_layers(feat_dims, relu_final=True, use_bn=False):
  function make_conv_layers (line 20) | def make_conv_layers(feat_dims, kernel=3, stride=1, padding=1, bnrelu_fi...
  function make_deconv_layers (line 38) | def make_deconv_layers(feat_dims, bnrelu_final=True):

FILE: common/nets/loss.py
  class ClothClsLoss (line 9) | class ClothClsLoss(nn.Module):
    method __init__ (line 10) | def __init__(self):
    method forward (line 26) | def forward(self, out, patch_idx, cloth_idx):
  class GenderClsLoss (line 49) | class GenderClsLoss(nn.Module):
    method __init__ (line 50) | def __init__(self):
    method forward (line 54) | def forward(self, out, gt):
  class SdfDPLoss (line 63) | class SdfDPLoss(nn.Module):
    method __init__ (line 64) | def __init__(self):
    method forward (line 67) | def forward(self, sdf, cloth_meshes_unposed, smpl_cloth_idx, smpl_clot...
  class RegLoss (line 100) | class RegLoss(nn.Module):
    method __init__ (line 101) | def __init__(self):
    method forward (line 105) | def forward(self, param, valid):
  class SdfParseLoss (line 110) | class SdfParseLoss(nn.Module):
    method __init__ (line 111) | def __init__(self):
    method forward (line 114) | def forward(self, sdf, cloth_meshes, parse_gt, sdf_thresh, cloth_meshe...

FILE: common/nets/module.py
  class ClothNet (line 8) | class ClothNet(nn.Module):
    method __init__ (line 9) | def __init__(self):
    method forward (line 34) | def forward(self, img_feat):

FILE: common/nets/resnet.py
  class ResNetBackbone (line 6) | class ResNetBackbone(nn.Module):
    method __init__ (line 8) | def __init__(self, resnet_type):
    method _make_layer (line 39) | def _make_layer(self, block, planes, blocks, stride=1):
    method forward (line 56) | def forward(self, x):
    method init_weights (line 68) | def init_weights(self):

FILE: common/timer.py
  class Timer (line 10) | class Timer(object):
    method __init__ (line 12) | def __init__(self):
    method tic (line 20) | def tic(self):
    method toc (line 25) | def toc(self, average=True):

FILE: common/utils/SMPLicit/SMPLicit/SMPL.py
  class SMPL (line 13) | class SMPL(nn.Module):
    method __init__ (line 14) | def __init__(self, model_path, joint_type = 'cocoplus', obj_saveable =...
    method save_obj (line 61) | def save_obj(self, verts, obj_mesh_name):
    method forward (line 73) | def forward(self, beta, theta, get_skin = False, theta_in_rodrigues=Tr...
    method deform_clothed_smpl (line 112) | def deform_clothed_smpl(self, theta, J, v_smpl, v_cloth):
    method unpose_and_deform_cloth (line 143) | def unpose_and_deform_cloth(self, v_cloth_posed, theta_from, theta_to,...
    method skeleton (line 198) | def skeleton(self,beta,require_body=False):

FILE: common/utils/SMPLicit/SMPLicit/SMPLicit.py
  class SMPLicit (line 13) | class SMPLicit(nn.Module):
    method __init__ (line 14) | def __init__(self, root_path, cloth_types):
    method get_right_shoe (line 95) | def get_right_shoe(self, sdf, unposed_cloth_mesh, do_marching_cube):
    method pose_mesh (line 111) | def pose_mesh(self, unposed_cloth_mesh, pose, unposed_smpl_joint, unpo...
    method pose_mesh_lower_body (line 133) | def pose_mesh_lower_body(self, unposed_cloth_mesh, pose, shape, Astar_...
    method forward (line 155) | def forward(self, z_cuts, z_styles, pose, shape, gender=[0], do_marchi...

FILE: common/utils/SMPLicit/SMPLicit/SMPLicit_options.py
  class Options (line 26) | class Options():
    method __init__ (line 27) | def __init__(self):

FILE: common/utils/SMPLicit/SMPLicit/network.py
  class Network (line 7) | class Network(nn.Module):
    method __init__ (line 8) | def __init__(self, n_z_style=1, point_pos_size=3, output_dim=1, n_z_cu...
    method forward (line 26) | def forward(self, z_cut, z_style, query):

FILE: common/utils/SMPLicit/SMPLicit/smplicit_core_test.py
  class Model (line 8) | class Model():
    method __init__ (line 9) | def __init__(self, filename, n_z_cut, n_z_style, num_clusters, name_cl...
    method load_networks (line 19) | def load_networks(self):
    method get_bbox (line 24) | def get_bbox(self, joint, mesh):
    method decode (line 102) | def decode(self, z_cut, z_style, smpl_joint, smpl_mesh, resolution, do...

FILE: common/utils/SMPLicit/SMPLicit/util_smpl.py
  function batch_rodrigues (line 27) | def batch_rodrigues(theta):
  function quat2mat (line 40) | def quat2mat(quat):
  function batch_global_rigid_transformation (line 62) | def batch_global_rigid_transformation(Rs, Js, parent, rotate_base = False):
  function batch_lrotmin (line 98) | def batch_lrotmin(theta):
  function batch_orth_proj (line 107) | def batch_orth_proj(X, camera):
  function reflect_pose (line 116) | def reflect_pose(poses):

FILE: common/utils/SMPLicit/SMPLicit/utils/sdf.py
  function create_grid (line 5) | def create_grid(resolution, b_min, b_max):
  function batch_eval (line 30) | def batch_eval(query_points, ref_points, z_cut, z_style, eval_func, scal...
  function eval_grid (line 40) | def eval_grid(query_points, ref_points, z_cut, z_style, eval_func, resol...
  function eval_grid_octree (line 44) | def eval_grid_octree(query_points, ref_points, z_cut, z_style, eval_func...

FILE: common/utils/dir.py
  function make_folder (line 4) | def make_folder(folder_name):
  function add_pypath (line 8) | def add_pypath(path):

FILE: common/utils/human_models.py
  class SMPL (line 8) | class SMPL(object):
    method __init__ (line 9) | def __init__(self):
    method get_custom_template_layer (line 28) | def get_custom_template_layer(self, v_template, gender):

FILE: common/utils/postprocessing.py
  function get_face_map (line 18) | def get_face_map(pix_to_face, faces):
  class Renderer (line 27) | class Renderer:
    method __init__ (line 28) | def __init__(self, device='cuda', focal=cfg.focal, princpt=cfg.princpt...
    method set_renderer (line 33) | def set_renderer(self, focal, princpt, img_shape, anti_aliasing=False):
    method rasterize_mesh (line 61) | def rasterize_mesh(self, mesh_vert, mesh_face):
    method render (line 67) | def render(self, img, mesh_vert, mesh_face):
  function rasterize_mesh_given_cam_param (line 95) | def rasterize_mesh_given_cam_param(mesh_vert, mesh_face, focal, princpt):
  function save_proj_faces (line 112) | def save_proj_faces(face_map, save_path):
  function merge_mesh (line 120) | def merge_mesh(verts, faces):
  function read_valid_point (line 131) | def read_valid_point(verts, indexs, valid):
  function pa_mpjpe (line 141) | def pa_mpjpe(predicted, target):
  function pairwise_distances (line 178) | def pairwise_distances(a, b, p=2, inv=False, num_samples=500):

FILE: common/utils/preprocessing.py
  function load_img (line 10) | def load_img(path, order='RGB'):
  function get_bbox (line 21) | def get_bbox(joint_img, joint_valid, extend_ratio=1.2):
  function process_bbox (line 38) | def process_bbox(bbox, img_width, img_height):
  function convert_focal_princpt (line 68) | def convert_focal_princpt(focal, princpt, img2bb_trans):
  function get_aug_config (line 78) | def get_aug_config():
  function augmentation (line 92) | def augmentation(img, bbox, data_split):
  function generate_patch_image (line 102) | def generate_patch_image(cvimg, bbox, scale, rot, do_flip, out_shape):
  function rotate_2d (line 126) | def rotate_2d(pt_2d, rot_rad):
  function gen_trans_from_patch_cv (line 134) | def gen_trans_from_patch_cv(c_x, c_y, src_width, src_height, dst_width, ...
  function process_db_coord (line 169) | def process_db_coord(joint_img, joint_valid, do_flip, img_shape, flip_pa...
  function process_human_model_output (line 196) | def process_human_model_output(human_model_param, cam_param, do_flip, im...
  function bilinear_interpolate (line 260) | def bilinear_interpolate(im, x, y):
  function iou_sil (line 286) | def iou_sil(sil_out, sil_target):

FILE: common/utils/transforms.py
  function cam2pixel (line 7) | def cam2pixel(cam_coord, f, c):
  function pixel2cam (line 13) | def pixel2cam(pixel_coord, f, c):
  function world2cam (line 19) | def world2cam(world_coord, R, t):
  function transform_joint_to_other_db (line 23) | def transform_joint_to_other_db(src_joint, src_name, dst_name):
  function rot6d_to_axis_angle (line 36) | def rot6d_to_axis_angle(x):
  function unwrap_xy_to_uv (line 52) | def unwrap_xy_to_uv(feat_xy, dp_fg, dp_I, dp_u, dp_v):

FILE: common/utils/vis.py
  function save_result (line 11) | def save_result(output, path):
  function render_result (line 31) | def render_result(verts, faces, img, path):
  function vis_keypoints_with_skeleton (line 36) | def vis_keypoints_with_skeleton(img, kps, kps_lines, kp_thresh=0.4, alph...
  function vis_keypoints (line 67) | def vis_keypoints(img, kps, alpha=1):
  function vis_3d_skeleton (line 84) | def vis_3d_skeleton(kpt_3d, kpt_3d_vis, kps_lines, filename=None):
  function save_obj (line 125) | def save_obj(v, f, file_name='output.obj'):
  function save_obj_with_color (line 134) | def save_obj_with_color(v, f, c, file_name='output.obj'):
  function vis_parse (line 143) | def vis_parse(img, parse, class_num):
  function vis_dp (line 153) | def vis_dp(img, dp_u, dp_v, filename):

FILE: data/DeepFashion2/DeepFashion2.py
  class DeepFashion2 (line 16) | class DeepFashion2(torch.utils.data.Dataset):
    method __init__ (line 17) | def __init__(self, transform, data_split):
    method load_data (line 37) | def load_data(self):
    method __len__ (line 111) | def __len__(self):
    method __getitem__ (line 114) | def __getitem__(self, idx):

FILE: data/MSCOCO/MSCOCO.py
  class MSCOCO (line 15) | class MSCOCO(torch.utils.data.Dataset):
    method __init__ (line 16) | def __init__(self, transform, data_split):
    method load_data (line 33) | def load_data(self):
    method __len__ (line 160) | def __len__(self):
    method __getitem__ (line 163) | def __getitem__(self, idx):
    method evaluate (line 351) | def evaluate(self, outs, cur_sample_idx):
    method print_eval_result (line 418) | def print_eval_result(self, eval_result):

FILE: data/PW3D/PW3D.py
  class PW3D (line 16) | class PW3D(torch.utils.data.Dataset):
    method __init__ (line 17) | def __init__(self, transform, data_split):
    method load_data (line 32) | def load_data(self):
    method __len__ (line 84) | def __len__(self):
    method __getitem__ (line 87) | def __getitem__(self, idx):
    method get_clothed_mesh (line 118) | def get_clothed_mesh(self, sequence, pid, pose, shape, trans, gender):
    method evaluate (line 134) | def evaluate(self, outs, cur_sample_idx):
    method print_eval_result (line 198) | def print_eval_result(self, eval_result):

FILE: data/dataset.py
  class MultipleDatasets (line 6) | class MultipleDatasets(Dataset):
    method __init__ (line 7) | def __init__(self, dbs, make_same_len=True):
    method __len__ (line 14) | def __len__(self):
    method __getitem__ (line 22) | def __getitem__(self, index):

FILE: demo/demo.py
  function parse_args (line 17) | def parse_args():

FILE: main/config.py
  class Config (line 7) | class Config:
    method set_args (line 67) | def set_args(self, gpu_ids, continue_train=False):

FILE: main/model.py
  class Model (line 15) | class Model(nn.Module):
    method __init__ (line 16) | def __init__(self, backbone, cloth_net, mode):
    method forward (line 35) | def forward(self, inputs, targets, meta_info, mode):
    method get_coords (line 130) | def get_coords(self, root_pose, params, cam_trans, gender):
  function init_weights (line 146) | def init_weights(m):
  function get_model (line 160) | def get_model(mode):

FILE: main/test.py
  function parse_args (line 9) | def parse_args():
  function main (line 28) | def main():

FILE: main/train.py
  function parse_args (line 9) | def parse_args():
  function main (line 26) | def main():