SYMBOL INDEX (428 symbols across 29 files)

FILE: libs/arguments/parse.py
  function read_yaml_file (line 11) | def read_yaml_file(path):
  function parse_args (line 22) | def parse_args():

FILE: libs/common/format.py
  function format_str_submission (line 11) | def format_str_submission(roll, pitch, yaw, x, y, z, score):
  function get_instance_str (line 25) | def get_instance_str(dic):
  function get_pred_str (line 44) | def get_pred_str(record):
  function save_txt_file (line 63) | def save_txt_file(img_path, prediction, params):

FILE: libs/common/img_proc.py
  function transform_preds (line 16) | def transform_preds(coords, center, scale, output_size):
  function get_affine_transform (line 26) | def get_affine_transform(center,
  function affine_transform (line 66) | def affine_transform(pt, t):
  function affine_transform_modified (line 71) | def affine_transform_modified(pts, t):
  function get_3rd_point (line 80) | def get_3rd_point(a, b):
  function get_dir (line 84) | def get_dir(src_point, rot_rad):
  function crop (line 93) | def crop(img, center, scale, output_size, rot=0):
  function simple_crop (line 107) | def simple_crop(input_image, center, crop_size):
  function np_random (line 137) | def np_random():
  function jitter_bbox_with_kpts (line 143) | def jitter_bbox_with_kpts(old_bbox, joints, parameters):
  function jitter_bbox_with_kpts_no_occlu (line 174) | def jitter_bbox_with_kpts_no_occlu(old_bbox, joints, parameters):
  function generate_xy_map (line 193) | def generate_xy_map(bbox, resolution, global_size):
  function crop_single_instance (line 213) | def crop_single_instance(data_numpy, bbox, joints, parameters, pth_trans...
  function get_tensor_from_img (line 251) | def get_tensor_from_img(path,
  function generate_target (line 347) | def generate_target(joints, joints_vis, parameters):
  function resize_bbox (line 411) | def resize_bbox(left, top, right, bottom, target_ar=1.):
  function enlarge_bbox (line 437) | def enlarge_bbox(left, top, right, bottom, enlarge):
  function modify_bbox (line 453) | def modify_bbox(bbox, target_ar, enlarge=1.1):
  function resize_crop (line 461) | def resize_crop(crop_size, target_ar=None):
  function bbox2cs (line 478) | def bbox2cs(bbox):
  function cs2bbox (line 485) | def cs2bbox(center, size):
  function kpts2cs (line 495) | def kpts2cs(keypoints,
  function draw_bboxes (line 542) | def draw_bboxes(img_path, bboxes_dict, save_path=None):
  function imread_rgb (line 556) | def imread_rgb(img_path):
  function save_cropped_patches (line 564) | def save_cropped_patches(img_path,
  function get_max_preds (line 608) | def get_max_preds(batch_heatmaps):
  function soft_arg_max_np (line 639) | def soft_arg_max_np(batch_heatmaps):
  function soft_arg_max (line 678) | def soft_arg_max(batch_heatmaps):
  function appro_cr (line 709) | def appro_cr(coordinates):
  function to_npy (line 722) | def to_npy(tensor):

FILE: libs/common/transformation.py
  function move_to (line 11) | def move_to(points, xyz=np.zeros((1,3))):
  function world_to_camera_frame (line 16) | def world_to_camera_frame(P, R, T):
  function camera_to_world_frame (line 32) | def camera_to_world_frame(P, R, T):
  function compute_similarity_transform (line 48) | def compute_similarity_transform(X, Y, compute_optimal_scale=False):
  function compute_rigid_transform (line 99) | def compute_rigid_transform(X, Y, W=None, verbose=False):
  function procrustes_transform (line 136) | def procrustes_transform(X, Y):
  function pnp_refine (line 143) | def pnp_refine(prediction, observation, intrinsics, dist_coeffs):

FILE: libs/common/utils.py
  function make_dir (line 18) | def make_dir(name):
  function save_checkpoint (line 30) | def save_checkpoint(states, is_best, output_dir, filename='checkpoint.pt...
  function get_model_summary (line 35) | def get_model_summary(model, *input_tensors, item_length=26, verbose=Fal...
  class AverageMeter (line 149) | class AverageMeter(object):
    method __init__ (line 153) | def __init__(self):
    method reset (line 157) | def reset(self):
    method update (line 164) | def update(self, val, n=1, others=None):
    method print_content (line 178) | def print_content(self):

FILE: libs/dataset/KITTI/car_instance.py
  function get_cr_indices (line 99) | def get_cr_indices():
  class KITTI (line 121) | class KITTI(bc.SupervisedDataset):
    method __init__ (line 125) | def __init__(self, cfgs, split, logger, scale=1.0):
    method _get_image_path_list (line 164) | def _get_image_path_list(self):
    method _initialize_unlabeled_data (line 176) | def _initialize_unlabeled_data(self, cfgs):
    method _load_image_list (line 184) | def _load_image_list(self):
    method _check_precomputed_file (line 199) | def _check_precomputed_file(self, path, name):
    method _save_precomputed_file (line 211) | def _save_precomputed_file(self, data_dic, pre_computed_path, name):
    method _prepare_key_points_custom (line 221) | def _prepare_key_points_custom(self, style, interp_params, vis_thresh=...
    method _prepare_key_points (line 264) | def _prepare_key_points(self, cfgs):
    method _save_cropped_instances (line 273) | def _save_cropped_instances(self):
    method _prepare_2d_pose_annot (line 304) | def _prepare_2d_pose_annot(self, threshold=4):
    method _prepare_detection_records (line 348) | def _prepare_detection_records(self, save=False, threshold = 0.1):
    method gather_annotations (line 352) | def gather_annotations(self,
    method read_single_file (line 383) | def read_single_file(self,
    method read_predictions (line 459) | def read_predictions(self, path):
    method _get_data_parameters (line 480) | def _get_data_parameters(self, cfgs):
    method _set_paths (line 533) | def _set_paths(self):
    method project_3d_to_2d (line 557) | def project_3d_to_2d(self, points, K):
    method render_car (line 565) | def render_car(self, ax, K, obj_class, rot_y, locs, dimension, shift):
    method show_statistics (line 575) | def show_statistics(self):
    method augment_pose_vector (line 611) | def augment_pose_vector(self,
    method get_representation (line 646) | def get_representation(self, p2d, p3d, in_rep, out_rep):
    method get_input_output_size (line 688) | def get_input_output_size(self):
    method interpolate (line 705) | def interpolate(self,
    method construct_box_3d (line 730) | def construct_box_3d(self, l, h, w, interp_params):
    method get_cam_cord (line 749) | def get_cam_cord(self, cam_cord, shift, ids, pose_vecs, rot_xz=False):
    method csv_read_annot (line 792) | def csv_read_annot(self, file_path, fieldnames):
    method csv_read_calib (line 831) | def csv_read_calib(self, file_path):
    method load_annotations (line 845) | def load_annotations(self, label_path, calib_path, fieldnames=FIELDNAM...
    method add_visibility (line 855) | def add_visibility(self, joints, img_width=1242, img_height=375):
    method get_inlier_indices (line 870) | def get_inlier_indices(self, p_2d, threshold=0.3):
    method filter_outlier (line 881) | def filter_outlier(self, p_2d, p_3d, threshold=0.3):
    method get_img_size (line 894) | def get_img_size(self, path):
    method get_2d_3d_pair (line 902) | def get_2d_3d_pair(self,
    method show_annot (line 1012) | def show_annot(self,
    method _generate_2d_3d_paris (line 1051) | def _generate_2d_3d_paris(self):
    method generate_pairs (line 1088) | def generate_pairs(self):
    method visualize (line 1128) | def visualize(self, plot_num = 1, save_dir=None):
    method get_collate_fn (line 1138) | def get_collate_fn(self):
    method inference (line 1141) | def inference(self, flags=[True, True]):
    method extract_ss_sample (line 1145) | def extract_ss_sample(self, cnt):
    method prepare_ft_dict (line 1171) | def prepare_ft_dict(self, idx):
    method __getitem__ (line 1217) | def __getitem__(self, idx):
  function prepare_data (line 1321) | def prepare_data(cfgs, logger):
  function get_dataset (line 1332) | def get_dataset(cfgs, logger, split):
  function collate_dict (line 1335) | def collate_dict(dict_list):
  function length_limit (line 1344) | def length_limit(instances, targets, target_weights, meta):
  function my_collate_fn (line 1368) | def my_collate_fn(batch):

FILE: libs/dataset/basic/basic_classes.py
  class SupervisedDataset (line 10) | class SupervisedDataset(torch.utils.data.Dataset):
    method __init__ (line 11) | def __init__(self, cfgs, split, logger=None):
    method generate_pairs (line 18) | def generate_pairs(self, synthetic=True):
    method normalize (line 26) | def normalize(self, statistics=None):
    method unnormalize (line 46) | def unnormalize(self, data, mean, std):
    method __len__ (line 49) | def __len__(self):
    method __getitem__ (line 52) | def __getitem__(self, idx):

FILE: libs/dataset/normalization/operations.py
  function get_statistics_1d (line 10) | def get_statistics_1d(data):
  function normalize_1d (line 21) | def normalize_1d(data, mean, std, individual=False):
  function unnormalize_1d (line 50) | def unnormalize_1d(normalized_data, mean, std):

FILE: libs/logger/logger.py
  function get_dirs (line 16) | def get_dirs(cfgs):
  function get_logger (line 30) | def get_logger(cfgs, head = '%(asctime)-15s %(message)s'):

FILE: libs/loss/function.py
  class JointsMSELoss (line 22) | class JointsMSELoss(nn.Module):
    method __init__ (line 23) | def __init__(self, use_target_weight):
    method forward (line 28) | def forward(self, output, target, target_weight, meta=None):
  function get_comp_dict (line 48) | def get_comp_dict(spec_list = ['None', 'None', 'None'],
  class JointsCompositeLoss (line 61) | class JointsCompositeLoss(nn.Module):
    method __init__ (line 66) | def __init__(self,
    method calc_hm_loss (line 95) | def calc_hm_loss(self, output, target):
    method calc_cross_ratio_loss (line 113) | def calc_cross_ratio_loss(self, pred_coor, target_cr, mask):
    method get_cr_mask (line 138) | def get_cr_mask(self, coordinates, threshold = 0.15):
    method calc_colinear_loss (line 155) | def calc_colinear_loss(self):
    method calc_coor_loss (line 159) | def calc_coor_loss(self, coordinates_pred, coordinates_gt):
    method forward (line 170) | def forward(self, output, target, target_weight=None, meta=None):
  class MSELoss1D (line 204) | class MSELoss1D(nn.Module):
    method __init__ (line 208) | def __init__(self, use_target_weight=False, reduction='mean'):
    method forward (line 213) | def forward(self, output, target, target_weight=None, meta=None):
  class SmoothL1Loss1D (line 217) | class SmoothL1Loss1D(nn.Module):
    method __init__ (line 221) | def __init__(self, use_target_weight=False):
    method forward (line 226) | def forward(self, output, target, target_weight=None, meta=None):
  class DecoupledSL1Loss (line 230) | class DecoupledSL1Loss(nn.Module):
    method __init__ (line 232) | def __init__(self, use_target_weight=None):
    method forward (line 236) | def forward(self, output, target, target_weight=None):
  class JointsOHKMMSELoss (line 242) | class JointsOHKMMSELoss(nn.Module):
    method __init__ (line 244) | def __init__(self, use_target_weight, topk=8):
    method ohkm (line 250) | def ohkm(self, loss):
    method forward (line 262) | def forward(self, output, target, target_weight):
  class WingLoss (line 287) | class WingLoss(nn.Module):
    method __init__ (line 289) | def __init__(self, use_target_weight, width=5, curvature=0.5, image_si...
    method forward (line 296) | def forward(self, output, target, target_weight):

FILE: libs/metric/criterions.py
  function get_distance (line 19) | def get_distance(gt, pred):
  function get_angle_error (line 40) | def get_angle_error(pred, meta_data, cfgs=None):
  function get_PCK (line 57) | def get_PCK(pred, gt):
  function get_distance_src (line 68) | def get_distance_src(output,
  class AngleError (line 145) | class AngleError():
    method __init__ (line 149) | def __init__(self, cfgs, num_joints=None):
    method update (line 156) | def update(self, prediction, meta_data, ground_truth=None, logger=None):
    method report (line 165) | def report(self, logger):
  class JointDistance2DSIP (line 173) | class JointDistance2DSIP():
    method __init__ (line 177) | def __init__(self, cfgs, num_joints=None):
    method update (line 193) | def update(self, prediction, meta_data, ground_truth=None, logger=None):
    method report (line 208) | def report(self, logger):
  function update_statistics (line 223) | def update_statistics(self, update, num_data, name_str):
  function update_rotation_error (line 241) | def update_rotation_error(self,
  function update_joints_3d_error (line 271) | def update_joints_3d_error(self,
  class RotationError3D (line 303) | class RotationError3D():
    method __init__ (line 307) | def __init__(self, cfgs):
    method update (line 317) | def update(self, prediction, ground_truth, meta_data=None, logger=None):
    method report (line 330) | def report(self, logger):
  class JointDistance3D (line 343) | class JointDistance3D():
    method __init__ (line 347) | def __init__(self, cfgs):
    method update (line 360) | def update(self, prediction, ground_truth, meta_data=None, logger=None):
    method report (line 374) | def report(self, logger):
  class RError3D (line 390) | class RError3D():
    method __init__ (line 391) | def __init__(self, cfgs, num_joints):
    method update (line 414) | def update(self, prediction, ground_truth, meta_data=None, logger=None):
    method report (line 433) | def report(self, logger):
  class RTError3D (line 451) | class RTError3D():
    method __init__ (line 452) | def __init__(self, cfgs, num_joints):
    method update (line 486) | def update(self, prediction, ground_truth, meta_data=None, logger=None):
    method report (line 513) | def report(self, logger):
  class Evaluator (line 540) | class Evaluator():
    method __init__ (line 544) | def __init__(self, metrics, cfgs=None, num_joints=9):
    method update (line 553) | def update(self,
    method report (line 570) | def report(self, logger):

FILE: libs/model/FCmodel.py
  class ResidualBlock (line 9) | class ResidualBlock(nn.Module):
    method __init__ (line 10) | def __init__(self,
    method forward (line 33) | def forward(self, x):
  class FCModel (line 45) | class FCModel(nn.Module):
    method __init__ (line 46) | def __init__(self,
    method forward (line 92) | def forward(self, x):
    method get_representation (line 97) | def get_representation(self, x):
  function get_fc_model (line 107) | def get_fc_model(stage_id,
  function get_cascade (line 123) | def get_cascade():

FILE: libs/model/egonet.py
  class EgoNet (line 28) | class EgoNet(nn.Module):
    method __init__ (line 29) | def __init__(self,
    method crop_single_instance (line 68) | def crop_single_instance(self,
    method load_cv2 (line 97) | def load_cv2(self, path, rgb=True):
    method crop_instances (line 105) | def crop_instances(self,
    method add_orientation_arrow (line 157) | def add_orientation_arrow(self, record):
    method write_annot_dict (line 181) | def write_annot_dict(self, annot_dict, records):
    method get_observation_angle_trans (line 203) | def get_observation_angle_trans(self, euler_angles, translations):
    method get_observation_angle_proj (line 219) | def get_observation_angle_proj(self, euler_angles, kpts, K):
    method get_template (line 238) | def get_template(self, prediction, interp_coef=[0.332, 0.667]):
    method kpts_to_euler (line 265) | def kpts_to_euler(self, template, prediction):
    method get_6d_rep (line 279) | def get_6d_rep(self, predictions, ax=None, color="black"):
    method gather_lifting_results (line 297) | def gather_lifting_results(self,
    method plot_one_image (line 341) | def plot_one_image(self,
    method post_process (line 385) | def post_process(self,
    method new_img_dict (line 410) | def new_img_dict(self):
    method get_keypoints (line 424) | def get_keypoints(self,
    method lift_2d_to_3d (line 469) | def lift_2d_to_3d(self, records, cuda=True):
    method forward (line 488) | def forward(self, annot_dict):

FILE: libs/model/heatmapModel/hrnet.py
  function conv3x3 (line 24) | def conv3x3(in_planes, out_planes, stride=1):
  function basicdownsample (line 29) | def basicdownsample(in_planes, out_planes):
  class BasicLinearModule (line 44) | class BasicLinearModule(nn.Module):
    method __init__ (line 45) | def __init__(self, in_channels, out_channels, mid_channels=512):
    method forward (line 53) | def forward(self, x):
  class BasicBlock (line 63) | class BasicBlock(nn.Module):
    method __init__ (line 66) | def __init__(self, inplanes, planes, stride=1, downsample=None):
    method forward (line 76) | def forward(self, x):
  class Bottleneck (line 95) | class Bottleneck(nn.Module):
    method __init__ (line 98) | def __init__(self, inplanes, planes, stride=1, downsample=None):
    method forward (line 113) | def forward(self, x):
  class HighResolutionModule (line 136) | class HighResolutionModule(nn.Module):
    method __init__ (line 137) | def __init__(self, num_branches, blocks, num_blocks, num_inchannels,
    method _check_branches (line 154) | def _check_branches(self, num_branches, blocks, num_blocks,
    method _make_one_branch (line 174) | def _make_one_branch(self, branch_index, block, num_blocks, num_channels,
    method _make_branches (line 212) | def _make_branches(self, num_branches, block, num_blocks, num_channels):
    method _make_fuse_layers (line 222) | def _make_fuse_layers(self):
    method get_num_inchannels (line 279) | def get_num_inchannels(self):
    method forward (line 282) | def forward(self, x):
  class PoseHighResolutionNet (line 309) | class PoseHighResolutionNet(nn.Module):
    method __init__ (line 311) | def __init__(self, cfgs, **kwargs):
    method _make_transition_layer (line 471) | def _make_transition_layer(
    method _make_layer (line 512) | def _make_layer(self, block, planes, blocks, stride=1):
    method _make_stage (line 531) | def _make_stage(self, layer_config, num_inchannels,
    method forward (line 563) | def forward(self, x):
    method init_weights (line 616) | def init_weights(self, pretrained=''):
    method modify_input_channel (line 649) | def modify_input_channel(self, num_channels):
    method load_my_state_dict (line 661) | def load_my_state_dict(self, state_dict):
  function is_freezed (line 669) | def is_freezed(name, freeze_names):
  function get_pose_net (line 675) | def get_pose_net(cfgs, is_train, **kwargs):

FILE: libs/model/heatmapModel/resnet.py
  function conv3x3 (line 22) | def conv3x3(in_planes, out_planes, stride=1):
  class BasicBlock (line 30) | class BasicBlock(nn.Module):
    method __init__ (line 33) | def __init__(self, inplanes, planes, stride=1, downsample=None):
    method forward (line 43) | def forward(self, x):
  class Bottleneck (line 62) | class Bottleneck(nn.Module):
    method __init__ (line 65) | def __init__(self, inplanes, planes, stride=1, downsample=None):
    method forward (line 80) | def forward(self, x):
  class PoseResNet (line 103) | class PoseResNet(nn.Module):
    method __init__ (line 105) | def __init__(self, block, layers, cfg, **kwargs):
    method _make_layer (line 136) | def _make_layer(self, block, planes, blocks, stride=1):
    method _get_deconv_cfg (line 153) | def _get_deconv_cfg(self, deconv_kernel, index):
    method _make_deconv_layer (line 166) | def _make_deconv_layer(self, num_layers, num_filters, num_kernels):
    method forward (line 193) | def forward(self, x):
    method init_weights (line 209) | def init_weights(self, pretrained=''):
  function get_pose_net (line 261) | def get_pose_net(cfg, is_train, **kwargs):

FILE: libs/optimizer/optimizer.py
  function prepare_optim (line 9) | def prepare_optim(model, cfgs):

FILE: libs/trainer/accuracy.py
  function get_distance (line 10) | def get_distance(gt, pred):
  function accuracy_pixel (line 27) | def accuracy_pixel(output,

FILE: libs/trainer/trainer.py
  function train_cascade (line 25) | def train_cascade(train_dataset, valid_dataset, cfgs, logger):
  function evaluate_cascade (line 73) | def evaluate_cascade(cascade,
  function get_loader (line 113) | def get_loader(dataset, cfgs, split, collate_fn=None):
  function train (line 127) | def train(train_dataset,
  function initialize_plot (line 265) | def initialize_plot():
  function update_curve (line 276) | def update_curve(ax, line, x_buffer, y_buffer):
  function logger_print (line 290) | def logger_print(epoch,
  function visualize_lifting_results (line 323) | def visualize_lifting_results(data,
  function evaluate (line 395) | def evaluate(eval_dataset,

FILE: libs/visualization/debug.py
  function draw_circles (line 18) | def draw_circles(ndarr,
  function save_batch_image_with_joints (line 51) | def save_batch_image_with_joints(batch_image,
  function save_batch_heatmaps (line 83) | def save_batch_heatmaps(batch_image,
  function save_debug_images (line 151) | def save_debug_images(epoch,

FILE: libs/visualization/egonet_utils.py
  function plot_2d_objects (line 14) | def plot_2d_objects(img_path, record, color_dict):
  function plot_3d_objects (line 62) | def plot_3d_objects(prediction, target, pose_vecs_gt, record, color):

FILE: libs/visualization/points.py
  function check_points (line 13) | def check_points(points, dimension):
  function set_3d_axe_limits (line 26) | def set_3d_axe_limits(ax, points=None, center=None, radius=None, ratio=1...
  function plot_3d_points (line 48) | def plot_3d_points(ax,
  function plot_lines (line 96) | def plot_lines(ax,
  function plot_mesh (line 132) | def plot_mesh(ax, vertices, faces, color='grey'):
  function plot_3d_coordinate_system (line 149) | def plot_3d_coordinate_system(ax,
  function plot_3d_bbox (line 173) | def plot_3d_bbox(ax,
  function plot_2d_bbox (line 193) | def plot_2d_bbox(ax,
  function plot_comparison_relative (line 214) | def plot_comparison_relative(points_pred, points_gt):
  function plot_scene_3dbox (line 244) | def plot_scene_3dbox(points_pred, points_gt=None, ax=None, color='r'):
  function get_area (line 270) | def get_area(points, indices, preserve_points=False):
  function interpolate (line 284) | def interpolate(start, end, num_interp):
  function get_interpolated_points (line 293) | def get_interpolated_points(points, indices, num_interp):
  function draw_pose_vecs (line 303) | def draw_pose_vecs(ax, pose_vecs=None, color='black'):
  function get_bbox_3d (line 321) | def get_bbox_3d(points, add_center=False, interp_style=""):
  function ray_intersect_triangle (line 364) | def ray_intersect_triangle(p0, p1, triangle):
  function get_visibility (line 414) | def get_visibility(box3d, triangles):

FILE: tools/inference.py
  function filter_detection (line 32) | def filter_detection(detected, thres=0.7):
  function merge (line 46) | def merge(dict_a, dict_b):
  function collate_dict (line 51) | def collate_dict(dict_list):
  function my_collate_fn (line 57) | def my_collate_fn(batch):
  function filter_conf (line 63) | def filter_conf(record, thres=0.0):
  function gather_dict (line 80) | def gather_dict(request,
  function make_output_dir (line 129) | def make_output_dir(cfgs, name):
  function inference (line 136) | def inference(testset, model, results, cfgs):
  function generate_empty_file (line 201) | def generate_empty_file(output_dir, label_dir):
  function main (line 215) | def main():

FILE: tools/inference_legacy.py
  function prepare_models (line 33) | def prepare_models(cfgs, is_cuda=True):
  function modify_bbox (line 59) | def modify_bbox(bbox, target_ar, enlarge=1.1):
  function crop_single_instance (line 67) | def crop_single_instance(img, bbox, resolution, pth_trans=None, xy_dict=...
  function crop_instances (line 91) | def crop_instances(annot_dict,
  function get_keypoints (line 149) | def get_keypoints(instances,
  function kpts_to_euler (line 211) | def kpts_to_euler(template, prediction):
  function get_template (line 225) | def get_template(prediction, interp_coef=[0.332, 0.667]):
  function get_observation_angle_trans (line 252) | def get_observation_angle_trans(euler_angles, translations):
  function get_observation_angle_proj (line 268) | def get_observation_angle_proj(euler_angles, kpts, K):
  function get_6d_rep (line 287) | def get_6d_rep(predictions, ax=None, color="black"):
  function format_str_submission (line 308) | def format_str_submission(roll, pitch, yaw, x, y, z, score):
  function get_instance_str (line 322) | def get_instance_str(dic):
  function get_pred_str (line 341) | def get_pred_str(record):
  function lift_2d_to_3d (line 360) | def lift_2d_to_3d(records, model, stats, template, cuda=True):
  function filter_detection (line 379) | def filter_detection(detected, thres=0.7):
  function add_orientation_arrow (line 393) | def add_orientation_arrow(record):
  function process_batch (line 417) | def process_batch(images,
  function to_npy (line 466) | def to_npy(tensor):
  function refine_with_perfect_size (line 475) | def refine_with_perfect_size(pred,
  function refine_with_predicted_bbox (line 518) | def refine_with_predicted_bbox(pred,
  function draw_pose_vecs (line 549) | def draw_pose_vecs(ax, pose_vecs=None, color='black'):
  function refine_solution (line 567) | def refine_solution(est_3d,
  function gather_lifting_results (line 597) | def gather_lifting_results(record,
  function save_txt_file (line 693) | def save_txt_file(img_path, prediction, params):
  function refine_one_image (line 705) | def refine_one_image(img_path,
  function post_process (line 819) | def post_process(records,
  function merge (line 844) | def merge(dict_a, dict_b):
  function collate_dict (line 849) | def collate_dict(dict_list):
  function my_collate_fn (line 855) | def my_collate_fn(batch):
  function filter_conf (line 861) | def filter_conf(record, thres=0.0):
  function gather_dict (line 878) | def gather_dict(request, references, filter_c=True):
  function inference (line 918) | def inference(testset, model_settings, results, cfgs):
  function generate_empty_file (line 1012) | def generate_empty_file(output_dir, label_dir):
  function main (line 1024) | def main():

FILE: tools/kitti-eval/evaluate_object_3d.cpp
  type DIFFICULTY (line 37) | enum DIFFICULTY{EASY=0, MODERATE=1, HARD=2}
  type METRIC (line 40) | enum METRIC{IMAGE=0, GROUND=1, BOX3D=2}
  type CLASSES (line 48) | enum CLASSES{CAR=0, PEDESTRIAN=1, CYCLIST=2}
  function initGlobals (line 61) | void initGlobals () {
  type tPrData (line 72) | struct tPrData {
    method tPrData (line 78) | tPrData () :
  type tBox (line 83) | struct tBox {
    method tBox (line 90) | tBox (string type, double x1,double y1,double x2,double y2,double alph...
  type tGroundtruth (line 95) | struct tGroundtruth {
    method tGroundtruth (line 102) | tGroundtruth () :
    method tGroundtruth (line 104) | tGroundtruth (tBox box,double truncation,int32_t occlusion) :
    method tGroundtruth (line 106) | tGroundtruth (string type,double x1,double y1,double x2,double y2,doub...
  type tDetection (line 111) | struct tDetection {
    method tDetection (line 117) | tDetection ():
    method tDetection (line 119) | tDetection (tBox box,double thresh) :
    method tDetection (line 121) | tDetection (string type,double x1,double y1,double x2,double y2,double...
  function loadDetections (line 131) | vector<tDetection> loadDetections(string file_name, bool &compute_aos,
  function loadGroundtruth (line 178) | vector<tGroundtruth> loadGroundtruth(string file_name,bool &success) {
  function saveStats (line 204) | void saveStats (const vector<double> &precision, const vector<double> &a...
  function imageBoxOverlap (line 227) | inline double imageBoxOverlap(tBox a, tBox b, int32_t criterion=-1){
  function imageBoxOverlap (line 263) | inline double imageBoxOverlap(tDetection a, tGroundtruth b, int32_t crit...
  function Polygon (line 269) | Polygon toPolygon(const T& g) {
  function groundBoxOverlap (line 294) | inline double groundBoxOverlap(tDetection d, tGroundtruth g, int32_t cri...
  function box3DOverlap (line 317) | inline double box3DOverlap(tDetection d, tGroundtruth g, int32_t criteri...
  function getThresholds (line 346) | vector<double> getThresholds(vector<double> &v, double n_groundtruth){
  function cleanData (line 381) | void cleanData(CLASSES current_class, const vector<tGroundtruth> &gt, co...
  function tPrData (line 455) | tPrData computeStatistics(CLASSES current_class, const vector<tGroundtru...
    method tPrData (line 78) | tPrData () :
  function eval_class (line 619) | bool eval_class (FILE *fp_det, FILE *fp_ori, CLASSES current_class,
  function saveAndPlotPlots (line 705) | void saveAndPlotPlots(string dir_name,string file_name,string obj_type,v...
  function eval (line 768) | bool eval(string result_sha,Mail* mail){
  function main (line 887) | int32_t main (int32_t argc,char *argv[]) {

FILE: tools/kitti-eval/evaluate_object_3d_offline.cpp
  type DIFFICULTY (line 37) | enum DIFFICULTY{EASY=0, MODERATE=1, HARD=2}
  type METRIC (line 40) | enum METRIC{IMAGE=0, GROUND=1, BOX3D=2}
  type CLASSES (line 48) | enum CLASSES{CAR=0, PEDESTRIAN=1, CYCLIST=2}
  function initGlobals (line 61) | void initGlobals () {
  type tPrData (line 72) | struct tPrData {
    method tPrData (line 78) | tPrData () :
  type tBox (line 83) | struct tBox {
    method tBox (line 90) | tBox (string type, double x1,double y1,double x2,double y2,double alph...
  type tGroundtruth (line 95) | struct tGroundtruth {
    method tGroundtruth (line 102) | tGroundtruth () :
    method tGroundtruth (line 104) | tGroundtruth (tBox box,double truncation,int32_t occlusion) :
    method tGroundtruth (line 106) | tGroundtruth (string type,double x1,double y1,double x2,double y2,doub...
  type tDetection (line 111) | struct tDetection {
    method tDetection (line 117) | tDetection ():
    method tDetection (line 119) | tDetection (tBox box,double thresh) :
    method tDetection (line 121) | tDetection (string type,double x1,double y1,double x2,double y2,double...
  function loadDetections (line 131) | vector<tDetection> loadDetections(string file_name, bool &compute_aos,
  function loadGroundtruth (line 178) | vector<tGroundtruth> loadGroundtruth(string file_name,bool &success) {
  function saveStats (line 204) | void saveStats (const vector<double> &precision, const vector<double> &a...
  function imageBoxOverlap (line 227) | inline double imageBoxOverlap(tBox a, tBox b, int32_t criterion=-1){
  function imageBoxOverlap (line 263) | inline double imageBoxOverlap(tDetection a, tGroundtruth b, int32_t crit...
  function Polygon (line 269) | Polygon toPolygon(const T& g) {
  function groundBoxOverlap (line 294) | inline double groundBoxOverlap(tDetection d, tGroundtruth g, int32_t cri...
  function box3DOverlap (line 317) | inline double box3DOverlap(tDetection d, tGroundtruth g, int32_t criteri...
  function getThresholds (line 346) | vector<double> getThresholds(vector<double> &v, double n_groundtruth){
  function cleanData (line 381) | void cleanData(CLASSES current_class, const vector<tGroundtruth> &gt, co...
  function tPrData (line 456) | tPrData computeStatistics(CLASSES current_class, const vector<tGroundtru...
    method tPrData (line 78) | tPrData () :
  function eval_class (line 622) | bool eval_class (FILE *fp_det, FILE *fp_ori, CLASSES current_class,
  function saveAndPlotPlots (line 708) | void saveAndPlotPlots(string dir_name,string file_name,string obj_type,v...
  function getEvalIndices (line 778) | vector<int32_t> getEvalIndices(const string& result_dir) {
  function eval (line 795) | bool eval(string gt_dir, string result_dir, Mail* mail){
  function main (line 917) | int32_t main (int32_t argc,char *argv[]) {

FILE: tools/kitti-eval/evaluate_object_3d_offline_r40.cpp
  type DIFFICULTY (line 37) | enum DIFFICULTY{EASY=0, MODERATE=1, HARD=2}
  type METRIC (line 40) | enum METRIC{IMAGE=0, GROUND=1, BOX3D=2}
  type CLASSES (line 48) | enum CLASSES{CAR=0, PEDESTRIAN=1, CYCLIST=2}
  function initGlobals (line 61) | void initGlobals () {
  type tPrData (line 72) | struct tPrData {
    method tPrData (line 78) | tPrData () :
  type tBox (line 83) | struct tBox {
    method tBox (line 90) | tBox (string type, double x1,double y1,double x2,double y2,double alph...
  type tGroundtruth (line 95) | struct tGroundtruth {
    method tGroundtruth (line 102) | tGroundtruth () :
    method tGroundtruth (line 104) | tGroundtruth (tBox box,double truncation,int32_t occlusion) :
    method tGroundtruth (line 106) | tGroundtruth (string type,double x1,double y1,double x2,double y2,doub...
  type tDetection (line 111) | struct tDetection {
    method tDetection (line 117) | tDetection ():
    method tDetection (line 119) | tDetection (tBox box,double thresh) :
    method tDetection (line 121) | tDetection (string type,double x1,double y1,double x2,double y2,double...
  function loadDetections (line 131) | vector<tDetection> loadDetections(string file_name, bool &compute_aos,
  function loadGroundtruth (line 178) | vector<tGroundtruth> loadGroundtruth(string file_name,bool &success) {
  function saveStats (line 204) | void saveStats (const vector<double> &precision, const vector<double> &a...
  function imageBoxOverlap (line 227) | inline double imageBoxOverlap(tBox a, tBox b, int32_t criterion=-1){
  function imageBoxOverlap (line 263) | inline double imageBoxOverlap(tDetection a, tGroundtruth b, int32_t crit...
  function Polygon (line 269) | Polygon toPolygon(const T& g) {
  function groundBoxOverlap (line 294) | inline double groundBoxOverlap(tDetection d, tGroundtruth g, int32_t cri...
  function box3DOverlap (line 317) | inline double box3DOverlap(tDetection d, tGroundtruth g, int32_t criteri...
  function getThresholds (line 346) | vector<double> getThresholds(vector<double> &v, double n_groundtruth){
  function cleanData (line 381) | void cleanData(CLASSES current_class, const vector<tGroundtruth> &gt, co...
  function tPrData (line 456) | tPrData computeStatistics(CLASSES current_class, const vector<tGroundtru...
    method tPrData (line 78) | tPrData () :
  function eval_class (line 622) | bool eval_class (FILE *fp_det, FILE *fp_ori, CLASSES current_class,
  function saveAndPlotPlots (line 708) | void saveAndPlotPlots(string dir_name,string file_name,string obj_type,v...
  function getEvalIndices (line 778) | vector<int32_t> getEvalIndices(const string& result_dir) {
  function eval (line 795) | bool eval(string gt_dir, string result_dir, Mail* mail){
  function main (line 917) | int32_t main (int32_t argc,char *argv[]) {

FILE: tools/kitti-eval/mail.h
  function class (line 8) | class Mail {

FILE: tools/train_IGRs.py
  function choose_loss_func (line 27) | def choose_loss_func(model_settings, cfgs):
  function train (line 49) | def train(model, model_settings, GPUs, cfgs, logger, final_output_dir):
  function evaluate (line 108) | def evaluate(model, model_settings, GPUs, cfgs, logger, final_output_dir...
  function main (line 127) | def main():

FILE: tools/train_lifting.py
  function main (line 24) | def main():