MCPcopy Create free account

hub / github.com/HRNet/HigherHRNet-Human-Pose-Estimation / functions

Functions201 in github.com/HRNet/HigherHRNet-Human-Pose-Estimation

↓ 6 callersMethodload_state_dict
Loads a state_dict created by an earlier call to state_dict(). If ``fp16_optimizer_instance`` was constructed from some ``init_optimi
lib/fp16_utils/fp16_optimizer.py:296
↓ 6 callersMethodupdate
(self, val, n=1)
lib/utils/utils.py:234
↓ 5 callersMethodstate_dict
Returns a dict containing the current state of this :class:`FP16_Optimizer` instance. This dict contains attributes of :class:`FP16_O
lib/fp16_utils/fp16_optimizer.py:276
↓ 4 callersMethod__init__
(self, loss_type)
lib/core/loss.py:42
↓ 4 callersFunctionget_affine_transform
(center, scale, rot, output_size,
lib/utils/transforms.py:59
↓ 3 callersMethod__init__
(self, inplanes, planes, stride=1, downsample=None)
lib/models/pose_higher_hrnet.py:36
↓ 3 callersFunction_get_loss_info
(loss_meters, loss_name)
lib/core/trainer.py:130
↓ 3 callersMethod_make_stage
(self, layer_config, num_inchannels, multi_scale_output=True)
lib/models/pose_higher_hrnet.py:443
↓ 3 callersMethod_make_transition_layer
( self, num_channels_pre_layer, num_channels_cur_layer)
lib/models/pose_higher_hrnet.py:391
↓ 3 callersMethodbackward
(self, loss)
lib/fp16_utils/loss_scaler.py:76
↓ 3 callersFunctionmake_input
(t, requires_grad=False, need_cuda=True)
lib/core/loss.py:21
↓ 3 callersMethodmaybe_print
(self, msg)
lib/fp16_utils/fp16_optimizer.py:189
↓ 3 callersFunctionsave_batch_maps
( batch_image, batch_maps, batch_mask, file_name, map_type='heatmap',
lib/utils/vis.py:157
↓ 3 callersMethodstep
If no closure is supplied, :attr:`step` should be called after ``fp16_optimizer_obj.backward(loss)``. :attr:`step` updates th
lib/fp16_utils/fp16_optimizer.py:336
↓ 3 callersFunctionupdate_config
(cfg, args)
lib/config/default.py:156
↓ 2 callersMethod_get_affine_matrix
(self, center, scale, res, rot=0)
lib/dataset/transforms/transforms.py:98
↓ 2 callersMethod_get_image_path
(self, file_name)
lib/dataset/COCODataset.py:88
↓ 2 callersMethod_get_image_path
(self, file_name)
lib/dataset/CrowdPoseDataset.py:77
↓ 2 callersMethod_master_params_to_model_params
(self)
lib/fp16_utils/fp16_optimizer.py:238
↓ 2 callersFunction_print_name_value
(logger, name_value, full_arch_name)
tools/valid.py:67
↓ 2 callersMethod_update_scale
(self, has_overflow=False)
lib/fp16_utils/fp16_optimizer.py:235
↓ 2 callersFunctionconv3x3
3x3 convolution with padding
lib/models/pose_higher_hrnet.py:27
↓ 2 callersFunctioncreate_logger
(cfg, cfg_name, phase='train')
lib/utils/utils.py:38
↓ 2 callersMethodevaluate
Perform evaluation on COCO keypoint task :param cfg: cfg dictionary :param preds: prediction :param output_dir: outpu
lib/dataset/COCODataset.py:159
↓ 2 callersFunctionget_3rd_point
(a, b)
lib/utils/transforms.py:102
↓ 2 callersFunctionget_multi_scale_size
(image, input_size, current_scale, min_scale)
lib/utils/transforms.py:155
↓ 2 callersFunctionnetwork_to_half
Convert model to half precision in a batchnorm-safe way.
lib/fp16_utils/fp16util.py:87
↓ 2 callersFunctionsetup_logger
(final_output_dir, rank, phase)
lib/utils/utils.py:22
↓ 1 callersFunctionBN_convert_float
Designed to work with network_to_half. BatchNorm layers need parameters in single precision. Find all layers and convert them back to flo
lib/fp16_utils/fp16util.py:71
↓ 1 callersMethod__init__
(self)
lib/fp16_utils/fp16util.py:47
↓ 1 callersMethod__len__
(self)
lib/dataset/COCODataset.py:132
↓ 1 callersMethod__len__
(self)
lib/dataset/CrowdPoseDataset.py:120
↓ 1 callersMethod_affine_joints
(self, joints, mat)
lib/dataset/transforms/transforms.py:124
↓ 1 callersMethod_check_branches
(self, num_branches, blocks, num_blocks, num_inchannels, num_channels)
lib/models/pose_higher_hrnet.py:124
↓ 1 callersMethod_check_overflow
(self)
lib/fp16_utils/fp16_optimizer.py:225
↓ 1 callersMethod_coco_keypoint_results_one_category_kernel
(self, data_pack)
lib/dataset/COCODataset.py:254
↓ 1 callersMethod_coco_keypoint_results_one_category_kernel
(self, data_pack)
lib/dataset/CrowdPoseDataset.py:240
↓ 1 callersMethod_do_python_keypoint_eval
(self, res_file, res_folder)
lib/dataset/COCODataset.py:295
↓ 1 callersMethod_do_python_keypoint_eval
(self, res_file, res_folder)
lib/dataset/CrowdPoseDataset.py:281
↓ 1 callersMethod_downscale_master
(self)
lib/fp16_utils/fp16_optimizer.py:248
↓ 1 callersMethod_forward_check
(self, outputs, heatmaps, masks, joints)
lib/core/loss.py:274
↓ 1 callersMethod_get_anno_file_name
(self)
lib/dataset/COCODataset.py:68
↓ 1 callersMethod_get_anno_file_name
(self)
lib/dataset/CrowdPoseDataset.py:67
↓ 1 callersMethod_get_deconv_cfg
(self, deconv_kernel)
lib/models/pose_higher_hrnet.py:378
↓ 1 callersMethod_has_inf_or_nan
(self, x)
lib/fp16_utils/loss_scaler.py:127
↓ 1 callersMethod_init_check
(self, heatmap_generator, joints_generator)
lib/dataset/CrowdPoseKeypoints.py:131
↓ 1 callersMethod_init_check
(self, heatmap_generator, joints_generator)
lib/dataset/COCOKeypoints.py:143
↓ 1 callersMethod_init_check
(self, cfg)
lib/core/loss.py:247
↓ 1 callersMethod_make_branches
(self, num_branches, block, num_blocks, num_channels)
lib/models/pose_higher_hrnet.py:168
↓ 1 callersMethod_make_deconv_layers
(self, cfg, input_channels)
lib/models/pose_higher_hrnet.py:341
↓ 1 callersMethod_make_final_layers
(self, cfg, input_channels)
lib/models/pose_higher_hrnet.py:311
↓ 1 callersMethod_make_fuse_layers
(self)
lib/models/pose_higher_hrnet.py:177
↓ 1 callersMethod_make_layer
(self, block, planes, blocks, stride=1)
lib/models/pose_higher_hrnet.py:426
↓ 1 callersMethod_make_one_branch
(self, branch_index, block, num_blocks, num_channels, stride=1)
lib/models/pose_higher_hrnet.py:144
↓ 1 callersMethod_model_grads_to_master_grads
(self)
lib/fp16_utils/fp16_optimizer.py:244
↓ 1 callersMethod_step_with_closure
(self, closure)
lib/fp16_utils/fp16_optimizer.py:386
↓ 1 callersMethod_write_coco_keypoint_results
(self, keypoints, res_file)
lib/dataset/COCODataset.py:226
↓ 1 callersMethod_write_coco_keypoint_results
(self, keypoints, res_file)
lib/dataset/CrowdPoseDataset.py:212
↓ 1 callersFunctionadd_joints
(image, joints, color, dataset='COCO')
lib/utils/vis.py:21
↓ 1 callersFunctionadd_path
(path)
tools/_init_paths.py:15
↓ 1 callersMethodadjust
(self, ans, det)
lib/core/group.py:177
↓ 1 callersFunctionaffine_transform
(pt, t)
lib/utils/transforms.py:96
↓ 1 callersFunctionaggregate_results
( cfg, scale_factor, final_heatmaps, tags_list, heatmaps, tags )
lib/core/inference.py:176
↓ 1 callersFunctionbuild_dataset
(cfg, is_train)
lib/dataset/build.py:24
↓ 1 callersFunctionbuild_transforms
(cfg, is_train=True)
lib/dataset/transforms/build.py:31
↓ 1 callersFunctioncheck_config
(cfg)
lib/config/default.py:199
↓ 1 callersFunctiondo_train
(cfg, model, data_loader, loss_factory, optimizer, epoch, output_dir, tb_log_dir, writer_dict, fp
lib/core/trainer.py:20
↓ 1 callersMethodevaluate
Perform evaluation on COCO keypoint task :param cfg: cfg dictionary :param preds: prediction :param output_dir: outpu
lib/dataset/CrowdPoseDataset.py:147
↓ 1 callersFunctionget_dir
(src_point, rot_rad)
lib/utils/transforms.py:107
↓ 1 callersFunctionget_final_preds
(grouped_joints, center, scale, heatmap_size)
lib/utils/transforms.py:195
↓ 1 callersMethodget_gaussian_kernel
(self, sigma)
lib/dataset/target_generators/target_generators.py:58
↓ 1 callersMethodget_joints
(self, anno)
lib/dataset/CrowdPoseKeypoints.py:94
↓ 1 callersMethodget_joints
(self, anno)
lib/dataset/COCOKeypoints.py:95
↓ 1 callersMethodget_mask
(self, anno, idx)
lib/dataset/CrowdPoseKeypoints.py:123
↓ 1 callersMethodget_mask
(self, anno, idx)
lib/dataset/COCOKeypoints.py:124
↓ 1 callersFunctionget_model_summary
:param model: :param input_tensors: :param item_length: :return:
lib/utils/utils.py:105
↓ 1 callersFunctionget_multi_stage_outputs
( cfg, model, image, with_flip=False, project2image=False, size_projected=None )
lib/core/inference.py:75
↓ 1 callersMethodget_num_inchannels
(self)
lib/models/pose_higher_hrnet.py:222
↓ 1 callersFunctionget_optimizer
(cfg, model)
lib/utils/utils.py:75
↓ 1 callersMethodhas_overflow
(self, params)
lib/fp16_utils/loss_scaler.py:59
↓ 1 callersMethodinit_weights
(self, pretrained='', verbose=True)
lib/models/pose_higher_hrnet.py:522
↓ 1 callersFunctionlink
(a, b, color)
lib/utils/vis.py:25
↓ 1 callersFunctionmain
()
tools/dist_train.py:80
↓ 1 callersFunctionmain
()
tools/crowdpose_concat_train_val.py:29
↓ 1 callersFunctionmain
()
tools/valid.py:87
↓ 1 callersFunctionmain_worker
( gpu, ngpus_per_node, args, final_output_dir, tb_log_dir )
tools/dist_train.py:127
↓ 1 callersFunctionmake_dataloader
(cfg, is_train=True, distributed=False)
lib/dataset/build.py:60
↓ 1 callersFunctionmake_heatmaps
(image, heatmaps)
lib/utils/vis.py:61
↓ 1 callersFunctionmake_tagmaps
(image, tagmaps)
lib/utils/vis.py:87
↓ 1 callersFunctionmake_test_dataloader
(cfg)
lib/dataset/build.py:91
↓ 1 callersFunctionmaster_params_to_model_params
Copy master parameters to model parameters. Args: model_params: List of model parameters created by :func:`prep_param_lists`.
lib/fp16_utils/fp16util.py:160
↓ 1 callersMethodmatch
(self, tag_k, loc_k, val_k)
lib/core/group.py:136
↓ 1 callersFunctionmatch_by_tag
(inp, params)
lib/core/group.py:25
↓ 1 callersFunctionmodel_grads_to_master_grads
Copy model gradients to master gradients. Args: model_params: List of model parameters created by :func:`prep_param_lists`.
lib/fp16_utils/fp16util.py:139
↓ 1 callersMethodnms
(self, det)
lib/core/group.py:130
↓ 1 callersMethodparse
(self, det, tag, adjust=True, refine=True)
lib/core/group.py:262
↓ 1 callersFunctionparse_args
()
tools/dist_train.py:45
↓ 1 callersFunctionparse_args
()
tools/crowdpose_concat_train_val.py:16
↓ 1 callersFunctionparse_args
()
tools/valid.py:48
↓ 1 callersMethodprocessKeypoints
(self, keypoints)
lib/dataset/COCODataset.py:145
next →1–100 of 201, ranked by callers