SYMBOL INDEX (235 symbols across 40 files)

FILE: common/base.py
  class Base (line 20) | class Base(object):
    method __init__ (line 23) | def __init__(self, log_name='logs.txt'):
    method _make_batch_generator (line 36) | def _make_batch_generator(self):
    method _make_model (line 40) | def _make_model(self):
  class Trainer (line 43) | class Trainer(Base):
    method __init__ (line 44) | def __init__(self):
    method get_optimizer (line 47) | def get_optimizer(self, model):
    method save_model (line 52) | def save_model(self, state, epoch):
    method load_model (line 57) | def load_model(self, model, optimizer):
    method set_lr (line 69) | def set_lr(self, epoch):
    method get_lr (line 81) | def get_lr(self):
    method _make_batch_generator (line 86) | def _make_batch_generator(self):
    method _make_model (line 94) | def _make_model(self):
  class Tester (line 111) | class Tester(Base):
    method __init__ (line 112) | def __init__(self, test_epoch):
    method _make_batch_generator (line 116) | def _make_batch_generator(self):
    method _make_model (line 122) | def _make_model(self):
    method _evaluate (line 137) | def _evaluate(self, outs, cur_sample_idx):
    method _print_eval_result (line 141) | def _print_eval_result(self, test_epoch):

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/backbone.py
  class FPN (line 10) | class FPN(nn.Module):
    method __init__ (line 11) | def __init__(self, pretrained=True):
    method _upsample_add (line 40) | def _upsample_add(self, x, y):
    method forward (line 44) | def forward(self, x):
  class ResNet (line 68) | class ResNet(nn.Module):
    method __init__ (line 69) | def __init__(self, block, layers, num_classes=1000):
    method _make_layer (line 90) | def _make_layer(self, block, planes, blocks, stride=1):
    method forward (line 105) | def forward(self, x):
  function resnet50 (line 122) | def resnet50(pretrained=False, **kwargs):
  function conv3x3 (line 130) | def conv3x3(in_planes, out_planes, stride=1):
  class BasicBlock (line 134) | class BasicBlock(nn.Module):
    method __init__ (line 137) | def __init__(self, inplanes, planes, stride=1, downsample=None):
    method forward (line 147) | def forward(self, x):
  class Bottleneck (line 166) | class Bottleneck(nn.Module):
    method __init__ (line 169) | def __init__(self, inplanes, planes, stride=1, downsample=None):
    method forward (line 185) | def forward(self, x):

FILE: common/nets/cbam.py
  class BasicConv (line 6) | class BasicConv(nn.Module):
    method __init__ (line 7) | def __init__(self, in_planes, out_planes, kernel_size, stride=1, paddi...
    method forward (line 14) | def forward(self, x):
  class Flatten (line 22) | class Flatten(nn.Module):
    method forward (line 23) | def forward(self, x):
  class ChannelGate (line 26) | class ChannelGate(nn.Module):
    method __init__ (line 27) | def __init__(self, gate_channels, reduction_ratio=16, pool_types=['avg...
    method forward (line 37) | def forward(self, x):
  function logsumexp_2d (line 62) | def logsumexp_2d(tensor):
  class ChannelPool (line 68) | class ChannelPool(nn.Module):
    method forward (line 69) | def forward(self, x):
  class SpatialGate (line 72) | class SpatialGate(nn.Module):
    method __init__ (line 73) | def __init__(self):
    method forward (line 78) | def forward(self, x):
  class CBAM (line 84) | class CBAM(nn.Module):
    method __init__ (line 85) | def __init__(self, gate_channels, reduction_ratio=16, pool_types=['avg...
    method forward (line 91) | def forward(self, x):

FILE: common/nets/hand_head.py
  class hand_regHead (line 5) | class hand_regHead(nn.Module):
    method __init__ (line 6) | def __init__(self, roi_res=32, joint_nb=21, stacks=1, channels=256, bl...
    method make_residual (line 51) | def make_residual(self, block, inplanes, planes, blocks, stride=1):
    method spatial_softmax (line 62) | def spatial_softmax(self, latents):
    method generate_output (line 69) | def generate_output(self, heatmaps):
    method forward (line 75) | def forward(self, x):
  class BasicBlock (line 96) | class BasicBlock(nn.Module):
    method __init__ (line 97) | def __init__(self, in_planes, out_planes, kernel_size,groups=1):
    method forward (line 107) | def forward(self, x):
  class Residual (line 111) | class Residual(nn.Module):
    method __init__ (line 112) | def __init__(self, numIn, numOut):
    method forward (line 127) | def forward(self, x):
  class Bottleneck (line 145) | class Bottleneck(nn.Module):
    method __init__ (line 148) | def __init__(self, inplanes, planes, stride=1, skip=None, groups=1):
    method forward (line 162) | def forward(self, x):
  class Hourglass (line 185) | class Hourglass(nn.Module):
    method __init__ (line 186) | def __init__(self, block, num_blocks, planes, depth):
    method _make_residual (line 193) | def _make_residual(self, block, num_blocks, planes):
    method _make_hour_glass (line 201) | def _make_hour_glass(self, block, num_blocks, planes, depth):
    method _hour_glass_forward (line 217) | def _hour_glass_forward(self, n, x):
    method forward (line 231) | def forward(self, x):
  class hand_Encoder (line 237) | class hand_Encoder(nn.Module):
    method __init__ (line 238) | def __init__(self, num_heatmap_chan=21, num_feat_chan=256, size_input_...
    method forward (line 266) | def forward(self, hm_list, encoding_list):

FILE: common/nets/mano_head.py
  function batch_rodrigues (line 7) | def batch_rodrigues(theta):
  function quat2mat (line 20) | def quat2mat(quat):
  function quat2aa (line 39) | def quat2aa(quaternion):
  function mat2quat (line 72) | def mat2quat(rotation_matrix, eps=1e-6):
  function rot6d2mat (line 134) | def rot6d2mat(x):
  function mat2aa (line 146) | def mat2aa(rotation_matrix):
  class mano_regHead (line 167) | class mano_regHead(nn.Module):
    method __init__ (line 168) | def __init__(self, mano_layer=mano.layer, feature_size=1024, mano_neur...
    method forward (line 190) | def forward(self, features, gt_mano_params=None):

FILE: common/nets/regressor.py
  class Regressor (line 8) | class Regressor(nn.Module):
    method __init__ (line 9) | def __init__(self):
    method forward (line 15) | def forward(self, feats, gt_mano_params=None):

FILE: common/nets/transformer.py
  class Transformer (line 6) | class Transformer(nn.Module):
    method __init__ (line 7) | def __init__(self, inp_res=32, dim=256, depth=2, num_heads=4, mlp_rati...
    method forward (line 26) | def forward(self, query, key):
  class Mlp (line 37) | class Mlp(nn.Module):
    method __init__ (line 38) | def __init__(self, in_features, hidden_features=None, out_features=Non...
    method forward (line 48) | def forward(self, x):
    method _init_weights (line 56) | def _init_weights(self):
  class Attention (line 63) | class Attention(nn.Module):
    method __init__ (line 64) | def __init__(self, dim, num_heads=1):
    method forward (line 71) | def forward(self, query, key, value, query2, key2, use_sigmoid):
  class Block (line 90) | class Block(nn.Module):
    method __init__ (line 92) | def __init__(self, dim, num_heads, mlp_ratio=4., act_layer=nn.GELU, no...
    method with_pos_embed (line 114) | def with_pos_embed(self, tensor, pos):
    method forward (line 117) | def forward(self, query, key, query_embed=None, key_embed=None):

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/camera.py
  function transform_mat (line 33) | def transform_mat(R: torch.tensor, t: torch.tensor) -> torch.Tensor:
  function create_camera (line 47) | def create_camera(camera_type='persp', **kwargs):
  class PerspectiveCamera (line 54) | class PerspectiveCamera(nn.Module):
    method __init__ (line 58) | def __init__(self, rotation=None, translation=None,
    method forward (line 106) | def forward(self, points):

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/fitting.py
  function to_tensor (line 7) | def to_tensor(tensor, dtype=torch.float32):
  function rel_change (line 14) | def rel_change(prev_val, curr_val):
  class FittingMonitor (line 17) | class FittingMonitor(object):
    method __init__ (line 18) | def __init__(self, summary_steps=1, visualize=False,
    method __enter__ (line 34) | def __enter__(self):
    method __exit__ (line 39) | def __exit__(self, exception_type, exception_value, traceback):
    method set_colors (line 42) | def set_colors(self, vertex_color):
    method run_fitting (line 49) | def run_fitting(self, optimizer, closure, params,
    method create_fitting_closure (line 92) | def create_fitting_closure(self,
  class ScaleTranslationLoss (line 128) | class ScaleTranslationLoss(nn.Module):
    method __init__ (line 130) | def __init__(self, init_joints_idxs, trans_estimation=None,
    method reset_loss_weights (line 153) | def reset_loss_weights(self, loss_weight_dict):
    method forward (line 162) | def forward(self, camera, joint_cam, joint_img, hand_translation, hand...

FILE: common/utils/mano.py
  class MANO (line 12) | class MANO(object):
    method __init__ (line 13) | def __init__(self):
    method get_layer (line 38) | def get_layer(self):

FILE: common/utils/manopth/mano/webuser/lbs.py
  function global_rigid_transformation (line 27) | def global_rigid_transformation(pose, J, kintree_table, xp):
  function verts_core (line 68) | def verts_core(pose, v, J, weights, kintree_table, want_Jtr=False, xp=ch...

FILE: common/utils/manopth/mano/webuser/posemapper.py
  class Rodrigues (line 27) | class Rodrigues(ch.Ch):
    method compute_r (line 30) | def compute_r(self):
    method compute_dr_wrt (line 33) | def compute_dr_wrt(self, wrt):
  function lrotmin (line 38) | def lrotmin(p):
  function posemap (line 51) | def posemap(s):

FILE: common/utils/manopth/mano/webuser/serialization.py
  function ready_arguments (line 31) | def ready_arguments(fname_or_dict):
  function load_model (line 73) | def load_model(fname_or_dict):

FILE: common/utils/manopth/mano/webuser/smpl_handpca_wrapper_HAND_only.py
  function ready_arguments (line 22) | def ready_arguments(fname_or_dict, posekey4vposed='pose'):
  function load_model (line 70) | def load_model(fname_or_dict, ncomps=6, flat_hand_mean=False, v_template...

FILE: common/utils/manopth/mano/webuser/verts.py
  function ischumpy (line 29) | def ischumpy(x):
  function verts_decorated (line 33) | def verts_decorated(trans,
  function verts_core (line 107) | def verts_core(pose,

FILE: common/utils/manopth/manopth/argutils.py
  function print_args (line 8) | def print_args(args):
  function save_args (line 16) | def save_args(args, save_folder, opt_prefix='opt', verbose=True):

FILE: common/utils/manopth/manopth/demo.py
  function generate_random_hand (line 10) | def generate_random_hand(batch_size=1, ncomps=6, mano_root='mano/models'):
  function display_hand (line 18) | def display_hand(hand_info, mano_faces=None, ax=None, alpha=0.2, batch_i...
  function cam_equal_aspect_3d (line 43) | def cam_equal_aspect_3d(ax, verts, flip_x=False):

FILE: common/utils/manopth/manopth/manolayer.py
  class ManoLayer (line 13) | class ManoLayer(Module):
    method __init__ (line 19) | def __init__(self,
    method forward (line 110) | def forward(self,

FILE: common/utils/manopth/manopth/rodrigues_layer.py
  function quat2mat (line 15) | def quat2mat(quat):
  function batch_rodrigues (line 43) | def batch_rodrigues(axisang):
  function th_get_axis_angle (line 57) | def th_get_axis_angle(vector):

FILE: common/utils/manopth/manopth/rot6d.py
  function compute_rotation_matrix_from_ortho6d (line 4) | def compute_rotation_matrix_from_ortho6d(poses):
  function robust_compute_rotation_matrix_from_ortho6d (line 26) | def robust_compute_rotation_matrix_from_ortho6d(poses):
  function normalize_vector (line 54) | def normalize_vector(v):
  function cross_product (line 63) | def cross_product(u, v):

FILE: common/utils/manopth/manopth/rotproj.py
  function batch_rotprojs (line 4) | def batch_rotprojs(batches_rotmats):

FILE: common/utils/manopth/manopth/tensutils.py
  function th_posemap_axisang (line 6) | def th_posemap_axisang(pose_vectors):
  function th_with_zeros (line 15) | def th_with_zeros(tensor):
  function th_pack (line 25) | def th_pack(tensor):
  function subtract_flat_id (line 34) | def subtract_flat_id(rot_mats):
  function make_list (line 45) | def make_list(tensor):

FILE: common/utils/manopth/setup.py
  function check_dependencies (line 8) | def check_dependencies():

FILE: common/utils/manopth/test/test_demo.py
  function test_generate_random_hand (line 6) | def test_generate_random_hand():

FILE: common/utils/optimizers/lbfgs_ls.py
  function _cubic_interpolate (line 11) | def _cubic_interpolate(x1, f1, g1, x2, f2, g2, bounds=None):
  function _strong_Wolfe (line 39) | def _strong_Wolfe(obj_func, x, t, d, f, g, gtd, c1=1e-4, c2=0.9, toleran...
  class LBFGS (line 172) | class LBFGS(Optimizer):
    method __init__ (line 199) | def __init__(self, params, lr=1, max_iter=20, max_eval=None,
    method _numel (line 216) | def _numel(self):
    method _gather_flat_grad (line 221) | def _gather_flat_grad(self):
    method _add_grad (line 233) | def _add_grad(self, step_size, update):
    method _clone_param (line 242) | def _clone_param(self):
    method _set_param (line 245) | def _set_param(self, params_data):
    method _directional_evaluate (line 249) | def _directional_evaluate(self, closure, x, t, d):
    method step (line 256) | def step(self, closure):

FILE: common/utils/optimizers/optim_factory.py
  function create_optimizer (line 27) | def create_optimizer(parameters, optim_type='lbfgs',

FILE: common/utils/preprocessing.py
  function load_img (line 8) | def load_img(path, order='RGB'):
  function get_bbox (line 19) | def get_bbox(joint_img, joint_valid, expansion_factor=1.0):
  function process_bbox (line 36) | def process_bbox(bbox, img_width, img_height, expansion_factor=1.25):
  function get_aug_config (line 65) | def get_aug_config():
  function augmentation (line 79) | def augmentation(img, bbox, data_split, do_flip=False):
  function generate_patch_image (line 89) | def generate_patch_image(cvimg, bbox, scale, rot, do_flip, out_shape):
  function rotate_2d (line 109) | def rotate_2d(pt_2d, rot_rad):
  function gen_trans_from_patch_cv (line 117) | def gen_trans_from_patch_cv(c_x, c_y, src_width, src_height, dst_width, ...

FILE: common/utils/transforms.py
  function cam2pixel (line 5) | def cam2pixel(cam_coord, f, c):
  function pixel2cam (line 11) | def pixel2cam(pixel_coord, f, c):
  function world2cam (line 17) | def world2cam(world_coord, R, t):
  function cam2world (line 21) | def cam2world(cam_coord, R, t):
  function rigid_transform_3D (line 25) | def rigid_transform_3D(A, B):
  function rigid_align (line 43) | def rigid_align(A, B):
  function transform_joint_to_other_db (line 48) | def transform_joint_to_other_db(src_joint, src_name, dst_name):

FILE: common/utils/vis.py
  function vis_keypoints_with_skeleton (line 11) | def vis_keypoints_with_skeleton(img, kps, kps_lines, kp_thresh=0.4, alph...
  function vis_keypoints (line 42) | def vis_keypoints(img, kps, alpha=1):
  function vis_mesh (line 59) | def vis_mesh(img, mesh_vertex, alpha=0.5):
  function vis_3d_skeleton (line 76) | def vis_3d_skeleton(kpt_3d, kpt_3d_vis, kps_lines, filename=None):
  function save_obj (line 115) | def save_obj(v, f, file_name='output.obj'):
  function render_mesh (line 123) | def render_mesh(img, mesh, face, cam_param):

FILE: data/DEX_YCB/DEX_YCB.py
  class DEX_YCB (line 43) | class DEX_YCB(torch.utils.data.Dataset):
    method __init__ (line 44) | def __init__(self, transform, data_split):
    method load_data (line 55) | def load_data(self):
    method __len__ (line 148) | def __len__(self):
    method __getitem__ (line 151) | def __getitem__(self, idx):
    method evaluate (line 212) | def evaluate(self, outs, cur_sample_idx):
    method print_eval_result (line 245) | def print_eval_result(self, test_epoch):

FILE: data/HO3D/HO3D.py
  class HO3D (line 18) | class HO3D(torch.utils.data.Dataset):
    method __init__ (line 19) | def __init__(self, transform, data_split):
    method load_data (line 31) | def load_data(self):
    method __len__ (line 75) | def __len__(self):
    method __getitem__ (line 78) | def __getitem__(self, idx):
    method evaluate (line 129) | def evaluate(self, outs, cur_sample_idx):
    method print_eval_result (line 155) | def print_eval_result(self, test_epoch):

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

FILE: demo/demo_fitting.py
  function parse_args (line 26) | def parse_args():
  function load_camera (line 45) | def load_camera(cam_path, cam_idx='0'):

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

FILE: main/model.py
  class Model (line 14) | class Model(nn.Module):
    method __init__ (line 15) | def __init__(self, backbone, FIT, SET, regressor):
    method forward (line 25) | def forward(self, inputs, targets, meta_info, mode):
    method get_mesh_scale_trans (line 59) | def get_mesh_scale_trans(self, pred_joint_img, pred_joint_cam, init_sc...
  function init_weights (line 107) | def init_weights(m):
  function get_model (line 120) | def get_model(mode):

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

FILE: main/train.py
  function parse_args (line 7) | def parse_args():
  function main (line 24) | def main():