MCPcopy Create free account

hub / github.com/YanjieZe/H-InDex / functions

Functions701 in github.com/YanjieZe/H-InDex

Method__init__
All inputs are expected in mjrl's format unless specified :param normalized_step_size: Normalized step size (under the KL metric). Tw
stage3_RL/mjrl/mjrl/algos/trpo.py:26
Method__init__
(self, env, plan_horizon, plan_paths=10, kappa=1.0,
stage3_RL/mjrl/mjrl/algos/model_accel/model_learning_mpc.py:6
Method__init__
(self, state_dim, act_dim, learn_reward=False, hidden_size=(64,64),
stage3_RL/mjrl/mjrl/algos/model_accel/nn_dynamics.py:8
Method__init__
(self, state_dim, act_dim, hidden_size=(64,64), s_shift = None, s_scale = No
stage3_RL/mjrl/mjrl/algos/model_accel/nn_dynamics.py:167
Method__init__
(self, learned_model=None, refine=False, kappa=5.0, plan_ho
stage3_RL/mjrl/mjrl/algos/model_accel/model_accel_npg.py:24
Method__init__
(self, sigma=[.1, 2.])
stage2_adapt/lib/utils/transforms.py:133
Method__init__
(self, transforms, p)
stage2_adapt/lib/utils/transforms.py:148
Method__init__
(self, brightness, contrast, saturation, hue)
stage2_adapt/lib/utils/transforms.py:173
Method__init__
(self, p)
stage2_adapt/lib/utils/transforms.py:184
Method__init__
(self, drop_prob=None)
stage2_adapt/lib/dino/vision_transformer_flexible.py:44
Method__init__
(self, dim, num_heads=8, qkv_bias=False, qk_scale=None, attn_drop=0.0, proj_drop=0.0)
stage2_adapt/lib/dino/vision_transformer_flexible.py:72
Method__init__
( self, dim, num_heads, mlp_ratio=4.0, qkv_bias=False, qk_scal
stage2_adapt/lib/dino/vision_transformer_flexible.py:103
Method__init__
(self, img_size=224, patch_size=16, in_chans=3, embed_dim=768)
stage2_adapt/lib/dino/vision_transformer_flexible.py:148
Method__init__
( self, img_size=[224], patch_size=16, in_chans=3, num_classes=0,
stage2_adapt/lib/dino/vision_transformer_flexible.py:166
Method__init__
( self, in_dim, out_dim, use_bn=False, norm_last_layer=True, nlayers=3, hidden_dim=2048, bottleneck_di
stage2_adapt/lib/dino/vision_transformer_flexible.py:355
Method__init__
(self, img_size, pretrain_k=200, tau_img=1.0, tau_mesh=1.0, corr_h=56, corr_w=56,
stage2_adapt/lib/dino/corr.py:10
Method__init__
(self, pretrain_path=None)
stage2_adapt/lib/dino/model.py:27
Method__init__
(self, cfg, root, image_set, is_train, transform=None, task_name="none")
stage2_adapt/lib/dataset/robotic.py:31
Method__init__
(self, cfg, root, image_set, is_train, transform=None)
stage2_adapt/lib/dataset/robotube.py:14
Method__init__
(self, cfg, root, image_set, is_train, transform=None)
stage2_adapt/lib/dataset/JointsDataset.py:31
Method__init__
(self, use_target_weight)
stage2_adapt/lib/core/loss.py:20
Method__init__
(self, use_target_weight, topk=8)
stage2_adapt/lib/core/loss.py:61
Method__init__
(self, cfg)
stage2_adapt/lib/core/loss.py:121
Method__init__
(self)
stage2_adapt/lib/core/function.py:207
Method__init__
(self, cfg)
stage2_adapt/lib/core/perceptual_loss.py:54
Method__init__
(self, inplanes, planes, stride=1, downsample=None)
stage2_adapt/lib/models/pose_double_head_resnet.py:71
Method__init__
(self, d_model, nheads=1, ffn=False)
stage2_adapt/lib/models/pose_double_head_resnet.py:111
Method__init__
(self, num_layers, d_model, nheads=1)
stage2_adapt/lib/models/pose_double_head_resnet.py:145
Method__init__
(self, d_model, height, width)
stage2_adapt/lib/models/pose_double_head_resnet.py:166
Method__init__
(self, block, layers, cfg, **kwargs)
stage2_adapt/lib/models/pose_double_head_resnet.py:192
Method__init__
(self, inplanes, planes, stride=1, downsample=None)
stage2_adapt/lib/models/pose_resnet.py:65
Method__init__
(self, block, layers, cfg, **kwargs)
stage2_adapt/lib/models/pose_resnet.py:105
Method__init__
(self, inplanes, planes, stride=1, downsample=None)
stage2_adapt/lib/models/pose_hrnet.py:63
Method__init__
(self, num_branches, blocks, num_blocks, num_inchannels, num_channels, fuse_method, multi_sca
stage2_adapt/lib/models/pose_hrnet.py:102
Method__init__
(self, cfg, **kwargs)
stage2_adapt/lib/models/pose_hrnet.py:276
Method__init__
(self, in_channels)
stage2_adapt/lib/models/imm_joint_model.py:62
Method__init__
(self, cfg, is_train, is_finetune, freeze_bn)
stage2_adapt/lib/models/imm_joint_model.py:103
Method__len__
(self)
stage2_adapt/lib/dataset/robotic.py:108
Method__len__
(self)
stage2_adapt/lib/dataset/robotube.py:102
Method__len__
(self,)
stage2_adapt/lib/dataset/JointsDataset.py:113
Method_get_db
(self)
stage2_adapt/lib/dataset/JointsDataset.py:62
Method_get_viewer
(self)
stage3_RL/mjrl/mjrl/envs/mujoco_env.py:141
Method_init_weights
(self, m)
stage2_adapt/lib/dino/vision_transformer_flexible.py:222
Method_init_weights
(self, m)
stage2_adapt/lib/dino/vision_transformer_flexible.py:380
Functionaccuracy
Calculate accuracy according to PCK, but uses ground truth heatmap rather than x,y locations First value to be returned is average accura
stage2_adapt/lib/core/evaluate.py:41
Methodaction_dim
(self)
stage3_RL/mjrl/mjrl/utils/gym_env.py:61
Methodaction_space
(self)
stage3_RL/mjrl/mjrl/utils/gym_env.py:73
Functionadd_hooks
(module)
stage2_adapt/lib/utils/utils.py:102
Functionaxis_angle2quat
(axis, angle)
stage3_RL/mj_envs/mj_envs/utils/quatmath.py:31
Functioncompute_advantages
(paths, baseline, gamma, gae_lambda=None, normalize=False)
stage3_RL/mjrl/mjrl/utils/process_samples.py:7
Methodcompute_cycle_loss
(self, img, mask, depth_weight, pointcorr)
stage2_adapt/lib/dino/corr.py:126
Methodcompute_path_rewards
(self, paths)
stage3_RL/mjrl/mjrl/envs/point_mass.py:43
Methodcompute_path_rewards
(self, paths)
stage3_RL/mjrl/mjrl/envs/peg_insertion_sawyer.py:49
Methodcompute_path_rewards
(self, paths)
stage3_RL/mjrl/mjrl/envs/reacher_sawyer.py:47
Functioncompute_returns
(paths, gamma)
stage3_RL/mjrl/mjrl/utils/process_samples.py:3
Functionconcat_tensor_dict_list
(tensor_dict_list)
stage3_RL/mjrl/mjrl/utils/tensor_utils.py:111
Functionconcat_tensor_dict_list_subsample
(tensor_dict_list, f)
stage3_RL/mjrl/mjrl/utils/tensor_utils.py:94
Functioncreate_logger
(cfg, cfg_name, phase='train')
stage2_adapt/lib/utils/utils.py:22
Functioncrop
(img, center, scale, output_size, rot=0)
stage3_RL/rrl/adaptation_models/transforms.py:110
Functioncrop
(img, center, scale, output_size, rot=0)
stage2_adapt/lib/utils/transforms.py:120
Functiondiscount_sum
discount sum a sequence with terminal value
stage3_RL/mjrl/mjrl/algos/model_accel/sampling.py:191
Methoddt
(self)
stage3_RL/mjrl/mjrl/envs/mujoco_env.py:117
Functioneuler2mat
Convert Euler Angles to Rotation Matrix. See rotation.py for notes
stage3_RL/mj_envs/mj_envs/utils/quatmath.py:36
Methodevaluate
(self, cfg, preds, output_dir, *args, **kwargs)
stage2_adapt/lib/dataset/JointsDataset.py:65
Methodevaluate_policy
(self, policy, num_episodes=5, horizon=None,
stage3_RL/rrl/multicam.py:315
Methodevaluate_success
(self, paths)
stage3_RL/mj_envs/mj_envs/hand_manipulation_suite/door_v0.py:128
Methodevaluate_success
(self, paths)
stage3_RL/mj_envs/mj_envs/hand_manipulation_suite/relocate_v0.py:123
Methodevaluate_success
(self, paths)
stage3_RL/mj_envs/mj_envs/hand_manipulation_suite/hammer_v0.py:130
Methodevaluate_success
(self, paths)
stage3_RL/mj_envs/mj_envs/hand_manipulation_suite/pen_v0.py:143
Methodevaluate_success
(self, paths)
stage3_RL/mjrl/mjrl/envs/point_mass.py:67
Methodfit
(self, paths, return_errors=False)
stage3_RL/mjrl/mjrl/baselines/linear_baseline.py:36
Methodfit
(self, paths, return_errors=False)
stage3_RL/mjrl/mjrl/baselines/mlp_baseline.py:61
Methodfit
(self, paths, return_errors=False)
stage3_RL/mjrl/mjrl/baselines/quadratic_baseline.py:44
Methodfit
(self, paths, return_errors=False)
stage3_RL/mjrl/mjrl/baselines/zero_baseline.py:9
Functionflatten_first_axis_tensor_dict
(tensor_dict)
stage3_RL/mjrl/mjrl/utils/tensor_utils.py:47
Functionflatten_tensors
(tensors)
stage3_RL/mjrl/mjrl/utils/tensor_utils.py:6
Functionflip_back
ouput_flipped: numpy.ndarray(batch_size, num_joints, height, width)
stage3_RL/rrl/adaptation_models/transforms.py:12
Functionflip_back
ouput_flipped: numpy.ndarray(batch_size, num_joints, height, width)
stage2_adapt/lib/utils/transforms.py:22
Functionfliplr_joints
flip coords
stage3_RL/rrl/adaptation_models/transforms.py:29
Methodforward
(self, X)
stage3_RL/rrl/utils.py:156
Methodforward
(self, images, pred_images)
stage3_RL/rrl/utils.py:180
Methodforward
(self, input)
stage3_RL/rrl/modules.py:10
Methodforward
(self, x)
stage3_RL/rrl/encoder.py:45
Methodforward
(self, x)
stage3_RL/rrl/encoder.py:98
Methodforward
(self, x)
stage3_RL/rrl/encoder.py:380
Methodforward
(self, x)
stage3_RL/rrl/encoder.py:573
Methodforward
(self, x)
stage3_RL/rrl/encoder.py:588
Methodforward
(self, img, params=None, return_params=False)
stage3_RL/rrl/adaptation_models/transforms.py:141
Methodforward
(self, img, params=None, return_params=False)
stage3_RL/rrl/adaptation_models/transforms.py:166
Methodforward
(self, img, params=None, return_params=False)
stage3_RL/rrl/adaptation_models/transforms.py:177
Methodforward
(self, x)
stage3_RL/rrl/adaptation_models/pose_double_head_resnet.py:39
Methodforward
(self, x)
stage3_RL/rrl/adaptation_models/pose_double_head_resnet.py:76
Methodforward
(self, qr, kv)
stage3_RL/rrl/adaptation_models/pose_double_head_resnet.py:117
Methodforward
(self, qr, kv)
stage3_RL/rrl/adaptation_models/pose_double_head_resnet.py:143
Methodforward
(self, x)
stage3_RL/rrl/adaptation_models/pose_double_head_resnet.py:176
Methodforward
(self, x)
stage3_RL/rrl/adaptation_models/pose_double_head_resnet.py:305
Methodforward
(self, x)
stage3_RL/rrl/adaptation_models/pose_resnet.py:33
Methodforward
(self, x)
stage3_RL/rrl/adaptation_models/pose_resnet.py:70
Methodforward
(self, x)
stage3_RL/rrl/adaptation_models/pose_resnet.py:183
Methodforward
(self, x)
stage3_RL/rrl/adaptation_models/pose_hrnet.py:31
← previousnext →401–500 of 701, ranked by callers