Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/GeWu-Lab/AnyTouch2
/ functions
Functions
813 in github.com/GeWu-Lab/AnyTouch2
⨍
Functions
813
◇
Types & classes
138
↳
Endpoints
2
Method
forward
(self, x)
sparsh/tactile_ssl/model/layers/drop_path.py:33
Method
forward
(self, x_or_x_list)
sparsh/tactile_ssl/model/layers/block.py:252
Method
forward
(self, x)
sparsh/tactile_ssl/model/layers/dino_head.py:43
Method
forward
(self, x: Tensor)
sparsh/tactile_ssl/model/layers/layer_scale.py:26
Method
forward
(self, x: Tensor, attn_bias=None)
sparsh/tactile_ssl/model/layers/attention.py:81
Method
forward
(self, q: Tensor, x: Tensor)
sparsh/tactile_ssl/model/layers/attention.py:123
Method
forward
(self, q: Tensor, x: Tensor, attn_bias=None)
sparsh/tactile_ssl/model/layers/attention.py:155
Method
forward
(self, q, x)
sparsh/tactile_ssl/model/layers/attention.py:196
Method
forward
(self, x: Tensor)
sparsh/tactile_ssl/model/layers/mlp.py:34
Method
forward
(self, q, kv)
sparsh/tactile_ssl/model/layers/decoder_block.py:107
Method
forward
(self, x: Tensor)
sparsh/tactile_ssl/model/layers/swiglu_ffn.py:30
Method
forward
(self, x: Tensor)
sparsh/tactile_ssl/model/layers/patch_embed.py:84
Method
forward
Args: pos: Position (normalised Index [-1, 1]) tensor of shape (num_pos, num_dims) Returns: encoded_pos: Enco
sparsh/tactile_ssl/model/layers/patch_embed.py:183
Method
forward
(self, embedding: torch.Tensor, target: torch.Tensor, **kwargs)
sparsh/tactile_ssl/probe/online_probe.py:34
Method
forward
(self, x, img_shape)
sparsh/tactile_ssl/probe/reconstruction.py:30
Method
forward
(self, x, img_shape, ids_restore)
sparsh/tactile_ssl/probe/reconstruction.py:56
Method
forward
(self, encoder_activations, mode="normal_shear")
sparsh/tactile_ssl/downstream_task/forcefield_sl.py:65
Method
forward
(self, x)
sparsh/tactile_ssl/downstream_task/pose_sl.py:83
Method
forward
(self, inputs)
sparsh/tactile_ssl/downstream_task/slip_decoders.py:54
Method
forward
(self, inputs)
sparsh/tactile_ssl/downstream_task/slip_decoders.py:116
Method
forward
(self, x)
sparsh/tactile_ssl/downstream_task/attentive_pooler.py:98
Method
forward
(self, x)
sparsh/tactile_ssl/downstream_task/attentive_pooler.py:137
Method
forward
(self, x, *args, **kwargs)
sparsh/tactile_ssl/downstream_task/sl_module.py:106
Method
forward
(self, x)
sparsh/tactile_ssl/downstream_task/textile_sl.py:67
Method
forward
(self, x)
sparsh/tactile_ssl/downstream_task/grasp_sl.py:60
Method
forward
(self, x)
sparsh/tactile_ssl/downstream_task/force_sl.py:64
Method
forward
(self, x, y)
sparsh/tactile_ssl/downstream_task/utils_forcefield/ssl_flow_loss.py:135
Method
forward
(self, depth, inv_K)
sparsh/tactile_ssl/downstream_task/utils_forcefield/ssl_utils.py:164
Method
forward
(self, points, K, T)
sparsh/tactile_ssl/downstream_task/utils_forcefield/ssl_utils.py:192
Method
forward
Forward pass. Args: x (tensor): input Returns: tensor: output
sparsh/tactile_ssl/downstream_task/utils_forcefield/layers/Fusion.py:23
Method
forward
(self, x, previous_stage=None)
sparsh/tactile_ssl/downstream_task/utils_forcefield/layers/Fusion.py:44
Method
forward
(self, x)
sparsh/tactile_ssl/downstream_task/utils_forcefield/layers/Reassemble.py:18
Method
forward
(self, x)
sparsh/tactile_ssl/downstream_task/utils_forcefield/layers/Reassemble.py:27
Method
forward
(self, x)
sparsh/tactile_ssl/downstream_task/utils_forcefield/layers/Reassemble.py:41
Method
forward
(self, x)
sparsh/tactile_ssl/downstream_task/utils_forcefield/layers/Reassemble.py:53
Method
forward
(self, x)
sparsh/tactile_ssl/downstream_task/utils_forcefield/layers/Reassemble.py:99
Method
forward
(self, x)
sparsh/tactile_ssl/downstream_task/utils_forcefield/layers/Reassemble.py:135
Method
forward
(self, x)
sparsh/tactile_ssl/downstream_task/utils_forcefield/layers/Head.py:19
Method
forward
(self, x)
sparsh/tactile_ssl/downstream_task/utils_forcefield/layers/Head.py:38
Method
forward
(self, x)
sparsh/tactile_ssl/downstream_task/utils_forcefield/layers/Head.py:56
Method
forward
(self, input_features, mode="normal_shear")
sparsh/tactile_ssl/downstream_task/utils_forcefield/layers/Head.py:87
Method
forward
(self, input_features)
sparsh/tactile_ssl/downstream_task/utils_forcefield/pose_estimator/pose_decoder.py:39
Method
forward
(self, input_image)
sparsh/tactile_ssl/downstream_task/utils_forcefield/pose_estimator/resnet_encoder.py:100
Method
forward
Predict poses between input frames for monocular sequences.
sparsh/tactile_ssl/downstream_task/utils_forcefield/pose_estimator/PoseEstimator.py:30
Method
forward
Args: student_output (BxD): backbone output of student
sparsh/tactile_ssl/loss/koleo_loss.py:38
Method
forward
Cross-entropy between softmax outputs of the teacher and student networks. student_patch_tokens: (B, N, D) tensor teacher_pat
sparsh/tactile_ssl/loss/ibot_patch_loss.py:98
Method
forward
Cross-entropy between softmax outputs of the teacher and student networks.
sparsh/tactile_ssl/loss/dino_loss.py:64
Method
forward_decoder
(self, x, ids_restore, sensor_type=None, data_type = None)
sparsh/tactile_ssl/model/point_mae.py:1158
Method
forward_diff_decoder
(self, x, ids_restore, sensor_type=None, data_type = None)
sparsh/tactile_ssl/model/point_mae.py:1232
Method
forward_idx
(self, x)
model/layers/gumbel_vector_quantizer.py:144
Method
forward_idx
(self, x)
sparsh/tactile_ssl/model/layers/gumbel_vector_quantizer.py:144
Method
forward_loss_img
(self, x, pred, mask, data_type = None)
sparsh/tactile_ssl/model/point_mae.py:1350
Method
forward_loss_pts
(self, points, pred, mask, data_type = None)
sparsh/tactile_ssl/model/point_mae.py:1372
Method
forward_masked
( self, student_patch_tokens_masked, teacher_patch_tokens_masked, student_mask
sparsh/tactile_ssl/loss/ibot_patch_loss.py:116
Method
forward_point_decoder
(self, points, points_num, ids_restore, sensor_type=None, data_type = None)
sparsh/tactile_ssl/model/point_mae.py:1301
Method
forward_with_x
(self, x: Tensor, normalized_coords: bool = False)
sparsh/tactile_ssl/model/layers/patch_embed.py:215
Function
get_2d_sincos_pos_embed
grid_size: int of the grid height and width return: pos_embed: [grid_size*grid_size, embed_dim] or [1+grid_size*grid_size, embed_dim] (w/
model/util/pos_embed.py:20
Function
get_2d_sincos_pos_embed
grid_size: int of the grid height and width return: pos_embed: [grid_size*grid_size, embed_dim] or [1+grid_size*grid_size, embed_dim] (w/
util/pos_embed.py:20
Method
get_activation
(name)
sparsh/tactile_ssl/downstream_task/forcefield_sl.py:159
Function
get_branges_scales
(xs: List[Tensor], sample_drop_ratio: float = 0.0)
model/layers/decoder_block.py:170
Function
get_branges_scales
(xs: List[Tensor], sample_drop_ratio: float = 0.0)
sparsh/tactile_ssl/model/layers/decoder_block.py:173
Function
get_entry_or
(cfg, key, default)
utils.py:62
Function
get_entry_or
(cfg, key, default)
sparsh/tactile_ssl/model/utils.py:62
Function
get_global_value
()
sparsh/tactile_ssl/model/point_mae.py:26
Method
get_intermediate_layers
(self, x, layers: List[int]=None)
sparsh/tactile_ssl/model/pretrained.py:37
Method
get_intermediate_layers
( self, x: torch.Tensor, n: Union[int, Sequence] = 1, # Layers or n last layers to ta
sparsh/tactile_ssl/model/vision_transformer.py:392
Method
get_lr
(self)
sparsh/tactile_ssl/model/custom_scheduler.py:29
Method
get_overall_metrics
(self, dataset, over_all_outputs=False)
sparsh/tactile_ssl/test/test_t4_grasp.py:71
Method
get_overall_metrics
(self, dataset, over_all_outputs=False)
sparsh/tactile_ssl/test/test_t2_slip.py:82
Method
get_overall_metrics
(self, dataset, over_all_outputs=False)
sparsh/tactile_ssl/test/test_t1_force.py:66
Method
get_overall_metrics
( self, dataset: data.Dataset, over_all_outputs: bool = False )
sparsh/tactile_ssl/test/test_task.py:39
Method
get_overall_metrics
(self, dataset, over_all_outputs=False)
sparsh/tactile_ssl/test/test_t3_pose.py:91
Method
get_overall_metrics
(self, dataset, over_all_outputs=False)
sparsh/tactile_ssl/test/test_t6_textile.py:73
Function
get_world_size
()
utils.py:45
Function
get_world_size
()
sparsh/tactile_ssl/model/utils.py:45
Method
global_avg
(self)
util/misc.py:66
Method
hook
(model, input, output)
sparsh/tactile_ssl/downstream_task/forcefield_sl.py:160
Function
init_distributed_mode
(args)
util/misc.py:216
Method
init_head
(self)
model/linear_probe.py:212
Method
init_head
(self)
sparsh/tactile_ssl/model/point_mae.py:1456
Function
init_weights_vit_timm
vit weight initialization, original timm impl (for reproducibility)
sparsh/tactile_ssl/model/vision_transformer.py:494
Method
initialize_decoder
(self)
sparsh/tactile_ssl/model/point_mae.py:352
Method
initialize_decoder
(self)
sparsh/tactile_ssl/model/point_mae.py:928
Function
interpolate_pos_embed
(model, checkpoint_model)
model/util/pos_embed.py:75
Function
interpolate_pos_embed
(model, checkpoint_model)
sparsh/tactile_ssl/model/util/pos_embed.py:75
Function
interpolate_pos_embed
(model, checkpoint_model)
util/pos_embed.py:75
Function
is_main_process
()
utils.py:55
Function
is_main_process
()
sparsh/tactile_ssl/model/utils.py:55
Function
load_label_from_csv
Load a csv file with pandas
utils.py:58
Function
load_label_from_csv
Load a csv file with pandas
sparsh/tactile_ssl/model/utils.py:58
Function
load_model
(args, model_without_ddp, optimizer, loss_scaler)
util/misc.py:315
Function
load_model_from_clip
(ckpt, model)
sparsh/train_task.py:79
Function
load_model_from_multi_old
(ckpt, model)
sparsh/train_task.py:65
Function
load_sample
(img, img_bg=None, enhance=False)
sparsh/tactile_ssl/data/digit/utils.py:172
Function
logging
(s, verbose=True, style=1)
utils.py:33
Function
logging
(s, verbose=True, style=1)
sparsh/tactile_ssl/model/utils.py:33
Function
make_dataset_pie_plot
(d, title=None, show=False)
utils.py:67
Function
make_dataset_pie_plot
(d, title=None, show=False)
sparsh/tactile_ssl/model/utils.py:67
Method
make_plots
(self, test_dset)
sparsh/tactile_ssl/test/test_t2_slip.py:165
Method
make_plots
(self, dataset, params: Optional[dict] = None)
sparsh/tactile_ssl/test/test_t1_force.py:124
← previous
next →
601–700 of 813, ranked by callers