MCPcopy Create free account

hub / github.com/GeWu-Lab/AnyTouch2 / functions

Functions813 in github.com/GeWu-Lab/AnyTouch2

⨍ Functions 813◇ Types & classes 138↳ Endpoints 2
Method__getitem__
(self, idx)
sparsh/tactile_ssl/data/vision_based_forces_slip_probes.py:140
Method__getitem__
Returns a single training item from the dataset as a dictionary.
sparsh/tactile_ssl/data/vision_tactile.py:93
Method__getstate__
(self)
sparsh/tactile_ssl/utils/signal_connector.py:180
Method__init__
(self, args, config, num_frames, tube_size)
model/tactile_mae.py:20
Method__init__
( self, num_queries=1, embed_dim=768, num_heads=16, mlp_ratio=4.0,
model/linear_probe.py:15
Method__init__
(self, embed_dim, num_heads, mlp_ratio, depth, norm_layer, init_std, qkv_bias, complete_block, classes=3)
model/linear_probe.py:102
Method__init__
(self, args, config, num_frames, tube_size)
model/linear_probe.py:153
Method__init__
Vector quantization using gumbel softmax Args: dim: input dimension (channels) num_vars: number of quantized vectors
model/layers/gumbel_vector_quantizer.py:12
Method__init__
(self, drop_prob=None)
model/layers/drop_path.py:29
Method__init__
( self, dim: int, num_heads: int, mlp_ratio: float = 4.0, qkv_bias: bo
model/layers/block.py:41
Method__init__
( self, dim: int, init_values: Union[float, Tensor] = 1e-5, inplace: bool = Fa
model/layers/layer_scale.py:16
Method__init__
( self, dim, num_heads=12, qkv_bias=False, proj_bias=True, att
model/layers/attention.py:99
Method__init__
( self, dim, num_heads, mlp_ratio=4.0, qkv_bias=False, act_lay
model/layers/attention.py:175
Method__init__
( self, in_features: int, hidden_features: Optional[int] = None, out_features:
model/layers/mlp.py:17
Method__init__
( self, dim, num_heads, mlp_ratio=4.0, qkv_bias=False, proj_bi
model/layers/decoder_block.py:40
Method__init__
( self, in_features: int, hidden_features: Optional[int] = None, out_features:
model/layers/swiglu_ffn.py:55
Method__init__
(self, args, mode='train')
data/downstream_dataset.py:23
Method__init__
(self, args, mode='train')
data/downstream_dataset.py:118
Method__init__
(self, args, mode='train')
data/downstream_dataset.py:223
Method__init__
(self, base_model: nn.Module, sensor_type: int)
sparsh/train_task.py:42
Method__init__
(self, backbone)
sparsh/tactile_ssl/utils/masking.py:22
Method__init__
(self, backbone)
sparsh/tactile_ssl/utils/masking.py:40
Method__init__
( self, crop_size: Union[int, Tuple[int, int]]=(224, 224), num_frames: int=16,
sparsh/tactile_ssl/utils/masking.py:106
Method__init__
(self)
sparsh/tactile_ssl/utils/__init__.py:197
Method__init__
(self, signal_handlers: Union[List[_HANDLER], _HANDLER])
sparsh/tactile_ssl/utils/signal_connector.py:28
Method__init__
(self, trainer)
sparsh/tactile_ssl/utils/signal_connector.py:42
Method__init__
( self, modal_dims: List[int], modal_lens: List[int], embed_dim: int,
sparsh/tactile_ssl/model/multimodal_transformer.py:28
Method__init__
(self, args, config)
sparsh/tactile_ssl/model/point_mae.py:31
Method__init__
(self, config: Union[CLIPVisionConfig, CLIPTextConfig])
sparsh/tactile_ssl/model/point_mae.py:70
Method__init__
(self, config: CLIPConfig)
sparsh/tactile_ssl/model/point_mae.py:183
Method__init__
(self, args, config, decoder_config, num_frames, add_time_attn, tube_size)
sparsh/tactile_ssl/model/point_mae.py:272
Method__init__
(self, config, decoder_config, num_frames, add_time_attn, tube_size, now_sensor='gelsight')
sparsh/tactile_ssl/model/point_mae.py:810
Method__init__
(self)
sparsh/tactile_ssl/model/pretrained.py:25
Method__init__
( self, optimizer, steps_per_epoch, start_lr, T_max, warmup_ep
sparsh/tactile_ssl/model/custom_scheduler.py:12
Method__init__
( self, input_dim: int, num_mask_tokens: int = 1, zero_init_mask_tokens: bool
sparsh/tactile_ssl/model/vision_transformer.py:426
Method__init__
Vector quantization using gumbel softmax Args: dim: input dimension (channels) num_vars: number of quantized vectors
sparsh/tactile_ssl/model/layers/gumbel_vector_quantizer.py:12
Method__init__
(self, drop_prob=None)
sparsh/tactile_ssl/model/layers/drop_path.py:29
Method__init__
( self, dim: int, num_heads: int, mlp_ratio: float = 4.0, qkv_bias: bo
sparsh/tactile_ssl/model/layers/block.py:44
Method__init__
( self, in_dim, out_dim, use_bn=False, nlayers=3, hidden_dim=2
sparsh/tactile_ssl/model/layers/dino_head.py:13
Method__init__
( self, dim: int, init_values: Union[float, Tensor] = 1e-5, inplace: bool = Fa
sparsh/tactile_ssl/model/layers/layer_scale.py:16
Method__init__
( self, dim, num_heads=12, qkv_bias=False, proj_bias=True, att
sparsh/tactile_ssl/model/layers/attention.py:102
Method__init__
( self, dim, num_heads, mlp_ratio=4.0, qkv_bias=False, act_lay
sparsh/tactile_ssl/model/layers/attention.py:178
Method__init__
( self, in_features: int, hidden_features: Optional[int] = None, out_features:
sparsh/tactile_ssl/model/layers/mlp.py:17
Method__init__
( self, dim, num_heads, mlp_ratio=4.0, qkv_bias=False, proj_bi
sparsh/tactile_ssl/model/layers/decoder_block.py:43
Method__init__
( self, in_features: int, hidden_features: Optional[int] = None, out_features:
sparsh/tactile_ssl/model/layers/swiglu_ffn.py:55
Method__init__
(self, num_frames: int, tubelet_size: int = 2, *args, **kw
sparsh/tactile_ssl/model/layers/patch_embed.py:120
Method__init__
( self, size: Union[int, List[int]], stride: Union[int, List[int]], embed_dim:
sparsh/tactile_ssl/model/layers/patch_embed.py:138
Method__init__
(self, config: DictConfig, dataset_name: str)
sparsh/tactile_ssl/data/vision_based_grasp_probes.py:25
Method__init__
( self, config: DictConfig, dataset_name: str, )
sparsh/tactile_ssl/data/vision_based_slip_probes.py:31
Method__init__
( self, config: DictConfig, dataset_name: str, )
sparsh/tactile_ssl/data/gelsight_grasp.py:22
Method__init__
( self, config: DictConfig, dataset_name: str, )
sparsh/tactile_ssl/data/digit_ycbslide.py:29
Method__init__
( self, config: DictConfig, digit_serial: str, gelsight_device_id: int, )
sparsh/tactile_ssl/data/vision_based_interactive.py:25
Method__init__
(self, config: DictConfig, dataset_name: str)
sparsh/tactile_ssl/data/vision_based_textile_probes.py:25
Method__init__
( self, config: DictConfig, dataset_name: str, )
sparsh/tactile_ssl/data/digit_slip.py:27
Method__init__
(self, config: DictConfig, dataset_name: str, is_train: bool = False)
sparsh/tactile_ssl/data/vision_based_pose_probes.py:25
Method__init__
( self, config: DictConfig, dataset_name: str, )
sparsh/tactile_ssl/data/vision_based_force_probes.py:143
Method__init__
( self, config: DictConfig, dataset_name: str, )
sparsh/tactile_ssl/data/vision_based_force_probes.py:151
Method__init__
( self, config: DictConfig, dataset_name: str, sensor_type: Literal["digit", "
sparsh/tactile_ssl/data/vision_tactile_forcefield.py:30
Method__init__
( self, config: DictConfig, dataset_name: str, is_train: bool = False, )
sparsh/tactile_ssl/data/vision_based_forces_slip_probes.py:32
Method__init__
( self, config: DictConfig, dataset_name: str, sensor_type: Literal["digit", "
sparsh/tactile_ssl/data/vision_tactile.py:30
Method__init__
( self, device, module: algorithm.Module, )
sparsh/tactile_ssl/test/test_t4_grasp.py:27
Method__init__
( self, device, module: algorithm.Module, )
sparsh/tactile_ssl/test/test_t2_slip.py:30
Method__init__
( self, device, module: algorithm.Module, )
sparsh/tactile_ssl/test/test_t1_force.py:27
Method__init__
( self, digit_serial: Optional[str], gelsight_device_id: Optional[int], device
sparsh/tactile_ssl/test/demo_t1_forcefield.py:26
Method__init__
( self, device, module: algorithm.Module, )
sparsh/tactile_ssl/test/test_task.py:13
Method__init__
( self, device, module: algorithm.Module, )
sparsh/tactile_ssl/test/test_t3_pose.py:28
Method__init__
( self, device, module: algorithm.Module, )
sparsh/tactile_ssl/test/test_t6_textile.py:29
Method__init__
( self, probe_name: str, decoder: nn.Module, loss_fn: partial, )
sparsh/tactile_ssl/probe/online_probe.py:19
Method__init__
( self, input_embed_dim: int = 768, *args, **kwargs )
sparsh/tactile_ssl/probe/reconstruction.py:41
Method__init__
( self, encoder: nn.Module, predictor: nn.Module, optim_cfg: partial,
sparsh/tactile_ssl/algorithm/ijepa.py:25
Method__init__
( self, encoder: nn.Module, dino_head: partial, optim_cfg: partial, lr
sparsh/tactile_ssl/algorithm/dinov2.py:32
Method__init__
( self, encoder: nn.Module, dino_head: partial, optim_cfg: partial, lr
sparsh/tactile_ssl/algorithm/dino.py:30
Method__init__
( self, encoder: nn.Module, predictor: nn.Module, optim_cfg: partial,
sparsh/tactile_ssl/algorithm/vjepa.py:36
Method__init__
( self, encoder: VisionTransformer, decoder: partial, optim_cfg: partial,
sparsh/tactile_ssl/algorithm/mae.py:21
Method__init__
Args: accelerator: The hardware to run on. Possible choices are: ``"cpu"``, ``"cuda"``, ``"mps"``, ``"gpu"``, ``"
sparsh/tactile_ssl/trainer/trainer.py:27
Method__init__
( self, image_size=(3, 224, 224), embed_dim='base', patch_size=16, res
sparsh/tactile_ssl/downstream_task/forcefield_sl.py:34
Method__init__
( self, embed_dim='base', num_heads=12, mlp_ratio=4.0, depth=1,
sparsh/tactile_ssl/downstream_task/pose_sl.py:36
Method__init__
( self, attn_pool=False, embed_dim='base', num_heads=12, mlp_ratio=4.0
sparsh/tactile_ssl/downstream_task/slip_decoders.py:71
Method__init__
( self, embed_dim=768, num_heads=12, mlp_ratio=4.0, depth=1, n
sparsh/tactile_ssl/downstream_task/attentive_pooler.py:110
Method__init__
( self, model_encoder: nn.Module, model_task: nn.Module, optim_cfg: partial,
sparsh/tactile_ssl/downstream_task/sl_module.py:19
Method__init__
( self, embed_dim='base', num_heads=12, mlp_ratio=4.0, depth=1,
sparsh/tactile_ssl/downstream_task/textile_sl.py:33
Method__init__
( self, model_encoder: nn.Module, model_task: nn.Module, optim_cfg: partial,
sparsh/tactile_ssl/downstream_task/slip_sl.py:31
Method__init__
( self, embed_dim='base', num_heads=12, mlp_ratio=4.0, depth=1,
sparsh/tactile_ssl/downstream_task/grasp_sl.py:28
Method__init__
( self, embed_dim='base', num_heads=12, mlp_ratio=4.0, depth=1,
sparsh/tactile_ssl/downstream_task/force_sl.py:26
Method__init__
(self, config, frame_ids, ssim_loss=None)
sparsh/tactile_ssl/downstream_task/utils_forcefield/ssl_flow_loss.py:12
Method__init__
(self, height, width)
sparsh/tactile_ssl/downstream_task/utils_forcefield/ssl_utils.py:148
Method__init__
(self, features)
sparsh/tactile_ssl/downstream_task/utils_forcefield/layers/Fusion.py:12
Method__init__
(self, start_index=0)
sparsh/tactile_ssl/downstream_task/utils_forcefield/layers/Reassemble.py:14
Method__init__
(self, in_features, start_index=1)
sparsh/tactile_ssl/downstream_task/utils_forcefield/layers/Reassemble.py:36
Method__init__
(self, conv, output_size)
sparsh/tactile_ssl/downstream_task/utils_forcefield/layers/Reassemble.py:48
Method__init__
(self, p, s, h, emb_dim, resample_dim)
sparsh/tactile_ssl/downstream_task/utils_forcefield/layers/Reassemble.py:59
Method__init__
p = patch size s = coefficient resample emb_dim <=> D (in the paper) resample_dim <=> ^D (in the paper) read
sparsh/tactile_ssl/downstream_task/utils_forcefield/layers/Reassemble.py:106
Method__init__
(self, scale_factor, mode, align_corners=False)
sparsh/tactile_ssl/downstream_task/utils_forcefield/layers/Head.py:12
Method__init__
(self, in_channels, out_channels)
sparsh/tactile_ssl/downstream_task/utils_forcefield/layers/Head.py:32
Method__init__
(self, features, use_skips=True)
sparsh/tactile_ssl/downstream_task/utils_forcefield/layers/Head.py:68
Method__init__
( self, num_ch_enc, num_input_features, num_frames_to_predict_for=None, stride=1 )
sparsh/tactile_ssl/downstream_task/utils_forcefield/pose_estimator/pose_decoder.py:17
Method__init__
(self, block, layers, num_classes=1000, num_input_images=1)
sparsh/tactile_ssl/downstream_task/utils_forcefield/pose_estimator/resnet_encoder.py:22
Method__init__
( self, num_encoder_layers, frame_ids=[0, -1], )
sparsh/tactile_ssl/downstream_task/utils_forcefield/pose_estimator/PoseEstimator.py:14
Method__init__
(self)
sparsh/tactile_ssl/loss/koleo_loss.py:21
Method__init__
(self, patch_out_dim, student_temp=0.1, center_momentum=0.9)
sparsh/tactile_ssl/loss/ibot_patch_loss.py:36
← previousnext →401–500 of 813, ranked by callers