MCPcopy Create free account

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

Functions249 in github.com/MediaBrain-SJTU/MemoNet

↓ 1 callersFunctionhomography_warp
r"""Function that warps image patchs or tensors by homographies. See :class:`~kornia.geometry.warp.HomographyWarper` for details. Args:
ETH/data/homography_warper.py:252
↓ 1 callersMethodinit_adj
(self, num_ped, batch_num)
ETH/models/layer_utils.py:91
↓ 1 callersMethodinit_adj
(self, num_ped)
models/layer_utils.py:104
↓ 1 callersMethodinit_parameters
(self)
ETH/models/layer_utils.py:253
↓ 1 callersMethodinit_parameters
(self)
models/layer_utils.py:274
↓ 1 callersMethodinitial_memory
(self)
ETH/models/model_train_trajectory.py:115
↓ 1 callersFunctioninitial_pos
(traj_batches)
sddloader.py:180
↓ 1 callersFunctioninitial_pos_new
(traj_batches)
sddloader.py:188
↓ 1 callersFunctionis_path_creatable
if any previous level of parent folder exists, returns true
ETH/utils/utils.py:96
↓ 1 callersFunctionis_path_exists_or_creatable
(pathname)
ETH/utils/utils.py:117
↓ 1 callersFunctionisfloat
(float_test)
ETH/utils/utils.py:47
↓ 1 callersFunctionislogical
(logical_test)
ETH/utils/utils.py:56
↓ 1 callersFunctionisnparray
(nparray_test)
ETH/utils/utils.py:36
↓ 1 callersFunctionisscalar
(scalar_test)
ETH/utils/utils.py:51
↓ 1 callersMethodk_means
return clustering ncluster of x. Args: x (Tensor): B, K, 2 ncluster (int, optional): Number of clusters. Defaults to
models/model_AIO.py:184
↓ 1 callersMethodk_means
return clustering ncluster of x. Args: x (Tensor): B, K, 2 ncluster (int, optional): Number of clusters. Defaults to
models/model_test_trajectory_res.py:76
↓ 1 callersMethodload_scene_map
(self)
ETH/data/preprocessor.py:104
↓ 1 callersFunctionmain
(config)
train_MemoNet.py:35
↓ 1 callersFunctionmain
(config)
test_MemoNet.py:42
↓ 1 callersFunctionmain
(config)
ETH/train.py:16
↓ 1 callersFunctionmain
(config)
ETH/test.py:16
↓ 1 callersMethodmake_nmp_mlp
(self)
ETH/models/layer_utils.py:66
↓ 1 callersMethodmake_nmp_mlp
(self)
models/layer_utils.py:69
↓ 1 callersFunctionmark_similar
(mask, sim_list)
sddloader.py:60
↓ 1 callersMethodnext_sample
(self)
ETH/data/dataloader.py:72
↓ 1 callersFunctionoffset_pos
(traj_batches)
sddloader.py:196
↓ 1 callersFunctionparse_config
()
train_MemoNet.py:5
↓ 1 callersFunctionparse_config
()
test_MemoNet.py:16
↓ 1 callersFunctionparse_config
()
ETH/train.py:5
↓ 1 callersFunctionparse_config
()
ETH/test.py:5
↓ 1 callersMethodprint_model_param
(self, model)
trainer/trainer_AIO.py:63
↓ 1 callersFunctionrecreate_dirs
(*dirs)
ETH/utils/utils.py:82
↓ 1 callersMethodreset
(self)
ETH/utils/utils.py:21
↓ 1 callersMethodreset_parameters
(self)
ETH/models/layer_utils.py:204
↓ 1 callersMethodreset_parameters
(self)
models/layer_utils.py:223
↓ 1 callersMethodrotate_traj
(self, past, future, past_abs)
ETH/trainer/test_trajectory_AIO.py:52
↓ 1 callersMethodshuffle
(self)
ETH/data/dataloader.py:51
↓ 1 callersFunctionsocial_and_temporal_filter
(p1_key, p2_key, all_data_dict, time_thresh=48, dist_tresh=100)
sddloader.py:47
↓ 1 callersFunctionsrc_norm_to_dst_norm
(dst_pix_trans_src_pix: torch.Tensor, dsize_src: Tuple[int, int], dsize_dst: Tuple[in
ETH/data/homography_warper.py:303
↓ 1 callersMethodto_map_points
(self, scene_pts)
ETH/data/map.py:24
↓ 1 callersMethodtorch_map
(self, device)
ETH/data/map.py:48
↓ 1 callersFunctiontransform_points
r"""Function that applies transformations to a set of points. Args: trans_01 (torch.Tensor): tensor for transformations of shape
ETH/data/homography_warper.py:168
↓ 1 callersFunctiontransform_warp_impl
Compute the transform in normalized cooridnates and perform the warping.
ETH/data/homography_warper.py:324
↓ 1 callersFunctionwarp_affine_crop
r"""Applies an affine transformation to a tensor. The function warp_affine transforms the source tensor using the specified matrix: .. m
ETH/data/homography_warper.py:421
↓ 1 callersMethodwarp_grid
r"""Computes the grid to warp the coordinates grid by an homography. Args: dst_homo_src (torch.Tensor): Homography or homographies
ETH/data/homography_warper.py:368
Method__call__
(self, frame)
ETH/data/preprocessor.py:157
Method__call__
(self)
ETH/data/dataloader.py:81
Method__enter__
(self)
ETH/utils/torch.py:34
Method__enter__
(self)
ETH/utils/torch.py:51
Method__enter__
(self)
ETH/utils/torch.py:68
Method__enter__
(self)
ETH/utils/torch.py:85
Method__exit__
(self, *args)
ETH/utils/torch.py:37
Method__exit__
(self, *args)
ETH/utils/torch.py:54
Method__exit__
(self, *args)
ETH/utils/torch.py:71
Method__exit__
(self, *args)
ETH/utils/torch.py:88
Method__init__
Initialization
sddloader.py:273
Method__init__
Initialization
sddloader.py:342
Method__init__
(self, cfg_id, cfg_info, tmp=False, create_dirs=False)
ETH/utils/config.py:12
Method__init__
(self)
ETH/utils/utils.py:18
Method__init__
(self, device, *models)
ETH/utils/torch.py:45
Method__init__
(self, *models)
ETH/utils/torch.py:62
Method__init__
(self, *models)
ETH/utils/torch.py:79
Method__init__
(self, start, finish, rate, cur_epoch=0)
ETH/utils/torch.py:190
Method__init__
(self, data_root, seq_name, parser, log, split='train', phase='training')
ETH/data/preprocessor.py:9
Method__init__
(self, parser, log, split='train', phase='training')
ETH/data/dataloader.py:10
Method__init__
(self, data, homography, origin=None, description=None)
ETH/data/map.py:36
Method__init__
( self, x_min: torch.Tensor, y_min: torch.Tensor, x_max: torch
ETH/data/homography_warper.py:349
Method__init__
The Trainer class handles the training procedure for training the autoencoder. :param config: configuration parameters (see train_ae.py)
ETH/trainer/test_trajectory_AIO.py:19
Method__init__
The Trainer class handles the training procedure for training the autoencoder. :param config: configuration parameters (see train_ae.py)
ETH/trainer/train_trajectory_AIO.py:20
Method__init__
( self, embedding_dim=64, h_dim=64, mlp_dim=1024, bottleneck_dim=64, activation='relu', batch_norm=False,
ETH/models/layer_utils.py:38
Method__init__
(self)
ETH/models/layer_utils.py:191
Method__init__
(self, past_len, future_len)
ETH/models/layer_utils.py:231
Method__init__
(self)
ETH/models/model_test_trajectory.py:10
Method__init__
(self, cfg)
ETH/models/model_train_trajectory.py:10
Method__init__
(self)
ETH/models/model_train_trajectory.py:50
Method__init__
(self, config)
trainer/test_final_trajectory.py:12
Method__init__
The Trainer class handles the training procedure for training the autoencoder. :param config: configuration parameters (see train_ae.
trainer/trainer_AIO.py:14
Method__init__
( self, embedding_dim=64, h_dim=64, mlp_dim=1024, bottleneck_dim=1024, activation='relu', batc
models/layer_utils.py:41
Method__init__
(self)
models/layer_utils.py:210
Method__init__
(self, past_len, future_len)
models/layer_utils.py:252
Method__init__
(self, settings, pretrained_model)
models/model_encdec_trajectory_social.py:12
Method__init__
(self, settings, pretrained_model)
models/model_selector_finetune_social.py:11
Method__init__
(self, settings, pretrained_model)
models/model_selector_warm_up_social.py:10
Method__init__
(self, settings)
models/model_encdec_social.py:10
Method__init__
(self, settings, pretrained_model=None)
models/model_AIO.py:11
Method__init__
(self, settings, pretrained_model)
models/model_test_trajectory_res.py:9
Method__setattr__
(self, name, value)
ETH/utils/config.py:53
Methodas_image
(self)
ETH/data/map.py:18
Methodas_image
(self)
ETH/data/map.py:54
Functionbatch_to
(dst, *args)
ETH/utils/torch.py:94
Functioncalc_interaction_nba
(x)
sddloader.py:65
Functioncalculate_loss
(x, reconstructed_x, mean, log_var, criterion, future, interpolated_future)
sddloader.py:205
Functioncalculate_loss_double
(x, reconstructed_x, mean, log_var, criterion, future, interpolated_future, mean2,log_var2)
sddloader.py:258
Functioncalculate_loss_mm
(x, reconstructed_x, mean, log_var, criterion, future, interpolated_future)
sddloader.py:217
Functioncalculate_loss_multi
(x, reconstructed_x, mean, log_var, criterion, future, interpolated_future)
sddloader.py:239
Functioncompute_flat_grad
(output, inputs, filter_input_ids=set(), retain_graph=False, create_graph=False)
ETH/utils/torch.py:134
Functionconvert_secs2time
format second to human readable way
ETH/utils/utils.py:68
Functionfilter_state_dict
(state_dict, filter_keys)
ETH/utils/torch.py:164
Functionfind_unique_common_from_lists
find common items from 2 lists, the returned elements are unique. repetitive items will be ignored if the common items in two elements are not in t
ETH/utils/utils.py:216
Methodfit
(self)
ETH/trainer/train_trajectory_AIO.py:52
← previousnext →101–200 of 249, ranked by callers