SYMBOL INDEX (168 symbols across 20 files) FILE: apps/render_smpl_depth.py function load_obj_mesh (line 19) | def load_obj_mesh(mesh_file, with_normal=False, with_texture=False, with... function extract_float (line 139) | def extract_float(text): function natural_sort (line 149) | def natural_sort(files): function load_ply (line 155) | def load_ply(file_name): function distortPoints (line 230) | def distortPoints(p, dist): function rasterize (line 252) | def rasterize(v, tri, size, K = np.identity(3), \ function render_view (line 297) | def render_view(intri, dists, c2ws, meshes, view, i): class Worker (line 328) | class Worker(Process): method __init__ (line 330) | def __init__(self, queue, lock): method run (line 335) | def run(self): FILE: apps/run_genebody.py function loss_string (line 26) | def loss_string(loss_dict): function print_write (line 32) | def print_write(file, string): function to8b (line 37) | def to8b(img): function prepare_data (line 50) | def prepare_data(opt, data, local_rank=0): function cal_metrics (line 94) | def cal_metrics(metrics, rgbs, gts): function train (line 105) | def train(opt, rank=0, local_rank = 0): FILE: genebody/download_tool.py function import_or_install (line 4) | def import_or_install(package): function parse_args (line 47) | def parse_args(): function newSession (line 66) | def newSession(): function save_hash (line 72) | def save_hash(path, code): function read_hash (line 76) | def read_hash(path): function checkHashes (line 81) | def checkHashes(localfile, cloud_hash, localroot, force): function getFiles (line 141) | def getFiles(originalUrl, download_path, force, download_root=None, req=... function fetch_with_pwd (line 267) | async def fetch_with_pwd(iurl, password): function havePwdGetFiles (line 298) | def havePwdGetFiles(iurl, password, download_path, force): function extractFiles (line 304) | def extractFiles(path, subset): function moveFiles (line 327) | def moveFiles(root): FILE: genebody/genebody.py function image_cropping (line 8) | def image_cropping(mask, padding=0.1): class GeneBodyReader (line 65) | class GeneBodyReader(): method __init__ (line 66) | def __init__(self, rootdir, loadsize=512): method get_views (line 75) | def get_views(self, subject): method get_frames (line 93) | def get_frames(self, subject): method get_cameras (line 99) | def get_cameras(self, subject): method get_smpl (line 102) | def get_smpl(self, subject, frame): method get_smpl_param (line 111) | def get_smpl_param(self, subject, frame): method get_data (line 133) | def get_data(self, subject, frame, camera_params, views): method get_near_far (line 172) | def get_near_far(self, verts, c2w, pad=0.5): method smpl_from_param (line 188) | def smpl_from_param(self, model_path, subject, smpl_param, smpl_scale): FILE: genebody/mesh.py function max_precision (line 49) | def max_precision(type1, type2): function decode (line 83) | def decode(content, structure, num, form): function load_ply (line 131) | def load_ply(file_name): function save_ply (line 192) | def save_ply(file_name, elems, _type = 'binary_little_endian', comments ... function normalize_v3 (line 280) | def normalize_v3(arr): function compute_normal (line 291) | def compute_normal(vertices, faces): function load_obj_mesh (line 312) | def load_obj_mesh(mesh_file, with_normal=False, with_texture=False, with... function write_obj_mesh (line 433) | def write_obj_mesh(filename, verts, faces): FILE: lib/data/GeneBodyDataset.py function mask_padding (line 22) | def mask_padding(mask, border = 5): function euler2rot (line 31) | def euler2rot(euler): function rot2euler (line 46) | def rot2euler(R): function gen_cam_views (line 52) | def gen_cam_views(transl, z_pitch, viewnum): class GeneBodyDataset (line 79) | class GeneBodyDataset(Dataset): method modify_commandline_options (line 81) | def modify_commandline_options(parser): method __init__ (line 83) | def __init__(self, opt, phase='eval', root=None, move_cam=0): method get_frames (line 126) | def get_frames(self, i = 0): method get_render_poses (line 145) | def get_render_poses(self, annots, move_cam=150): method __len__ (line 159) | def __len__(self): method image_cropping (line 165) | def image_cropping(self, mask): method get_near_far (line 208) | def get_near_far(self, smpl_verts, w2c): method get_image (line 230) | def get_image(self, sid, num_views, view_id=None, random_sample=False,... method smpl_from_param (line 387) | def smpl_from_param(self, model_path, subject, smpl_param, smpl_scale): method get_item (line 410) | def get_item(self, index): method __getitem__ (line 475) | def __getitem__(self, index): FILE: lib/geometry.py function rot2euler (line 4) | def rot2euler(R): function euler2rot (line 10) | def euler2rot(euler): function batch_rodrigues (line 25) | def batch_rodrigues(theta): function quat_to_rotmat (line 41) | def quat_to_rotmat(quat): function index (line 64) | def index(feat, uv, mode='bilinear'): function orthogonal (line 81) | def orthogonal(points, calibrations, transforms=None): function perspective (line 100) | def perspective(points, w2c, camera): FILE: lib/mesh_util.py function save_obj_mesh (line 7) | def save_obj_mesh(mesh_path, verts, faces): function save_obj_mesh_with_color (line 18) | def save_obj_mesh_with_color(mesh_path, verts, faces, colors): function save_obj_mesh_with_uv (line 30) | def save_obj_mesh_with_uv(mesh_path, verts, faces, uvs): FILE: lib/metrics.py function chamfer (line 21) | def chamfer(x_verts, gt_verts, x_normals=None, gt_normals=None): function fscore (line 36) | def fscore(dist1, dist2, threshold=1.0): function psnr (line 51) | def psnr(x, gt): function ssim_channel (line 64) | def ssim_channel(x, gt): function ssim (line 85) | def ssim(x, gt): function lpips (line 107) | def lpips(x, gt, net=lpips_net): FILE: lib/metrics_torch.py class LPIPS (line 8) | class LPIPS(torch.nn.Module): method __init__ (line 9) | def __init__(self): method forward (line 13) | def forward(self, x, gt): function psnr (line 27) | def psnr(x, gt): function gaussian (line 46) | def gaussian(window_size, sigma): function create_window (line 51) | def create_window(window_size, channel=1): function ssim_ (line 58) | def ssim_(img1, img2, window_size=11, window=None, size_average=True, fu... class SSIM (line 114) | class SSIM(torch.nn.Module): method __init__ (line 115) | def __init__(self, window_size=11, size_average=True, val_range=None): method forward (line 125) | def forward(self, img1, img2): FILE: lib/model/Embedder.py class PositionalEncoding (line 8) | class PositionalEncoding: method __init__ (line 12) | def __init__(self, d, num_freqs=10, min_freq=None, max_freq=None, freq... method create_embedding_fn (line 19) | def create_embedding_fn(self, d): method embed (line 44) | def embed(self, inputs): class SphericalHarmonics (line 47) | class SphericalHarmonics: method __init__ (line 51) | def __init__(self, d = 3, rank = 3): method Lengdre_polynormial (line 56) | def Lengdre_polynormial(self, x, omx = None): method SH (line 69) | def SH(self, xyz): method embed (line 90) | def embed(self, inputs): FILE: lib/model/GNR.py class GNR (line 12) | class GNR(nn.Module): method __init__ (line 14) | def __init__(self, opt): method image_rescale (line 40) | def image_rescale(self, images, masks): method get_image_feature (line 46) | def get_image_feature(self, data): method forward (line 57) | def forward(self, data, train_shape=False): method render_path (line 66) | def render_path(self, data): method reconstruct (line 74) | def reconstruct(self, data): FILE: lib/model/HGFilters.py class HourGlass (line 12) | class HourGlass(nn.Module): method __init__ (line 13) | def __init__(self, num_modules, depth, num_features, norm='batch'): method _generate_network (line 22) | def _generate_network(self, level): method _forward (line 34) | def _forward(self, level, inp): method forward (line 60) | def forward(self, x): class HGFilter (line 64) | class HGFilter(nn.Module): method __init__ (line 65) | def __init__(self, opt): method forward (line 114) | def forward(self, x): FILE: lib/model/NeRF.py class NeRF (line 9) | class NeRF(nn.Module): method __init__ (line 10) | def __init__(self, opt, D=8, W=256, input_ch=3, input_ch_atts=3, input... method forward (line 98) | def forward(self, x, attdirs=None, alpha_only=False, smpl_vis=None): method weighted_softmax (line 191) | def weighted_softmax(self, attention, weight): FILE: lib/model/NeRFRenderer.py class NeRFRenderer (line 23) | class NeRFRenderer: method __init__ (line 24) | def __init__(self, opt, nerf, nerf_fine=None, projection='perspective'... method cal_loss (line 82) | def cal_loss(self, rgb, rgb_gt): method get_rays_orthogonal (line 99) | def get_rays_orthogonal(self, bbox, calib): method get_rays_perspective (line 120) | def get_rays_perspective(self, bbox, w2c, cam): method make_att_input (line 155) | def make_att_input(self, pts, viewdirs, calibs, smpl): method make_nerf_input (line 182) | def make_nerf_input(self, pts, feats, images, smpl, calibs, mesh_param... method make_nerf_output (line 243) | def make_nerf_output(self, nerf_output, t_vals, norm, source_rgb, is_t... method render_rays (line 276) | def render_rays(self, ray_batch, feats, images, masks, calibs, smpl, m... method inside_pts_vh (line 364) | def inside_pts_vh(self, pts, masks, smpl, calibs, persps=None): method render (line 393) | def render(self, feats, images, masks, calibs, bbox, mesh_param, smpl=... method render_path (line 423) | def render_path(self, feats, images, masks, calibs, bbox, mesh_param, ... method train_shape (line 462) | def train_shape(self, feats, images, masks, calibs, bbox, mesh_param, ... method reconstruct (line 480) | def reconstruct(self, feats, images, masks, calibs, bbox, mesh_param, ... method octree_reconstruct (line 534) | def octree_reconstruct(self, coords, masks, **kwargs): method get_plucker_line (line 608) | def get_plucker_line(self, countour, w2c, cam, mesh_param): method warp_mat (line 640) | def warp_mat(self, face_verts): method canonical_warpping (line 660) | def canonical_warpping(self, verts, verts_can, pts): method multiview_consistency (line 672) | def multiview_consistency(self, pts, feats, calibs, depth, persps=None): FILE: lib/model/SRFilters.py class SRFilters (line 5) | class SRFilters(nn.Module): method __init__ (line 9) | def __init__(self, order=2, in_ch=256, out_ch=128): method forward (line 17) | def forward(self, feat, images): FILE: lib/net_ddp.py function worker_init_fn (line 11) | def worker_init_fn(worker_id): function ddp_init (line 16) | def ddp_init(args): function create_network (line 32) | def create_network(opt, net, local_rank=0): function synchronize (line 81) | def synchronize(): class ddpSampler (line 86) | class ddpSampler: method __init__ (line 90) | def __init__(self, dataset, rank=None, num_replicas=None): method indices (line 105) | def indices(self): method len (line 113) | def len(self): method distributed_concat (line 116) | def distributed_concat(self, tensor, num_total_examples): FILE: lib/net_util.py function conv3x3 (line 14) | def conv3x3(in_planes, out_planes, strd=1, padding=1, bias=False): function init_weights (line 19) | def init_weights(net, init_type='normal', init_gain=0.02): function init_net (line 55) | def init_net(net, init_type='normal', init_gain=0.02, gpu_ids=[]): function imageSpaceRotation (line 73) | def imageSpaceRotation(xy, rot): function cal_gradient_penalty (line 85) | def cal_gradient_penalty(netD, real_data, fake_data, device, type='mixed... function get_norm_layer (line 123) | def get_norm_layer(norm_type='instance'): class Flatten (line 142) | class Flatten(nn.Module): method forward (line 143) | def forward(self, input): class ConvBlock (line 146) | class ConvBlock(nn.Module): method __init__ (line 147) | def __init__(self, in_planes, out_planes, norm='batch'): method forward (line 174) | def forward(self, x): FILE: lib/options.py class BaseOptions (line 6) | class BaseOptions(): method __init__ (line 7) | def __init__(self): method initialize (line 10) | def initialize(self, parser): method gather_options (line 136) | def gather_options(self): method print_options (line 148) | def print_options(self, opt): method parse (line 160) | def parse(self): FILE: lib/ply_util.py function max_precision (line 47) | def max_precision(type1, type2): function decode (line 80) | def decode(content, structure, num, form): function load_ply (line 127) | def load_ply(file_name): function save_ply (line 187) | def save_ply(file_name, elems, _type = 'binary_little_endian', comments ...