MCPcopy Create free account

hub / github.com/MediaBrain-SJTU/MemoNet / functions

Functions249 in github.com/MediaBrain-SJTU/MemoNet

Methodfit
(self)
trainer/test_final_trajectory.py:60
Methodforward
r"""Warps an image or tensor from source into reference frame. Args: patch_src (torch.Tensor): The image or tensor to warp.
ETH/data/homography_warper.py:390
Methodforward
Inputs: batch1: actor_dim, 0th -> 0th actor is ego batch2: scene_dim, all the other agents under ith ego agent - h_states: Tensor of shape (b
ETH/models/layer_utils.py:110
Methodforward
(self, x)
ETH/models/layer_utils.py:178
Methodforward
X: b, T, 2 return: b, F
ETH/models/layer_utils.py:212
Methodforward
>>> Input: x_true: N, T_p, 2 x_hat: N, T_p, 2 f: N, f (128+16) >>> Output: x_hat_aft
ETH/models/layer_utils.py:263
Methodforward
(self, past, abs_past, end_pose)
ETH/models/model_test_trajectory.py:157
Methodforward
(self, past, abs_past, end_pose)
ETH/models/model_train_trajectory.py:341
Methodforward
Inputs: - h_states: Tensor of shape (num_layers, batch, h_dim) - seq_start_end: A list of tuples which delimit sequences with
models/layer_utils.py:121
Methodforward
(self, x)
models/layer_utils.py:197
Methodforward
X: b, T, 2 return: b, F
models/layer_utils.py:231
Methodforward
>>> Input: x_true: N, T_p, 2 x_hat: N, T_p, 2 f: N, f (128+16) >>> Output: x_hat_aft
models/layer_utils.py:284
Methodforward
(self, past, abs_past, seq_start_end, end_pose, future)
models/model_encdec_trajectory_social.py:107
Methodforward
(self, past, abs_past, seq_start_end, end_pose, future)
models/model_selector_finetune_social.py:69
Methodforward
(self, past, abs_past, seq_start_end, end_pose, future)
models/model_selector_warm_up_social.py:68
Methodforward
(self, past, abs_past, seq_start_end, end_pose, future)
models/model_encdec_social.py:50
Methodforward
(self, past, abs_past, seq_start_end, end_pose, future)
models/model_AIO.py:358
Methodforward
(self, past, abs_past, seq_start_end, end_pose)
models/model_test_trajectory_res.py:174
Functiongenerate_pooled_data
(b_size, t_tresh, d_tresh, train=True, scene=None, verbose=True)
sddloader.py:165
Methodget_cropped_maps
(self, world_pts, patch_size, rotation=None, device='cpu')
ETH/data/map.py:21
Methodget_cropped_maps
Returns rotated patches of the map around the transformed scene points. ___________________ | | | |
ETH/data/map.py:156
Functionget_flat_grad_from
(inputs, grad_grad=False)
ETH/utils/torch.py:119
Functionget_flat_params_from
(models)
ETH/utils/torch.py:98
Methodget_last_epoch
(self)
ETH/utils/config.py:37
Methodget_memory_index
(self, state_past, memory_past)
ETH/models/model_train_trajectory.py:77
Methodget_memory_index_batch
(self, state_past, memory_past)
ETH/models/model_train_trajectory.py:86
Methodget_memory_index_batch
(self, state_past, memory_past)
models/model_selector_warm_up_social.py:58
Functionget_noise
(shape, noise_type)
ETH/models/layer_utils.py:29
Functionget_noise
(shape, noise_type)
models/layer_utils.py:29
Functionget_scheduler
(optimizer, policy, nepoch_fix=None, nepoch=None, decay_step=None, decay_gamma=0.1)
ETH/utils/torch.py:171
Methodget_state_encoding
(self, past, abs_past, seq_start_end, end_pose, future)
models/model_encdec_trajectory_social.py:62
Methodget_state_encoding
(self, past, abs_past, seq_start_end, end_pose, future)
models/model_test_trajectory_res.py:52
Functionget_timestring
()
ETH/utils/utils.py:78
Functioninitialize_weights
(modules)
ETH/utils/utils.py:184
Functionisfile
(pathname)
ETH/utils/utils.py:122
Methodk_means
return clustering ncluster of x. Args: x (Tensor): B, K, 2 ncluster (int, optional): Number of clusters. Defaults to 20. iter (int, option
ETH/models/model_train_trajectory.py:53
Functionlambda_rule
(epoch)
ETH/utils/torch.py:173
Functionload_list_from_folder
load a list of files or folders from a system path parameters: folder_path: root to search ext_filter: a string to r
ETH/utils/utils.py:262
Functionload_txt_file
load data or string from text file
ETH/utils/utils.py:250
Functionmkdir_if_missing
(input_path)
ETH/utils/utils.py:145
Functionnaive_social
(p1_key, p2_key, all_data_dict)
sddloader.py:13
Methodprint_model_param
(self, model)
trainer/test_final_trajectory.py:53
Functionrotation_2d_torch
(x, theta, origin=None)
ETH/utils/torch.py:15
Methodset_epoch
(self, epoch)
ETH/utils/torch.py:200
Functionset_flat_params_to
(model, flat_params)
ETH/utils/torch.py:110
Functionset_optimizer_lr
(optimizer, lr)
ETH/utils/torch.py:160
Methodupdate
(self, val, n=1)
ETH/utils/utils.py:28
Methodval
(self)
ETH/utils/torch.py:203
Methodvisualize_data
(self, data)
ETH/data/map.py:194
← previous201–249 of 249, ranked by callers