↓ 2 callersFunctionget_affine_transform(
center, scale, rot, output_size,
shift=np.array([0, 0], dtype=np.float32), inv=0
)
stage3_RL/rrl/adaptation_models/transforms.py:54
↓ 2 callersMethodset_transformations(self, s_shift=None, s_scale=None,
a_shift=None, a_scale=None,
stage3_RL/mjrl/mjrl/algos/model_accel/nn_dynamics.py:193
↓ 2 callersFunctiontest_time_adaptation unsupervised keypoint feature adaptation videos: shape (num_videos, timestep, num_frames, 256, 256, 3)
stage3_RL/rrl/adaptation.py:49
↓ 1 callersMethod__init__(self, state_dim, act_dim,
hidden_size=(64,64),
s_shift = None,
stage3_RL/mjrl/mjrl/algos/model_accel/nn_dynamics.py:263
↓ 1 callersMethod_check_branches(self, num_branches, blocks, num_blocks,
num_inchannels, num_channels)
stage3_RL/rrl/adaptation_models/pose_hrnet.py:109
↓ 1 callersMethod_check_branches(self, num_branches, blocks, num_blocks,
num_inchannels, num_channels)
stage2_adapt/lib/models/pose_hrnet.py:119
↓ 1 callersMethod_make_one_branch(self, branch_index, block, num_blocks, num_channels,
stride=1)
stage3_RL/rrl/adaptation_models/pose_hrnet.py:129
↓ 1 callersMethod_make_one_branch(self, branch_index, block, num_blocks, num_channels,
stride=1)
stage2_adapt/lib/models/pose_hrnet.py:139