MCPcopy Create free account

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

Functions249 in github.com/MediaBrain-SJTU/MemoNet

↓ 19 callersMethodrepeat
Inputs: -tensor: 2D tensor of any shape -num_reps: Number of times to repeat each row Outpus: -repeat_tensor:
models/layer_utils.py:79
↓ 6 callersMethodget
(self, name, default=None)
ETH/utils/config.py:63
↓ 5 callersFunctionisstring
(string_test)
ETH/utils/utils.py:60
↓ 5 callersFunctionmake_mlp
(dim_list, activation='relu', batch_norm=True, dropout=0)
ETH/models/layer_utils.py:15
↓ 5 callersFunctionmake_mlp
(dim_list, activation='relu', batch_norm=True, dropout=0)
models/layer_utils.py:15
↓ 5 callersFunctionprint_log
print a string to a log file parameters: print_str: a string to print log: a opened file to save the log same_line:
ETH/utils/utils.py:197
↓ 4 callersMethodis_epoch_end
(self)
ETH/data/dataloader.py:65
↓ 4 callersFunctionis_path_exists
(pathname)
ETH/utils/utils.py:112
↓ 4 callersFunctionis_path_valid
(pathname)
ETH/utils/utils.py:89
↓ 4 callersMethodk_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_test_trajectory.py:60
↓ 3 callersMethod__init__
(self, input_dim, output_dim, hidden_size=(1024, 512), activation='relu', discrim=False, dropout=-1)
ETH/models/layer_utils.py:160
↓ 3 callersMethod__init__
(self, input_dim, output_dim, hidden_size=(1024, 512), activation='relu', discrim=False, dropout=-1)
models/layer_utils.py:179
↓ 3 callersMethodfit
(self)
trainer/trainer_AIO.py:70
↓ 3 callersMethodget_attention
(self, past, abs_past, end_pose, gt_destination)
ETH/models/model_train_trajectory.py:301
↓ 3 callersMethodget_memory_index
(self, state_past, memory_past)
ETH/models/model_test_trajectory.py:85
↓ 3 callersMethodget_memory_index
(self, state_past, memory_past)
models/model_AIO.py:163
↓ 3 callersMethodget_memory_index_batch
(self, state_past, memory_past)
ETH/models/model_test_trajectory.py:94
↓ 3 callersMethodget_state_encoding
(self, past, abs_past, seq_start_end, end_pose, future)
models/model_AIO.py:93
↓ 3 callersMethodreconstruct_destination
(self, past, abs_past, end_pose, future)
ETH/models/model_train_trajectory.py:166
↓ 3 callersMethodrotate_traj
(self, past, future, past_abs)
ETH/trainer/train_trajectory_AIO.py:70
↓ 3 callersFunctionsafe_path
convert path to a valid OS format, e.g., empty string '' to '.', remove redundant '/' at the end from 'aa/' to 'aa' parameters: input_pa
ETH/utils/utils.py:163
↓ 3 callersMethodto_map_points
(self, scene_pts)
ETH/data/map.py:179
↓ 2 callersMethodTotalFrame
(self)
ETH/data/preprocessor.py:63
↓ 2 callersMethod__init__
(self, cfg)
ETH/models/model_train_trajectory.py:99
↓ 2 callersMethodattention_loss
(self, sim, distance)
ETH/trainer/train_trajectory_AIO.py:103
↓ 2 callersFunctioncollect_data
(set_name, dataset_type = 'image', batch_size=512, time_thresh=48, dist_tresh=100, scene=None, verbose=True, r
sddloader.py:94
↓ 2 callersMethoddecode_state_into_intention
(self, past, norm_past_state, abs_past_state_social, norm_fut_state)
models/model_AIO.py:103
↓ 2 callersMethodedge2node
(self, x, rel_rec, rel_send)
ETH/models/layer_utils.py:76
↓ 2 callersMethodedge2node
(self, x, rel_rec, rel_send)
models/layer_utils.py:92
↓ 2 callersFunctionencode_onehot
(labels)
ETH/models/layer_utils.py:7
↓ 2 callersFunctionencode_onehot
(labels)
models/layer_utils.py:7
↓ 2 callersMethodfit
(self)
ETH/trainer/test_trajectory_AIO.py:48
↓ 2 callersMethodgenerate_memory
(self, train_dataset, filter_memory=False, threshold_distance=0.5)
models/model_AIO.py:208
↓ 2 callersMethodget_destination_from_memory
(self, past, abs_past, seq_start_end, end_pose, return_abs=False)
models/model_AIO.py:271
↓ 2 callersMethodget_memory_index_batch
(self, state_past, memory_past)
models/model_AIO.py:173
↓ 2 callersMethodget_sim
(self, past, abs_past, end_pose, gt_destination)
ETH/models/model_train_trajectory.py:330
↓ 2 callersFunctionisfolder
if '.' exists in the subfolder, the function still justifies it as a folder. e.g., /mnt/dome/adhoc_0.5x/abc is a folder if '.' exists after all sla
ETH/utils/utils.py:131
↓ 2 callersFunctionisinteger
(integer_test)
ETH/utils/utils.py:40
↓ 2 callersFunctionislist
(list_test)
ETH/utils/utils.py:64
↓ 2 callersFunctionmulti_linspace
This function is just like np.linspace, but will create linearly spaced vectors from a start to end vector. Inputs: a - Start vector.
ETH/data/homography_warper.py:203
↓ 2 callersMethodnode2edge
(self, x, rel_rec, rel_send)
ETH/models/layer_utils.py:81
↓ 2 callersMethodnode2edge
(self, x, rel_rec, rel_send)
models/layer_utils.py:97
↓ 2 callersFunctionnormal_transform_pixel
(height, width)
ETH/data/homography_warper.py:289
↓ 2 callersFunctionprepare_seed
(rand_seed)
test_MemoNet.py:9
↓ 2 callersFunctionprepare_seed
(rand_seed)
ETH/utils/utils.py:177
↓ 2 callersMethodreconstruct_trajectory
(self, past, abs_past, end_pose, future)
ETH/models/model_train_trajectory.py:196
↓ 2 callersFunctionsafe_list
copy a list to the buffer for use parameters: input_data: a list outputs: safe_data: a copy of input data
ETH/utils/utils.py:150
↓ 2 callersMethodstep
(self)
ETH/utils/torch.py:197
↓ 1 callersMethodAttentionLoss
(self, sim, distance)
trainer/trainer_AIO.py:100
↓ 1 callersMethodFutureData
(self, frame)
ETH/data/preprocessor.py:75
↓ 1 callersMethodFutureMotion
(self, DataTuple, valid_id)
ETH/data/preprocessor.py:137
↓ 1 callersMethodGetID
(self, data)
ETH/data/preprocessor.py:57
↓ 1 callersMethodPreData
(self, frame)
ETH/data/preprocessor.py:66
↓ 1 callersMethodPreMotion
(self, DataTuple, valid_id)
ETH/data/preprocessor.py:117
↓ 1 callersMethod__getattribute__
(self, name)
ETH/utils/config.py:46
↓ 1 callersMethod__init__
(self, *models)
ETH/utils/torch.py:28
↓ 1 callersMethod__init__
(self, data, homography, description=None)
ETH/data/map.py:13
↓ 1 callersMethod__init__
(self, cfg)
ETH/models/model_test_trajectory.py:45
↓ 1 callersMethod_evaluate
(self, generator)
ETH/trainer/train_trajectory_AIO.py:217
↓ 1 callersMethod_train_single_epoch
(self)
ETH/trainer/train_trajectory_AIO.py:156
↓ 1 callersMethod_train_single_epoch
(self)
trainer/trainer_AIO.py:110
↓ 1 callersMethodadd_memory
(self, past, abs_past, end_pose, future)
ETH/models/model_train_trajectory.py:121
↓ 1 callersFunctionangle_to_rotation_matrix
Creates a rotation matrix out of angles in degrees Args: angle: (torch.Tensor): tensor of angles in degrees, any shape. Returns:
ETH/data/homography_warper.py:28
↓ 1 callersMethodbatch_rotate
As the input is a map and the warp_affine works on an image coordinate system we would have to flip the y axis updown, negate the ang
ETH/data/map.py:72
↓ 1 callersFunctionconvert_points_from_homogeneous
r"""Function that converts points from homogeneous to Euclidean space. Examples:: >>> input = torch.rand(2, 4, 3) # BxNx3 >>> out
ETH/data/homography_warper.py:141
↓ 1 callersFunctionconvert_points_to_homogeneous
r"""Function that converts points from Euclidean to homogeneous space. Examples:: >>> input = torch.rand(2, 4, 3) # BxNx3 >>> out
ETH/data/homography_warper.py:125
↓ 1 callersFunctioncreate_batched_meshgrid
Generates a coordinate grid for an image. When the flag `normalized_coordinates` is set to True, the grid is normalized to be in the range [-1
ETH/data/homography_warper.py:217
↓ 1 callersMethoddecode_state_into_sim
(self, norm_past_state, abs_past_state_social, future)
models/model_AIO.py:138
↓ 1 callersMethoddecode_state_into_sim_warm
(self, norm_past_state, abs_past_state_social)
models/model_AIO.py:122
↓ 1 callersFunctiondeg2rad
r"""Function that converts angles from degrees to radians. Args: tensor (torch.Tensor): Tensor of arbitrary shape. Returns: to
ETH/data/homography_warper.py:14
↓ 1 callersMethodevaluate
(self, generator)
ETH/trainer/test_trajectory_AIO.py:84
↓ 1 callersMethodevaluate
(self, dataset)
trainer/test_final_trajectory.py:67
↓ 1 callersFunctionevaluate_addressor
(train_dataset, dataset, model, config, device)
trainer/evaluations.py:34
↓ 1 callersFunctionevaluate_intention
(dataset, model, config, device)
trainer/evaluations.py:5
↓ 1 callersFunctionevaluate_trajectory
(dataset, model, config, device)
trainer/evaluations.py:68
↓ 1 callersMethodfilter_memory
(self, generator)
ETH/models/model_train_trajectory.py:130
↓ 1 callersFunctionfind_min_dist
given two time frame arrays, find then min dist
sddloader.py:34
↓ 1 callersFunctionfind_min_time
given two time frame arrays, find then min dist (time)
sddloader.py:19
↓ 1 callersMethodfix_process_to_get_destination
(self, past, abs_past, seq_start_end, end_pose)
models/model_test_trajectory_res.py:112
↓ 1 callersMethodfixed_process_to_get_destination
(self, past, abs_past, seq_start_end, end_pose, future)
models/model_encdec_trajectory_social.py:74
↓ 1 callersMethodfixed_process_to_get_destination
(self, past, abs_past, seq_start_end, end_pose, future)
models/model_AIO.py:352
↓ 1 callersMethodgenerate_memory
(self)
ETH/trainer/train_trajectory_AIO.py:113
↓ 1 callersMethodget_cropped_maps_from_scene_map_batch
Returns rotated patches of each map around the transformed scene points. ___________________ | | | |
ETH/data/map.py:92
↓ 1 callersMethodget_destination
(self, past, abs_past, end_pose)
ETH/models/model_test_trajectory.py:105
↓ 1 callersMethodget_destination
(self, past, abs_past, end_pose)
ETH/models/model_train_trajectory.py:250
↓ 1 callersFunctionget_ethucy_split
(dataset)
ETH/data/ethucy_split.py:3
↓ 1 callersMethodget_heading
(self, cur_data, valid_id)
ETH/data/preprocessor.py:98
↓ 1 callersMethodget_memory_index
(self, state_past, memory_past)
models/model_selector_finetune_social.py:51
↓ 1 callersMethodget_memory_index
(self, state_past, memory_past)
models/model_selector_warm_up_social.py:50
↓ 1 callersMethodget_memory_index
(self, state_past, memory_past)
models/model_test_trajectory_res.py:65
↓ 1 callersMethodget_memory_index_batch
(self, state_past, memory_past)
models/model_selector_finetune_social.py:59
↓ 1 callersMethodget_memory_index_batch
(self, state_past, memory_past)
models/model_test_trajectory_res.py:100
↓ 1 callersMethodget_padded_map
(self, padding_x, padding_y, device)
ETH/data/map.py:59
↓ 1 callersMethodget_parameters
(self, mode)
ETH/models/model_train_trajectory.py:38
↓ 1 callersMethodget_pred_mask
(self, cur_data, valid_id)
ETH/data/preprocessor.py:92
↓ 1 callersFunctionget_rotation_matrix2d
r"""Calculates an affine matrix of 2D rotation. The function calculates the following matrix: .. math:: \begin{bmatrix} \a
ETH/data/homography_warper.py:48
↓ 1 callersMethodget_seq_and_frame
(self, index)
ETH/data/dataloader.py:54
↓ 1 callersMethodget_state_encoding
(self, past, abs_past, end_pose, future)
ETH/models/model_train_trajectory.py:238
↓ 1 callersMethodget_trajectory
(self, past, abs_past, seq_start_end, end_pose)
models/model_AIO.py:321
↓ 1 callersMethodget_valid_id
(self, pre_data, fut_data)
ETH/data/preprocessor.py:82
next →1–100 of 249, ranked by callers