Functions — github.com/NiteshBharadwaj/part-affinity

hub / github.com/NiteshBharadwaj/part-affinity / functions

Functions74 in github.com/NiteshBharadwaj/part-affinity

⨍ Functions 74 ◇ Types & classes 7

↓ 12 callersFunctionmake_standard_block

(feat_in, feat_out, kernel, stride=1, padding=1, use_bn=True)

src/model/helper.py:15

↓ 7 callersFunctionresize_hm

(heatmap, hm_size)

src/data_process/process_utils.py:77

↓ 3 callersFunctioncheck_annot

(annot)

src/data_process/coco_process_utils.py:35

↓ 3 callersMethodget_item_raw

(self, index, to_resize = True)

src/data_process/coco.py:27

↓ 3 callersFunctioninit

(model)

src/model/helper.py:5

↓ 3 callersFunctionnormalize

(img)

src/data_process/process_utils.py:101

↓ 3 callersMethodparse

(self)

src/opts/base_opts.py:37

↓ 3 callersMethodupdate

(self, val, n=1)

src/training/train_net.py:75

↓ 3 callersFunctionvisualize_heatmap

(img, heat_maps, displayname = 'heatmaps')

src/visualization/visualize.py:27

↓ 3 callersFunctionvisualize_paf

(img, pafs,name='pafs')

src/visualization/visualize.py:65

↓ 2 callersFunctioncompute_resized_coords

Given the index/coordinates of a cell in some input array (e.g. image), provides the new coordinates if that array was resized by making it

src/evaluation/post.py:49

↓ 2 callersFunctioncreate_data_loaders

(opt)

src/data_process/data_loader_provider.py:5

↓ 2 callersFunctioncreate_model

(opt)

src/model/model_provider.py:16

↓ 2 callersFunctiondenormalize

(img)

src/data_process/process_utils.py:108

↓ 2 callersFunctionmake_paf_block_stage1

(inp_feats, output_feats)

src/model/paf_model.py:49

↓ 2 callersFunctionmake_paf_block_stage2

(inp_feats, output_feats)

src/model/paf_model.py:58

↓ 2 callersFunctionparse_criterion

(criterion)

src/model/model_provider.py:7

↓ 2 callersFunctionreshape_paf

(paf)

src/visualization/visualize.py:78

↓ 2 callersFunctionrotate_2d

(pt_2d, rot_rad)

src/data_process/process_utils.py:115

↓ 2 callersFunctionstep

(data_loader, model, criterion_hm, criterion_paf, to_train=False, optimizer=None, viz_output=False)

src/training/train_net.py:7

↓ 2 callersFunctionvalidate_net

(test_loader, model, criterion_hm, criterion_paf, save_dir=None, epoch=0, viz_output=False)

src/training/train_net.py:56

↓ 2 callersFunctionvisualize_masks

(img, ignore_mask)

src/visualization/visualize.py:21

↓ 1 callersFunctionDrawGaussian

(img, pt, sigma)

src/data_process/process_utils.py:35

↓ 1 callersFunctionGaussian

(sigma)

src/data_process/process_utils.py:19

↓ 1 callersFunctionNMS

NonMaximaSuppression: find peaks (local maxima) in a set of grayscale images :param heatmaps: set of grayscale images on which to find local

src/evaluation/post.py:75

↓ 1 callersMethod__init__

(self, backend_outp_feats, n_joints, n_paf, stage1)

src/model/paf_model.py:30

↓ 1 callersFunctionadd_neck

(keypoints)

src/data_process/coco_process_utils.py:86

↓ 1 callersFunctionadjust_learning_rate

(optimizer, epoch, dropLR, LR)

src/training/train_net.py:82

↓ 1 callersFunctionaffine_augment

(img, ignore_mask, keypoints, rot_angle, scale_aug_factor)

src/data_process/process_utils.py:155

↓ 1 callersFunctionappend_result

Build the outputs to be evaluated :param image_id: int, the id of the current image :param person_to_joint_assoc: numpy array of joints associ

src/evaluation/post.py:434

↓ 1 callersFunctionapply_mask

(image, mask, color, alpha=0.5)

src/visualization/visualize.py:6

↓ 1 callersMethodaugment

(self, img, ignore_mask, keypoints, opts)

src/data_process/coco.py:44

↓ 1 callersFunctionclean_annot

(coco, data_path, split)

src/data_process/coco_process_utils.py:128

↓ 1 callersFunctioncolor_augment

(img, ignore_mask, keypoints, color_aug)

src/data_process/process_utils.py:94

↓ 1 callersFunctioncreate_data_sets

(opt)

src/data_process/data_loader_provider.py:23

↓ 1 callersFunctioncreate_optimizer

(opt, model)

src/model/model_provider.py:31

↓ 1 callersFunctiondecode_pose

(img_orig, param, heatmaps, pafs)

src/evaluation/post.py:401

↓ 1 callersFunctioneval_COCO

Evaluate images on Coco test set :param outputs: list of dictionaries, the models' processed outputs :param dataDir: string, path to the MSCOC

src/evaluation/coco.py:7

↓ 1 callersFunctioneval_net

(data_loader, model, opts)

src/evaluation/eval_net.py:15

↓ 1 callersFunctionfind_connected_joints

For every type of limb (eg: forearm, shin, etc.), look for every potential pair of joints (eg: every wrist-elbow combination) and evaluate th

src/evaluation/post.py:155

↓ 1 callersFunctionfind_peaks

Given a (grayscale) image, find local maxima whose value is above a given threshold (param['thre1']) :param img: Input image (2d array) w

src/evaluation/post.py:33

↓ 1 callersFunctionflip

(img, ignore_mask, keypoints, flip_indices)

src/data_process/process_utils.py:56

↓ 1 callersFunctiongen_trans_from_patch_cv

(c_x, c_y, src_width, src_height, dst_width, dst_height, scale, rot)

src/data_process/process_utils.py:125

↓ 1 callersFunctionget_heatmap

(coco, img, keypoints, sigma)

src/data_process/coco_process_utils.py:39

↓ 1 callersFunctionget_ignore_mask

(coco, img, annots)

src/data_process/coco_process_utils.py:110

↓ 1 callersMethodget_imgs_multiscale

(self, index, scales, flip = False)

src/data_process/coco.py:65

↓ 1 callersFunctionget_keypoints

(coco, img, annots)

src/data_process/coco_process_utils.py:101

↓ 1 callersFunctionget_paf

(coco, img, keypoints, sigma_paf, variable_width)

src/data_process/coco_process_utils.py:52

↓ 1 callersFunctiongroup_limbs_of_same_person

Associate limbs belonging to the same person together. :param connected_limbs: See 'return' doc of find_connected_joints() :param joint_l

src/evaluation/post.py:262

↓ 1 callersMethodinit

(self)

src/opts/base_opts.py:8

↓ 1 callersMethodload_image

(self, img_path)

src/data_process/coco.py:60

↓ 1 callersFunctionmain

()

src/eval.py:10

↓ 1 callersFunctionmain

()

src/main.py:10

↓ 1 callersFunctionplot_pose

(img_orig, joint_list, person_to_joint_assoc, bool_fast_plot=True, plot_ear_to_shoulder=False)

src/evaluation/post.py:349

↓ 1 callersMethodreset

(self)

src/training/train_net.py:69

↓ 1 callersFunctionresize

(img, ignore_mask, keypoints, imgSize)

src/data_process/process_utils.py:68

↓ 1 callersFunctionresize_hm_paf

(heatmap, paf, ignore_mask, hm_size)

src/data_process/process_utils.py:84

↓ 1 callersFunctiontrain_net

(train_loader, test_loader, model, criterion_hm, criterion_paf, optimizer, n_epochs, val_interva

src/training/train_net.py:45

↓ 1 callersFunctionvisualize_keypoints

(img, keypoints, body_part_map)

src/visualization/visualize.py:37

↓ 1 callersFunctionvisualize_output

(input_, heatmaps_t, pafs_t, ignore_masks_t, outputs)

src/visualization/visualize.py:98

↓ 1 callersFunctionvisualize_output_single

(img, heatmap_t, paf_t, ignore_mask_t, heatmap_o, paf_o)

src/visualization/visualize.py:84

Method__getitem__

(self, index)

src/data_process/coco.py:54

Method__init__

(self, backend, backend_outp_feats, n_joints, n_paf, n_stages=7)

src/model/paf_model.py:7

src/training/train_net.py:66

Method__init__

(self)

src/opts/base_opts.py:5

Method__init__

(self)

src/opts/viz_opts.py:5

Method__init__

(self, data_path, opt, split='train')

src/data_process/coco.py:13

Method__len__

(self)

src/data_process/coco.py:82

Functionapply_heatmap

(img, heatmap)

src/visualization/visualize.py:15

Methodforward

(self, x)

src/model/paf_model.py:16

Methodforward

(self, x)

src/model/paf_model.py:43

src/opts/viz_opts.py:8