SYMBOL INDEX (96 symbols across 9 files)

FILE: src/hist_cub.py
  function cub (line 24) | def cub(x, a, b, c):
  function subsample (line 33) | def subsample(a, p=0.0005, seed=0):
  function read_cos_opt (line 42) | def read_cos_opt(path, fname="cos_hist.npy"):
  function plot_hist (line 53) | def plot_hist(cos_opt_dir, hist_smpl_fpath, params, out_dir, bins=10, xy...
  function kldiv (line 121) | def kldiv(p_hist, q_hist):
  function calc_histogram (line 127) | def calc_histogram(x, bins=10, range=(0, 1)):
  function step (line 132) | def step(params, angles_opt, p_hist, bone_idx=None):
  function optimize (line 154) | def optimize(cos_opt_dir, hist_smpl_fpath, bone_idx=None):
  function main (line 201) | def main():

FILE: src/losses.py
  class MSE (line 9) | class MSE(nn.Module):
    method __init__ (line 10) | def __init__(self, ignore=None):
    method forward (line 16) | def forward(self, y_pred, y_data):
  class Parallel (line 25) | class Parallel(nn.Module):
    method __init__ (line 26) | def __init__(self, skeleton, ignore=None, ground_parallel=None):
    method forward (line 40) | def forward(self, y_pred3d, y_data, z, spine_j, writer=None, global_st...
  class MimickedSelfContactLoss (line 155) | class MimickedSelfContactLoss(nn.Module):
    method __init__ (line 156) | def __init__(self, geodesics_mask):
    method forward (line 164) | def forward(

FILE: src/pose.py
  function parse_args (line 50) | def parse_args():
  function freeze_layers (line 223) | def freeze_layers(model):
  function project_and_normalize_to_spin (line 239) | def project_and_normalize_to_spin(vertices_3d, camera):
  function project_and_normalize_to_spin_legs (line 253) | def project_and_normalize_to_spin_legs(vertices_3d, A, camera):
  function rotation_matrix_to_angle_axis (line 286) | def rotation_matrix_to_angle_axis(rotmat):
  function get_smpl_output (line 303) | def get_smpl_output(smpl, rotmat, betas, use_betas=True, zero_hands=False):
  function get_predictions (line 331) | def get_predictions(model_hmr, smpl, input_img, use_betas=True, zero_han...
  function get_pred_and_data (line 348) | def get_pred_and_data(
  function normalize_keypoints_to_spin (line 379) | def normalize_keypoints_to_spin(keypoints_2d, img_size):
  function unnormalize_keypoints_from_spin (line 397) | def unnormalize_keypoints_from_spin(keypoints_2d, shift, scale, ax2):
  function get_vertices_in_heatmap (line 405) | def get_vertices_in_heatmap(contact_heatmap):
  function get_contact_heatmap (line 424) | def get_contact_heatmap(model_contact, img_path, thresh=0.5):
  function discretize (line 445) | def discretize(parametrization, n_bins=100):
  function get_mapping_from_params_to_verts (line 453) | def get_mapping_from_params_to_verts(verts, params):
  function find_contacts (line 461) | def find_contacts(y_data_conts, keypoints_2d, bone_to_params, thresh=12,...
  function optimize (line 529) | def optimize(
  function optimize_ft (line 772) | def optimize_ft(
  function create_bone (line 1042) | def create_bone(i, j, keypoints_2d):
  function is_parallel_to_plane (line 1051) | def is_parallel_to_plane(bone, thresh=21):
  function is_close_to_plane (line 1055) | def is_close_to_plane(bone, plane, thresh):
  function get_selector (line 1061) | def get_selector():
  function calc_cos (line 1070) | def calc_cos(joints_2d, joints_3d):
  function get_natural (line 1087) | def get_natural(keypoints_2d, vertices, right_foot_inds, left_foot_inds,...
  function get_cos (line 1165) | def get_cos(keypoints_3d_pred, use_angle_transf, loss_parallel):
  function save_mesh_with_winding_numbers (line 1202) | def save_mesh_with_winding_numbers(sc_module, vertices, smpl, save_path):
  function get_contacts (line 1223) | def get_contacts(
  function save_all (line 1264) | def save_all(
  function spin_step (line 1348) | def spin_step(
  function eft_step (line 1403) | def eft_step(
  function dc_step (line 1496) | def dc_step(
  function us_step (line 1577) | def us_step(
  function main (line 1652) | def main():

FILE: src/pose_estimation.py
  function transform (line 81) | def transform(img):
  function get_affine_transform (line 89) | def get_affine_transform(
  function get_3rd_point (line 127) | def get_3rd_point(a, b):
  function get_dir (line 132) | def get_dir(src_point, rot_rad):
  function process_image (line 142) | def process_image(path, input_img_size, pixel_std=200):
  function get_final_preds (line 166) | def get_final_preds(batch_heatmaps, center, scale, post_process=False):
  function transform_preds (line 199) | def transform_preds(coords, center, scale, output_size):
  function affine_transform (line 207) | def affine_transform(pt, t):
  function get_max_preds (line 213) | def get_max_preds(batch_heatmaps):
  function infer_single_image (line 245) | def infer_single_image(model, img_path, input_img_size=(288, 384), retur...

FILE: src/renderer.py
  class Renderer (line 8) | class Renderer:
    method __init__ (line 14) | def __init__(self, focal_length=5000, img_res=224, faces=None):
    method visualize_tb (line 22) | def visualize_tb(self, vertices, camera_translation, images):
    method __call__ (line 40) | def __call__(self, vertices, camera_translation, image):
  function overlay_mesh (line 84) | def overlay_mesh(

FILE: src/spin/hmr.py
  function rot6d_to_rotmat (line 9) | def rot6d_to_rotmat(x):
  class Bottleneck (line 31) | class Bottleneck(nn.Module):
    method __init__ (line 38) | def __init__(self, inplanes, planes, stride=1, downsample=None):
    method forward (line 52) | def forward(self, x):
  class HMR (line 75) | class HMR(nn.Module):
    method __init__ (line 78) | def __init__(self, block, layers, smpl_mean_params):
    method _make_layer (line 123) | def _make_layer(self, block, planes, blocks, stride=1):
    method forward (line 145) | def forward(self, x, init_pose=None, init_shape=None, init_cam=None, n...
  function hmr (line 187) | def hmr(smpl_mean_params, pretrained=True, **kwargs):

FILE: src/spin/smpl.py
  class SMPL (line 37) | class SMPL(_SMPL):
    method __init__ (line 93) | def __init__(self, *args, **kwargs):
    method forward (line 103) | def forward(self, *args, **kwargs):
  class SMPLX (line 124) | class SMPLX(_SMPLX):
    method __init__ (line 176) | def __init__(self, *args, **kwargs):
    method forward (line 182) | def forward(self, *args, **kwargs):

FILE: src/spin/utils.py
  function get_transform (line 12) | def get_transform(center, scale, res, rot=0):
  function transform (line 40) | def transform(pt, center, scale, res, invert=0, rot=0):
  function crop (line 51) | def crop(img, center, scale, res, rot=0):
  function bbox_from_openpose (line 89) | def bbox_from_openpose(openpose_file, rescale=1.2, detection_thresh=0.2):
  function bbox_from_json (line 105) | def bbox_from_json(bbox_file):
  function process_image (line 119) | def process_image(img_file, bbox_file=None, openpose_file=None, input_re...

FILE: src/utils.py
  function load_json (line 21) | def load_json(path):
  function save_json (line 26) | def save_json(o, path):
  function load_pkl (line 31) | def load_pkl(path):
  function save_pkl (line 36) | def save_pkl(o, path):
  function plot_3D (line 41) | def plot_3D(joints, vertices, faces):
  function draw_keypoints (line 71) | def draw_keypoints(
  function save_results_image (line 120) | def save_results_image(
  function save_3d_model_on_img (line 347) | def save_3d_model_on_img(
  function save_mesh_with_colors (line 388) | def save_mesh_with_colors(vertices, faces, save_path, mask=None, inds=No...
  function save_pose_params (line 406) | def save_pose_params(rotmat, camera, betas, vertices, smpl, contact, sav...